Cod sursa(job #2454778)

Utilizator mihai50000Mihai-Cristian Popescu mihai50000 Data 9 septembrie 2019 20:37:12
Problema Secventa 5 Scor 90
Compilator cpp-32 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("secv5.in");
ofstream fout("secv5.out");

const int DIM = (1 << 20) + 7;

unordered_map <unsigned int, int> H1;
unordered_map <unsigned int, int> H2;

unsigned int v[DIM];

unsigned int n;

unsigned int l, r;

main()
{
	fin >> n >> l >> r;
	
	for(int i = 1; i <= n; i++)
		fin >> v[i];
	
	long long l1 = 1, l2 = 1;
	long long ans = 0;
	
	for(int i = 1; i <= n; i++)
	{
		H1[v[i]]++;
		H2[v[i]]++;
		
		while(H1.size() > r)
		{
			H1[v[l1]]--;
			
			if(H1[v[l1]] == 0)
				H1.erase(v[l1]);
			
			l1++;
		}
		
		while(H2.size() >= l)
		{
			H2[v[l2]]--;
			
			if(H2[v[l2]] == 0)
				H2.erase(v[l2]);
			
			l2++;
		}
		
		ans += i - l1 + 1;
		ans -= i - l2 + 1;
	}
	
	fout << ans;
}