关于队里面最菜的在博客打卡第二十六天这件事

题目链接：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;
    }

}