DTRawr - Online C++0x Compiler & Debugging Tool

#include <bits/stdc++.h>
using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, d;
    cin >> n >> d;
    vector<int> v(n);
    unordered_map<int, int> freq;
    for (int i = 0; i < n; i++)
    {
        cin >> v[i];
        freq[v[i]]++;
    }

    priority_queue<pair<int, int>> pq;
    for (auto it : freq)
        pq.push({it.second, it.first});

    queue<pair<int, int>> cooldown;
    vector<int> ans;

    while (!pq.empty())
    {
        pair<int, int> top = pq.top();
        pq.pop();

        int count = top.first;
        int num = top.second;

        ans.push_back(num);
        count--;

        cooldown.push({count, num});

        if (cooldown.size() >= d)
        {
            pair<int, int> front = cooldown.front();
            cooldown.pop();
            if (front.first > 0)
                pq.push(front);
        }
    }

    if (ans.size() == n)
    {
        cout << "Yes" << endl;
        for (int x : ans)
            cout << x << " ";
        cout << endl;
    }
    else
    {
        cout << "No" << endl;
    }

    return 0;
}