Cod sursa(job #10946)

Utilizator vlad_DVlad Dumitriu vlad_D Data 30 ianuarie 2007 02:05:04
Problema Secventa 5 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
/*
by vlad D.
   :)
*/
#include <cstdio>
#include <set>
using namespace std;
unsigned int ret;
unsigned int N, A, B;
unsigned int sol[1048588], unde[1048590];
set<unsigned int> Q;
unsigned int i, j;
int main() {
	freopen("secv5.in", "r", stdin);
	freopen("secv5.out", "w", stdout);
	scanf("%u %u %u", &N, &B, &A);
	for (i=0; i<N; i++) {
		int X;
		scanf("%u", &X);
		if (Q.find(X) == Q.end()) {Q.insert(X); sol[i+1] = sol[i] + 1;} 
		else sol[i+1] = sol[i];
		}
	Q.clear();
	//for (i=1; i<=N; i++) printf("%lld ", sol[i]);
	B--;
	unsigned int  lw = 0;
	for (i=N; i>=1; i--) unde[sol[i]] = i;
	for (i=1; i<=N; i++) {
		//printf("ll : d %d\n", i, lw);
		if (sol[i] <= A) ret+=i;
		else ret+= i - unde[sol[i]-A+1]+1;  
		//ret+= i - lw;
		}
	//printf("ret: %lld\n", ret);
	if (B > 0) {
	lw = 0;
	for (i=1; i<=N; i++) {
		if (sol[i] <= B) ret-=i;
		else ret-=i - unde[sol[i]-B+1]+1;
	}}
	printf("%u\n", ret);
	return 0;
}