Cod sursa(job #1825728)

Utilizator liviu23Liviu Andrei liviu23 Data 9 decembrie 2016 16:42:23
Problema Secventa 5 Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.53 kb
#include <algorithm>
#include <fstream>
#include <vector>
#define f first
#define s second
#define DIM (1<<20)+5
#define M 205019
#define ll long long
using namespace std;

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

bool findNum(ll x,int add) {
    int i=x%M,st=0,dr=h[i].size()-1,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;
}

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

ll countSeq(int len) {
    ll s=0,ans=0;
    int j=1;
    for(int 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(int 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(int j=0;j<h[i].size();j++)
        h[i][j].s=0;
    fout<<a1-countSeq(l-1);
    return 0;
}