Cod sursa(job #1401970)

Utilizator retrogradLucian Bicsi retrograd Data 26 martie 2015 11:25:41
Problema Partitie Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.83 kb
#include<fstream>
#include<algorithm>
#include<vector>

using namespace std;
typedef int var;

ifstream fin("partitie.in");
ofstream fout("partitie.out");

#define MAXN 500001

vector<pair<var, var> >V;
#define mp make_pair
var COLOR[MAXN];
var LAST[MAXN];

int main() {

    var n, d, val;
    fin>>n>>d;

    for(var i=1; i<=n; i++) {
        fin>>val;
        V.push_back(mp(val, i));
        LAST[i] = -2000000000;
    }
    sort(V.begin(), V.end());

    var maxc = -1, cur_color;
    for(auto v : V) {
        var val = v.first;
        var col;
        for(col = 1; val < LAST[col] + d; col++);
        COLOR[v.second] = col;
        LAST[col] = val;
        maxc = max(maxc, col);
    }

    fout<<maxc<<'\n';
    for(var i=1; i<=n; i++) {
        fout<<COLOR[i]<<'\n';
    }

    return 0;
}