C - Coffee Break

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <queue>
const int maxn=2e5+10;
using namespace std;
struct node {
    int val,indx,ans;
    friend bool operator < (node a,node b){  //规定优先队列的排列顺序（小->大）
        return a.val>b.val;
    }
};  //定义结构体
int n,m,d;
node a[maxn];
priority_queue<node>q;
bool cmp(node a,node b){
    return a.indx<b.indx;
}
bool cmp1(node a,node b){
    return a.val<b.val;
}
int main(){
    cin>>n>>m>>d;
    for (int i=1; i<=n; i++) {  //输入
        int x;cin>>x;
        a[i].val=x;
        a[i].indx=i;          //indx 为当前点的位置--为输出做准备
    }
    sort(a+1, a+1+n, cmp1);    //以时间val从小到大排序--依次入队列。
    a[1].ans=1;q.push(a[1]);   //首先将最小的入队列，第一天的第一杯咖啡
    int t=1;                 //t 为喝咖啡的天数
    for (int i=2; i<=n; i++) {  //遍历2-n个咖啡点
        if (a[i].val>q.top().val+d) { //如果当前咖啡时间和队首（最小的）咖啡时间相差大于d
            a[i].ans=q.top().ans;    //说明这一杯咖啡可以在队首元素这一天喝掉
            q.pop();               //队首出队列，当前a[i]会取而代之
        }
        else a[i].ans=++t;          // 如果<=d,说明当前咖啡时间不能在队列中任一天数喝，就需要新一天来喝
        q.push(a[i]);              //更新这一天喝的刚喝的咖啡，方便下一次比较
    }
    sort(a+1, a+1+n, cmp); //以位置从小到大排序--符合输出要求。
    cout<<t<<endl;
    for (int i=1; i<=n; i++) cout<<a[i].ans<<‘ ‘;
    cout<<endl;
}