Cod sursa(job #2970523)

Utilizator bem.andreiIceman bem.andrei Data 25 ianuarie 2023 14:00:24
Problema Partitie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.05 kb
#include <bits/stdc++.h>

using namespace std;
ifstream r("partitie.in");
ofstream w("partitie.out");

const int nmax=300005;
pair<int,int> v[nmax];
int ans[nmax];
multiset<pair<int,int> >s;
int main()
{
    int n,d;
    r>>n>>d;
    for(int i=1; i<=n; i++)
    {
        r>>v[i].first;
        v[i].second=i;
    }
    sort(v+1,v+n+1);
    s.insert(v[1]);
    ans[v[1].second]=1;
    int nr=1;
    for(int i=2; i<=n; i++)
    {
        pair<int,int> x=*s.begin();
        pair<int,int> y=*s.end();
        if(y.first-v[i].first>=d)
        {
            s.erase(s.end());
            s.insert(v[i]);
            ans[v[i].second]=ans[y.second];
        }
        else if(v[i].first-x.first>=d)
        {
            s.erase(s.begin());
            s.insert(v[i]);
            ans[v[i].second]=ans[x.second];
        }
        else
        {
            ans[v[i].second]=++nr;
            s.insert(v[i]);
        }
    }
    w<<s.size()<<"\n";
    for(int i=1; i<=n; i++)
    {
        w<<ans[i]<<"\n";
    }
    return 0;
}