Pagini recente » Cod sursa (job #2493660) | Cod sursa (job #429640) | Cod sursa (job #735380) | tema | Cod sursa (job #859029)
Cod sursa(job #859029)
#include <deque>
#include <fstream>
#include <iostream>
#include <tr1/unordered_map>
#include <map>
using namespace std;
using namespace std::tr1;
#define FORIT(it, v)for(typeof((v).begin()) it=(v).begin();it!=(v).end();++it)
const int MAX_N = (1 << 21) + 10;
typedef unordered_map<int, int>::iterator mit;
typedef unsigned long long uint64;
typedef unsigned int uint32;
ifstream fin("secv5.in");
ofstream fout("secv5.out");
int N, L, U;
uint32 sir[MAX_N];
int dq[MAX_N], dqFront, dqBack;
uint64 compute_subsets(uint32*, int, int);
void debug(map<int, int>&, deque<int>&);
int main() {
fin >> N >> L >> U;
for (int i = 1; i <= N; ++i) {
fin >> sir[i];
}
fout << compute_subsets(sir, N, U) - compute_subsets(sir, N, L - 1);
}
uint64 compute_subsets(uint32* v, int n, int limit) {
if (limit == 0) return 0;
uint64 result = 0;
//deque<int> dq;
dqFront = 1, dqBack = 0;
unordered_map<int, int> hash;
int numValues = 0;
for (int i = 1; i <= n; ++i) {
dq[++dqBack] = v[i]; //dq.push_back(v[i]);
int valBack = dq[dqBack]; //dq.back();
mit itBack = hash.find(valBack);
if (itBack == hash.end()) {
++numValues;
if (numValues > limit) {
int valFront = dq[dqFront]; //dq.front();
mit itFront = hash.find(valFront);
while (itFront->second > 1) {
--itFront->second;
++dqFront; //dq.pop_front();
if (/*dq.front()*/ dq[dqFront] != valFront) {
valFront = dq[dqFront]; //dq.front();
itFront = hash.find(valFront);
}
}
hash.erase(itFront);
++dqFront; //dq.pop_front();
--numValues;
}
hash.insert(make_pair(valBack, 1));
} else {
++itBack->second;
}
result += dqBack - dqFront + 1; //dq.size();
//debug(hash, dq);
}
return result;
}
void debug(map<int, int>& h, deque<int>& d) {
cerr << "ZE map: \n";
FORIT (it, h) {
cerr << it->first << ' ' << it->second << endl;
}
cerr << "ZE deq: \n";
FORIT (it, d) {
cerr << *it << endl;
}
cerr << "Added: " << d.size() << endl;
cerr << endl;
}