Cod sursa(job #2112287)

Utilizator Mihai145Oprea Mihai Adrian Mihai145 Data 23 ianuarie 2018 12:07:30
Problema Partitie Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.66 kb
#include <fstream>
#include <algorithm>
#define NMAX 300002
using namespace std;
ifstream fin("partitie.in");
ofstream fout("partitie.out");
int n, d, nrp, v[NMAX], poz[NMAX];
inline bool cmp(const int a, const int b)
{
    return v[a]<v[b];
}
int main()
{
    int i, j;
    fin>>n>>d;
    for(i=1; i<=n; i++)
    {
        fin>>v[i];
        poz[i]=i;
    }
    sort(poz+1,poz+n+1,cmp);
    for(i=1; i<=n; i++)
    {
        for(j=i+1; j<=n && v[poz[j]]-v[poz[i]]<d ;)
            j++;
        nrp=max(nrp,j-i);
    }
    fout<<nrp<<'\n';
    for(i=1; i<=n; i++) v[poz[i]]=i%nrp+1;
    for(i=1; i<=n; i++) fout<<v[i]<<'\n';
    return 0;
}