题目链接:Problem - B2 - Codeforces
这是一道神奇的题,写完之和和别人交流发现是用思维写的,然而我用二分去写的(写出来了是没想到) 要处理很多东西,然后发现了用神奇的优先队列可以优化到n * log(n) * log(n),再加上数据范围不是很大,最大是2x10^7 , 一秒不到就可以过。二分的是他的每个颜色的最大长度,并且用优先队列去处理判断的条件。
题目:
ac代码:
#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#include<unordered_map>
using namespace std;
typedef long long int LL ;
typedef pair<int,int >PII ;
const int N = 2e5 + 10;
int a[N];
int n, k ;
vector<int> ansv[N];
int res[N];
bool check(int mid , priority_queue<int> q)
{
for(int i = 1; i <= k ; i ++)
{
vector<int> v;
for(int j = 1; j <= mid ; j ++)
{
if(!q.size())
{
return false;
}
int x = q.top();
q.pop();
if(x <= 0)return false;
v.push_back(x-1);
}
for(auto it : v)q.push(it);
}
return true;
}
int main()
{
ios :: sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t;
cin >> t;
while(t--)
{
cin >> n >> k ;
unordered_map<int,int>mp;
for(int i = 1; i <= n ; i++)
{
cin >> a[i];
mp[a[i]] ++ ;
res[i] = 0 ;
ansv[i].clear();
}
priority_queue<int> q;
for(auto it : mp)q.push(it.second);
for(int i = 1; i <= n ; i ++)
{
ansv[a[i]].push_back(i);
}
int l = 1, r = n ;
while(l < r)
{
int mid = l + r + 1>> 1;
if(check(mid,q))l = mid;
else r = mid - 1;
}
priority_queue<PII> q2;
for(auto it : mp)
{
q2.push({it.second,it.first});
}
for(int i = 1; i <= k ; i ++)
{
vector<PII> v;
for(int j = 1 ; j <= l ; j ++)
{
auto x = q2.top();
q2.pop();
res[ansv[x.second].back()] = i ;
ansv[x.second].pop_back();
v.push_back({x.first-1,x.second});
}
for(auto it : v)q2.push({it});
}
for(int i = 1; i <= n ; i ++)cout <<res[i]<<' ';
cout <<endl;
}
}