Cod sursa(job #1135831)

Utilizator alexalghisiAlghisi Alessandro Paolo alexalghisi Data 8 martie 2014 14:21:29
Problema Nums Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.69 kb
#include <iostream>
#include <string>
#include <vector>
#include <fstream>
#include <algorithm>
#include <map>
#include <cstring>

#define DN 100003
using namespace std;


ifstream f("nums.in");
ofstream g("nums.out");

struct operatii{

    int op,k;
    string s;
};
struct st{
    string nr;
    int ind;
} v[DN];

int sz,arb[4*DN],n,poz,k,S;
int p[DN];
map < string , int > mp;
operatii M[DN];


bool mic(st a, st b) /// a<b ?
{
    if(a.nr.size() == b.nr.size()){

        return a.nr < b.nr;
    }
    return a.nr.size() < b.nr.size();
}

void update(int nod,int left,int right){

    if(left == right){

        ++arb[nod];
        return;
    }
    int mij = ( left + right)>>1;
    if( poz <= mij) update( 2*nod , left , mij);
        else update(2*nod+1,mij+1,right);

    arb[nod] = arb[2*nod] + arb[2*nod+1];
}

void fnd(int nod,int left,int right){

    if(left == right){

        g<<v[left].nr<<"\n";
        return;
    }
    int mij = ( left + right)>>1;
    if( S + arb[2*nod] >= k && arb[2*nod] ){

        fnd( 2*nod , left , mij);
    }
    else{
        S+=arb[2*nod];
        fnd(2*nod+1,mij+1,right);
    }
    arb[nod] = arb[2*nod] + arb[2*nod+1];
}

int main()
{
    int m;
    f>>m;
    for(int i=1;i<=m;++i){

        f>>M[i].op;

        if(M[i].op){
            f>>v[++sz].nr;
            v[sz].ind = sz;

            M[i].s = v[sz].nr;
        }
        else
            f>>M[i].k;
    }

    sort(v+1,v+sz+1,mic);
    for(int i=1;i<=sz;++i)
        p[ v[i].ind ] = i;

    n = sz;

    for(int i=1,c=0;i<=m;++i){

        if(M[i].op){

            poz = p[ ++c];
            update(1,1,n);
        }else{

            k = M[i].k;
            S=0;
            fnd(1,1,n);
        }
    }
    return 0;
}