Codeforces Round #734 (Div. 3)——B2

Codeforces Round #734 (Div. 3)

题意

1. 给你n个数字，和k种颜色，现在给这n个数字上色，每个数字可以不涂色
2. k种颜色必须都用到，且每种颜色用到的数量相同
3. 相同的数字颜色不能相同
4. 尽可能多的涂色

思路

• 首先我们肯定，如果某种数字的个数超过了k，那么超过的部分一定不能涂色 。然后现在我们用结构体存储每种数字，和它的下标，超过的部分不存储。
• 现在我们对存储过的数字，进行染色。假设现在有tt个数字，我们将这些数字按照值从小到大排序（从大到小也行），按照题目的要求，我们染色的数字个数，必须是k倍数。令 m=tt- tt % k。我们遍历这m个数字，对每个数字染色, 颜色从1，2，…k这样分配，剩余的数字无法染色。这样既能保证同种数字的颜色都不相同，又能保证恰好染了k的倍数个数字

代码

#include<bits/stdc++.h>

#define mem(a,b) memset(a,b,sizeof a)
using namespace std;
const int N =2e5+9;
int col[N],cnt[N];
struct node
{
	int x,idx;
	bool operator<(const node &t) const
	{
		return x<t.x;
	}
}e[N];
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		int n,k;
		mem(col,0);
		mem(cnt,0);
		scanf("%d%d",&n,&k);
		int tt=0;
		for(int i=0; i<n; i++)
		{
			int x;
			scanf("%d",&x);
			if(cnt[x]<k) e[tt++]={x,i};
			cnt[x]++;
		}
		sort(e,e+tt);
		int m=tt-tt%k;
		int c=0;
		for(int i=0; i<m; i++)
		{
			col[e[i].idx]=c+1;
			c++;
			c%=k;
		}
		for(int i=0; i<n; i++)
		{
			printf("%d ",col[i]);
		}
		puts("");
	}
	
	return 0;
 } 

总结

这道题开复现赛的时候没做出来，明明不难，就是没想到。关键一点在于，我没想到把这些数字按照值的的大小排序，害，继续加油