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;
}