得到至少k个相同的数

codeofrces 1213D2

题意:

     给一个长度为n(<=2e5)的序列，可以进行操作，每次可以对一个元素除以二向下取整。需要得到至少k（<=n)个相同的数，可以把这些数全部变成零，所以肯定有解。问至少需要多少次操作。

题目链接

思路：

    因为是除以二向下取整，所以不会出现超过2e5的数，对原数组的每一个数进行操作，把这个数可能得到的所有数（包括它自己）和与之对应的操作次数记录下来。然后从0到2e5，进行枚举。并且更新答案，找出最小值。

#pragma warning(disable:4996)
#include<iostream>
#include<cstring>
#include<cstdio>
#include<set>
#include<string>
#include<cmath>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
vector<int> t[200005];
int main()
{
	int n, k, i, j, a, cnt, ans;
	while (scanf("%d%d", &n, &k) == 2) {
		for (i = 0; i <= 200000; i++)t[i].clear();
		while (n--) {
			scanf("%d", &a);
			cnt = 0;
			while (a) {
				t[a].push_back(cnt);
				//记录下这些数和操作次数。
				a /= 2;
				cnt++;

			}
			t[0].push_back(cnt);
		}
		ans = 20000000;
		for (i = 0; i <= 200000; i++) {
			cnt = 0;
			if (t[i].size() < k)continue;
			sort(t[i].begin(), t[i].end());
			//排序，然后取操作数最小的k个数
			for (j = 0; j < k; j++) {
				cnt += t[i][j];
			}
			ans = min(ans, cnt);
		}
		printf("%d\n", ans);
	}
	return 0;
}