CF1551B Wonderful Coloring

CF1551B1 Wonderful Coloring - 1

题目描述

AC代码：

#include<bits/stdc++.h>
using namespace std;
long long num,ans,n;
int a[30];
string p;
int main(){
	int t;
	cin>>t;
	while(t--){
		memset(a,0,sizeof(a));
		num=ans=0;
		cin>>p;
		for(int i=0;i<p.length();i++)
			a[p[i]-'a'+1]++;
		for(int i=1;i<=26;i++){
			if(a[i]>=2) ans++;
			if(a[i]==1) num++;
		}
		//for(int i=1;i<=26;i++) cout<<a[i]<<" ";
		//cout<<endl; 
		ans+=num/2;
		cout<<ans<<endl;
	}
	return 0;
}

CF1551B2 Wonderful Coloring - 2

题目描述

AC代码：

#include<bits/stdc++.h>
using namespace std;
int main() {
	int t,n,k;
	cin>>t;
	while(t--){
		int cnt[200005],ans[200005],x;
		vector<pair <int,int> > v;
		cin>>n>>k;
		for(int i=0;i<=n;i++)
			ans[i]=cnt[i]=0;
		for(int i=0;i<n;i++){
			cin>>x;	
			if(cnt[x]<k) v.push_back({x,i});
            cnt[x]++;		
		}
		sort(v.begin(),v.end());
		int groups=v.size()/k;
		for(int i=0;i<groups*k;i++){
			ans[v[i].second]=i%k+1;
		}
		for(int i=0;i<n;i++) cout<<ans[i]<<" ";
		cout<<endl;
	}
	return 0;
}