Cod sursa(job #1146958)

Utilizator misinozzz zzz misino Data 19 martie 2014 14:17:40
Problema Nums Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.72 kb
#include<fstream>
#include<cstring>
#include<algorithm>
#define N 100100
using namespace std;
ifstream f("nums.in");
ofstream g("nums.out");
int i,n,mini,pp,aint[4*N],nr,p[N];
struct numere{int poz;string nr;}v[N];
struct operatii{int t,k;}op[N];
inline bool cmp(numere a,numere b){
    if(a.nr.size()==b.nr.size())
    return a.nr<b.nr;
    return a.nr.size()<b.nr.size();
}
inline void update(int nod,int li,int ls,int poz){
    if(li==ls)
    {
        ++aint[nod];
        return ;
    }
    int mij=(li+ls)>>1;
    if(poz<=mij)
    update(nod*2,li,mij,poz);
    else
    update(nod*2+1,mij+1,ls,poz);
    aint[nod]=aint[nod*2]+aint[nod*2+1];
}
inline void query(int nod,int li,int ls,int care){
    if(li==ls)
    {
        g<<v[li].nr<<'\n';
        return ;
    }
    int mij=(li+ls)>>1;
    if(aint[nod*2]>=care)
    query(nod*2,li,mij,care);
    else
    query(nod*2+1,mij+1,ls,care-aint[nod*2]);
}
int main()
{
    f>>n;
    for(i=1;i<=n;++i)
    {
        f>>op[i].t;
        if(op[i].t)
        {
            f>>v[++nr].nr;
            v[nr].poz=nr;
        }
        else
        {
            f>>op[i].k;
        }
    }
    sort(v+1,v+nr+1,cmp);
    for(i=1;i<=nr;++i)
    {
        mini=v[i].poz;
        pp=i;
        if(i<nr&&v[i].nr==v[i+1].nr)
        {
            ++i;
            while(i<nr&&v[i].nr==v[i+1].nr)
            mini=min(mini,v[i].poz),++i;
            --i;
        }
        p[mini]=pp;
    }
    pp=0;
    for(i=1;i<=n;++i)
    {
        if(op[i].t)
        {
            if(!p[++pp])
            continue;
            update(1,1,nr,p[pp]);
        }
        else
        {
            query(1,1,nr,op[i].k);
        }
    }
    return 0;
}