Cod sursa(job #1280432)

Utilizator dragangabrielDragan Andrei Gabriel dragangabriel Data 1 decembrie 2014 22:31:07
Problema Secventa 5 Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.65 kb
#include<fstream>
#include<algorithm>
#include<unordered_map>
using namespace std;

unordered_map<unsigned int,int>s;
int n,x,y,k,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);
    g.close();
    return 0;
}