#99-【桶排序】摩天楼

版权声明:反正也没有人会转,下一个 https://blog.csdn.net/drtlstf/article/details/82729202

Description

有N栋摩天楼,第i栋摩天楼的高度是Hi。FJ有一种魔法,对一栋摩天楼用一次魔法就可以使得该摩天楼的高度加1。同一栋摩天楼可以多次使用魔法。FJ的目标是使得至少有M栋摩天楼的高度是相同的。问至少需要使用多少次魔法?

Input

多组测试数据。
第一行,一个整数G,表示有G组测试数据。1 <= G <= 5。 
每组测试数据格式如下:
第一行,N和M 。 <= N <= 50。 1 <= M <= N。
第二行,N个整数,第i个整数表示Hi,1 <= Hi <=50

Output

共G行,每行一个整数。

Sample Input

3
6 3
1 1 2 2 3 3 
3 1
1 2 3 
8 5
1 1 1 1 50 50 50 50 

Sample Output

1
0
49

HINT

先桶排序,随后计算最小高度。

#include <iostream>
#include <cstring>
#include <algorithm>

#define SIZE 55
#define INF 2e+09

using namespace std;

int bucket[SIZE];

int main(int argc, char** argv)
{
	int t, n, m, i, j, x, _max, tot, tot2, temp, res;
	
	scanf("%d", &t);
	while (t--)
	{
		_max = tot = 0; // 对于每组数据的初始化
		memset(bucket, 0, sizeof (bucket));
		scanf("%d%d", &n, &m);
		for (i = 0; i < n; ++i) // 进行桶排序
		{
			scanf("%d", &x);
			++bucket[x];
			_max = max(_max, x);
		}
		res = INF;
		for (i = 1; i <= _max; ++i)
		{
			temp = tot2 = 0;
			tot += bucket[i];
			if ((tot < m) || (!bucket[i])) // 还没有凑齐m栋或这个高度没有楼,跳过(第二个条件是一种贪心)
			{
				continue;
			}
			for (j = i; j; --j)
			{
				tot2 += bucket[j];
				if (tot2 < m) // 计算以这个高度为基准需要的花费
				{
					temp += bucket[j] * (i - j);
				}
				else
				{
					temp += (m - tot2 + bucket[j]) * (i - j);
					break;
				}
			}
			res = min(res, temp);
		}
		printf("%d\n", res);
	}
	
	return 0;
}

猜你喜欢

转载自blog.csdn.net/drtlstf/article/details/82729202