Cod sursa(job #23464)

Utilizator wefgefAndrei Grigorean wefgef Data 28 februarie 2007 19:53:44
Problema Secventa 5 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.96 kb
#include <cstdio>
#include <string.h>
#include <algorithm>

using namespace std;

#define Nmax (1 << 20)

#define x first
#define y second

pair<unsigned int, int> u[Nmax];

int v[Nmax];
char frec[Nmax];
int n, m, a, b;

void readdata()
{
	freopen("secv5.in", "r", stdin);
	freopen("secv5.out", "w", stdout);
	
	scanf("%d %d %d", &n, &a, &b);
	for (int i = 0; i < n; ++i)
	{
		scanf("%u", &u[i].x);
		u[i].y = i;
	}
}

long long eval(int k)
{
	int i, j = 0, cnt = 0;
	long long rez = 0;
	
	if (k == a-1) memset(frec, 0, sizeof(frec));
	for (i = 0; i < n; ++i)
	{
		cnt += ++frec[v[i]] == 1;
		
		while (cnt > k)
			if (--frec[v[j++]] == 0) --cnt;

		rez += i-j+1;
	}
	return rez;
}

void solve()
{
	int i;
	
	sort(u, u+n);
	
	v[ u[0].y ] = 0;
	for (i = 1; i < n; ++i)
	{
		if (u[i].x > u[i-1].x) ++m;
		v[ u[i].y ] = m;
	}
	
	printf("%lld\n", eval(b) - eval(a-1));
}

int main()
{
	readdata();
	solve();
	return 0;
}