Cod sursa(job #703568)

Utilizator milijrCristian Militaru milijr Data 2 martie 2012 12:54:17
Problema Secventa Scor 10
Compilator cpp Status done
Runda pregatireoji_6 Marime 2.77 kb
#include <stdio.h>
#include <vector>
using namespace std;

#define MAXN 500005
#define MAXDIFNR 600020
#define MAXNR 30000
vector<int> poz[MAXDIFNR];
int n, k, nrInterv = 1, nrAuxInterv = 1, indice;
pair<int, int> intervale[MAXN];
vector<int> subinterv[MAXN];
pair<int, int> interv;
bool intervBun;
void dfs(int nod)
{
	if(subinterv[nod].empty() || subinterv[nod].front() > nrAuxInterv)
	{
		interv.first = intervale[nod].first + 1;
		interv.second = intervale[nod].second - 1;
		
		intervBun = true;
		nrInterv++;
		subinterv[nod].push_back(nrInterv);
		intervale[nrInterv] = pair<int, int> (interv.first - 1, indice);
		nrInterv++;
		subinterv[nod].push_back(nrInterv);
		intervale[nrInterv] = pair<int, int> (indice, interv.second + 1);
		
		return;
	}
	vector<int>::iterator it;
	for(it = subinterv[nod].begin(); it != subinterv[nod].end(); it++)
	{
		if(intervale[*it].second > indice)
		{
			if(intervale[*it].second - intervale[*it].first > k)
				dfs(*it);
			else
				return;
		}
	}
}
int nr[MAXN];
int main()
{
	FILE  *in = fopen("secventa.in",  "r");
	FILE *out = fopen("secventa.out", "w");
	
	fscanf(in, "%i %i", &n, &k);
	intervale[1] = pair<int, int>(0, n + 1);
	int i, aux, j;
	
	for(i = 1; i <= n; i++)
	{
		fscanf(in, "%i", &aux);
		poz[aux + MAXNR].push_back(i);
	}
	vector<int>::iterator it;
	pair<int, int> intervMax;
	bool gasit, stored = 0;
	int maxim, st, dr, piv;
	vector<int> used[2];
	used[0].push_back(0);
	used[0].push_back(n + 1);
	int i1, i2, n1, n2;
	for(i = 0; i < MAXDIFNR; i++)
	{
		if(!poz[i].empty())
		{
			nrAuxInterv = nrInterv;
			gasit = false;
			for(it = poz[i].begin(); it != poz[i].end(); it++)
			{
				//binary search:::
				st = 0;
				dr = used[stored].size() - 1;
				while(st < dr - 1)
				{
					piv = (st + dr) / 2;
					if(used[stored][piv] < *it)
						st = piv;
					else
						dr = piv;
				}
				if(maxim != i && used[stored][st + 1] - used[stored][st] > k)
				{
					intervMax.first = used[stored][st];
					maxim = i;
				}
				/*intervBun = false;
				indice = *it;
				dfs(1);
				if(intervBun == true && gasit == false)
				{
					gasit = true;
					intervMax.first = interv.first;
					intervMax.second = interv.second;
					maxim = i;
				}*/
			}
			used[1 - stored].clear();
			i1 = i2 = 0;
			n1 = used[stored].size();
			n2 = poz[i].size();
			//sorare prin interclasare:
			while(i1 != n1 || i2 != n2)
			{
				if((i1 == n1 || poz[i][i2] < used[stored][i1]) && i2 != n2)
				{
					used[1 - stored].push_back(poz[i][i2]);
					i2++;
				}
				else
				{
					used[1 - stored].push_back(used[stored][i1]);
					i1++;
				}
			}
			stored = 1 - stored;
		}
	}
	fprintf(out, "%i %i %i\n", intervMax.first + 1, intervMax.first + k, maxim - MAXNR);
}