Cod sursa(job #1281270)

Utilizator dragangabrielDragan Andrei Gabriel dragangabriel Data 2 decembrie 2014 23:09:17
Problema Secventa 5 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.67 kb
#include<fstream>
#include<unordered_map>
using namespace std;
 
unordered_map<unsigned int,int>s;
int n,x,y,k;
unsigned long v[1100000];
 
long long solve(int x)
{
    long long rez=0;
    int j=1;
    for(int i=1;i<=n;i++)
    {
        s[v[i]]++;
        while(s.size()>x)
        {
            s[v[j]]--;
            if (!s[v[j]]) s.erase(v[j]);
            j++;
        }
        rez=rez+i-j+1;
    }
    s.clear();
    return rez;
}
 
int main()
{
    ifstream f("secv5.in");
    ofstream g("secv5.out");
    f>>n>>x>>y;
    for(int i=1;i<=n;i++)
        f>>v[i];
    g<<solve(y)-solve(x-1);
    f.close();
    g.close();
    return 0;
}