Cod sursa(job #3133790)

Utilizator AnastasiaStefanescuAnastasia Stefanescu AnastasiaStefanescu Data 26 mai 2023 22:59:47
Problema Distincte Scor 15
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.61 kb
#include <fstream>
using namespace std;

ifstream fin("distincte.in");
ofstream fout("distincte.out");

int aint[400005];
int A[100001], poz_ult_aparitie[100001], n, m, k;

void build(int nod,int st,int dr){
    if(st>dr)
        return;
    if(st==dr){
        aint[nod]=A[st];
        return;
    }
    int mijl=(st+dr)/2;
    build(2*nod,st,mijl);
    build(2*nod+1,mijl+1,dr);
    aint[nod]=aint[2*nod] + aint[2*nod+1];
}

void update(int nod,int st,int dr,int poz,int val){
    if(st>dr || poz>dr || poz<st)
        return;
    if(st==dr)
    {
        aint[nod] = val;
        return;
    }
    
    int mijl=(st+dr)/2;
    if(poz <= mijl)
        update(2*nod,st,mijl,poz,val);
    else
        update(2*nod+1,mijl+1,dr,poz,val);
    aint[nod]= aint[2*nod] + aint[2*nod+1];
}

int query(int nod,int st,int dr,int qst,int qdr){
    if(st>dr || qst>dr || qdr<st)
        return 0;
    if(qst<=st && dr<=qdr)
        return aint[nod];
    else
    {
        int mijl=(st+dr)/2;
        return query(2*nod,st,mijl,qst,qdr) + query(2*nod+1,mijl+1,dr,qst,qdr);
    }
}

void init(){
    int i;
    for (i = 1; i<= k; i++)
        poz_ult_aparitie[i] = 0;
}

int main()
{
    int i,x,y, j;
    
    fin>>n>>k >> m;
    for(i = 1; i<= n; i++)
    {
        fin >> A[i];
        
    }
    
    for(i=1;i<=m;i++)
    {
        fin>>x>>y;
        init();
        for(j = x; j<= y; j++)
        {
            if(poz_ult_aparitie[A[j]] != 0)
                update(1,1,n, poz_ult_aparitie[A[j]], 0);
            update(1, 1, n, j, A[j]);
            poz_ult_aparitie[A[j]] = j;
        }
        fout<<query(1,1,n,x,y)<<"\n";
    }
    
    return 0;
}