#99-【桶排序】摩天楼

Description

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

Input

多组测试数据。
第一行，一个整数G，表示有G组测试数据。1 <= G <= 5。 
每组测试数据格式如下：
第一行，N和M 。 <= N <= 50。 1 <= M <= 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;
}