Cod sursa(job #1281248)

Utilizator dragangabrielDragan Andrei Gabriel dragangabriel Data 2 decembrie 2014 22:46:30
Problema Secventa 5 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.66 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;
}