Cod sursa(job #1994297)

Utilizator mouse_wirelessMouse Wireless mouse_wireless Data 24 iunie 2017 16:15:14
Problema Secventa 5 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.18 kb
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cassert>
#include <map>
#include <set>
#include <climits>
#include <cmath>
#include <list>
#include <functional>
#include <unordered_set>
#include <unordered_map>
using namespace std;
typedef long long LL;
typedef unsigned int uint;

#ifdef INFOARENA
#define ProblemName "secv5"
#endif

#define MCONCAT(A, B) A B
#ifdef ProblemName
#define InFile MCONCAT(ProblemName, ".in")
#define OuFile MCONCAT(ProblemName, ".out")
#else
#define InFile "fis.in"
#define OuFile "fis.out"
#endif

#define MAXN 1058576
uint v[MAXN];
int N, L, U;

LL ans(int k) {
	unordered_map<uint, int> S;
	int diff = 0;
	int l = 0;
	LL rez = 0;
	for (int r = 0; r < N; ++r) {
		if ((++S[v[r]]) == 1)
			++diff;
		while (diff > k) {
			if ((--S[v[l]]) == 0) {
				--diff;
				S.erase(v[l]);
			}
			++l;
		}
		rez += r - l + 1;
	}
	return rez;
}

int main() {
	assert(freopen(InFile, "r", stdin));
	assert(freopen(OuFile, "w", stdout));
	scanf("%d%d%d", &N, &L, &U);
	for (int i = 0; i < N; ++i)
		scanf("%u", &v[i]);
	printf("%lld\n", ans(U) - ans(L - 1));
	return 0;
}