Cod sursa(job #432382)

Utilizator tamas_iuliaTamas Iulia tamas_iulia Data 2 aprilie 2010 12:16:50
Problema Secventa 5 Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.97 kb
#include <stdio.h>
#include <vector>
#define Nmax 1048600
#define pb push_back
#define mp make_pair
#define x first
#define nr second
#define UL unsigned long
#define Mod 666013UL
#define LL long long
using namespace std;

vector< pair<UL,UL> > G[Mod];
vector< pair<UL,UL> > H[Mod];

UL v[Nmax];
int n,P,U,i,j; 
UL x,nr,last_nr,ok;
LL sumaU,sumaP;

char s[256];

inline int Find(UL x){
	UL list=x%Mod;
	vector< pair<UL,UL> >:: iterator it;
	
	for(it=G[list].begin(); it!=G[list].end(); ++it)
		if(it->x == x){
			last_nr=it->nr;
			return 1;
		}
	return -1;
}

inline vector< pair<UL,UL> >::iterator find(UL x){
	UL list=x%Mod;
	vector< pair<UL,UL> >:: iterator it;
	for(it=H[list].begin(); it!=H[list].end(); ++it)
		if(it->x==x) return it;
	return H[list].end();
}

inline void add(UL x){
	vector< pair<UL,UL> >:: iterator it=find(x);	
	if(it==H[x%Mod].end()) 
		H[x%Mod].pb(mp(x,1));
	else it->nr++;
}
inline void sterg(UL x){
	vector< pair<UL,UL> >:: iterator it=find(x);	
	if(it->nr >1) it->nr--;
	else H[x%Mod].erase(it), ok=1;
}

void work(int U, LL& sumaU){
	int i,Lu=1, nr=0;
	
	for(i=1;i<=n;++i){
		if( find(v[i]) == H[v[i]%Mod].end() ){
			add(v[i]); nr++;
			
			if( nr > U ){
				ok=0; // tre sa sterg pana cand ajung sa se stearga de tot unul dintre nr
				sterg(v[Lu]); // sterg 
				for(Lu++; !ok ; ){ 
					sterg(v[Lu]);
					Lu++;
				}
				nr--;
			}
		}
		else add(v[i]);
		
		sumaU=(LL) sumaU-Lu+i+1;
	}
	for(i=Lu; i<=n; ++i) H[v[i]%Mod].clear();

}	


int main(){
	freopen("secv5.in","r",stdin);
	freopen("secv5.out","w",stdout);
	scanf("%d%d%d\n",&n,&P,&U);
	
	for(i=1;i<=n;++i){
		fgets(s,255,stdin); x=0;
		for(j=0; s[j]>='0' && s[j]<='9'; ++j)
			x=x*10+s[j]-'0';
		
		if(Find(x) == -1){
			G[x%Mod].pb(mp(x,++nr));
			v[i]=nr;
		}
		else v[i]=last_nr;
		
	}
	
	sumaU=sumaP=0;
	work(U,sumaU);
	work(P-1,sumaP);
	
	printf("%lld\n",sumaU-sumaP);
	fclose(stdin); fclose(stdout);
	return 0;
}