Cod sursa(job #125747)

Utilizator tvladTataranu Vlad tvlad Data 20 ianuarie 2008 17:23:46
Problema Partitie Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.08 kb
#include <cstdio>
#include <algorithm>
#include <set>
#include <queue>
using namespace std;

const int N = 300000;

int n,d;
pair<int,int> a[N];
int m[N];

bool pair_cmp_second ( const pair<int,int> &a, const pair<int,int> &b ) { return a.second < b.second; }

int main() {
	freopen("partitie.in","rt",stdin);
	freopen("partitie.out","wt",stdout);
	scanf("%d %d",&n,&d);
	for (int i = 0; i < n; ++i) {
		scanf("%d",&a[i].first);
		a[i].second = i;
	}
	sort(a,a+n);
	// A walk in the park :)
	deque<int> deq;
	set<int> ok;
	int max = 1;
	m[0] = 1; deq.push_back(0);
	int ds = 1;
	for (int i = 1; i < n; ++i) {
		while (ds != 0 && a[deq.front()].first <= a[i].first-d) {
			ok.insert(m[deq.front()]);
			deq.pop_front();
			--ds;
		}
		if (ok.empty()) {
			++max;
			ok.insert(max);
		}
		m[i] = (*(ok.begin()));
		ok.erase(ok.find(m[i]));
		deq.push_back(i);
		++ds;
	}
	// Back home
	for (int i = 0; i < n; ++i) a[i].first = i;
	sort(a,a+n,pair_cmp_second);
	printf("%d\n",max);
	for (int i = 0; i < n; ++i) {
		printf("%d\n",m[a[i].first]);
	}
	return 0;
}