Cod sursa(job #1825707)

Utilizator liviu23Liviu Andrei liviu23 Data 9 decembrie 2016 16:23:35
Problema Secventa 5 Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.55 kb
#include <algorithm>
#include <fstream>
#include <vector>
#define f first
#define s second
#define DIM 1000105
#define M 205019
#define uint unsigned
using namespace std;

uint n,l,u,v[DIM];
vector<pair<uint,uint> > h[M+5];

bool findNum(uint x,int add) {
    uint i=x%M,st=0,dr=h[i].size(),mij;
    while(st<dr) {
        mij=(st+dr)/2;
        if(x<h[i][mij].f)
            st=mij+1;
        else if(x>h[i][mij].f)
            dr=mij-1;
        else {
            h[i][mij].s+=add;
            if(add==-1&&h[i][mij].s==0)
                return false;
            if(add==1&&h[i][mij].s==1)
                return false;
            return true;
        }
    }
    return false;
}

uint findNextDist(uint i) {
    for(;i<=n;i++)
        if(!findNum(v[i],-1))
            return i+1;
    return n;
}

long long countSeq(uint len) {
    uint j=1,s=0;
    long long ans=0;
    for(uint i=1;i<=n;i++) {
        if(!findNum(v[i],1)) {
            if(s==len)
                j=findNextDist(j);
            else s++;
        }
        ans+=i-j+1;
    }
    return ans;
}

int main()
{
    ios_base::sync_with_stdio(false);
    ifstream fin("secv5.in");
    ofstream fout("secv5.out");
    fin>>n>>l>>u;
    for(uint i=1;i<=n;i++) {
        fin>>v[i];
        h[v[i]%M].push_back({v[i],0});
    }
    for(int i=0;i<M;i++)
        sort(h[i].begin(),h[i].end());
    long long a1=countSeq(u);
    for(int i=0;i<M;i++)
    for(uint j=0;j<h[i].size();j++)
        h[i][j].s=0;
    fout<<a1-countSeq(l-1);
    return 0;
}