Cod sursa(job #1798539)

Utilizator Bodo171Bogdan Pop Bodo171 Data 5 noiembrie 2016 11:53:31
Problema Nums Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.59 kb
#include <iostream>
#include<fstream>
#include<algorithm>
using namespace std;
const int nmax=100005;
struct numar
{
    string str;
    int ind;
}v[nmax];
string s[nmax];
int aib[nmax],poz[nmax],stat[nmax],first[nmax];
int i,n,m,p,u,idx,k,sum,key,tip,j;
bool ap[nmax];
bool comp(numar unu,numar doi)
{
     if(unu.str.size()==doi.str.size()) return unu.str<doi.str;
    return (unu.str.size()<doi.str.size());
}
inline int lbit(int x)
{
    return ((x^(x-1))&x);
}
void update(int poz)
{
    for(idx=poz;idx<=key;idx+=lbit(idx))
      aib[idx]++;
}
int compute(int poz)
{
    sum=0;
    for(idx=poz;idx>0;idx-=lbit(idx))
        sum+=aib[idx];
    return sum;
}
int cb(int nr)
{
    p=1;u=key;
    while(u-p>1)
    {
        m=(p+u)/2;
        if(compute(m)<nr) p=m;
        else u=m;
    }
    return u;
}
void norm()
{
    for(i=1;i<=k;i++)
    {
        if(v[i].str!=v[i-1].str)
            {key++;first[key]=i;}
        poz[v[i].ind]=key;
    }
}
int main()
{
    ifstream f("nums.in");
    ofstream g("nums.out");
    f>>n;
    for(i=1;i<=n;i++)
    {
        f>>tip;
        if(tip==1)
        {
            f>>s[i];
            k++;
            v[k].str=s[i];
            v[k].ind=i;
        }else
        {
            f>>stat[i];
        }
    }
    sort(v+1,v+k+1,comp);
    norm();
    for(i=1;i<=n;i++)
    {
        if(stat[i]!=0)
        {
            g<<v[first[cb(stat[i])]].str<<'\n';
        }
        else
        {
            if(!ap[poz[i]])update(poz[i]);
            ap[poz[i]]=1;
        }
    }
    return 0;
}