Cod sursa(job #2209596)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 3 iunie 2018 20:49:49
Problema Nums Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.18 kb
#include<bits/stdc++.h>
using namespace std;
ifstream f("nums.in");
ofstream g("nums.out");
int n;
int aib[100002];
bool prs[100002];
struct qx
{
    int tip,pos;
    string x;
};
qx v[100002];
int cate=0;
struct str
{
    string x;
};
str v2[100002];
void add(int nod)
{
    for(;nod<=cate;nod+=(nod&(-nod)))
        aib[nod]++;
}
int compute(int nod)
{
    int sol=0;
    for(;nod;nod-=(nod&(-nod)))
        sol+=aib[nod];
    return sol;
}
bool cmp(str a, str b)
{
    if(a.x.size()!=b.x.size())
        return a.x.size()<b.x.size();
    for(int i=0;i<a.x.size();++i)
        if(a.x[i]!=b.x[i])
            return a.x[i]<b.x[i];
}
int cc(string a, string b)
{
    if(a.size()<b.size())
        return -1;
    if(a.size()>b.size())
        return 1;
    for(int i=0;i<a.size();++i)
    {
        if(a[i]>b[i])
            return 1;
        if(a[i]<b[i])
            return -1;
    }
    return 0;
}
int cb(string s)
{
    int b=1;
    int e=cate;
    while(b<=e)
    {
        int mid=(b+e)/2;
        int val=cc(v2[mid].x,s);
        if(val==0)
            return mid;
        if(val==-1)
            b=mid+1;
        else
            e=mid-1;
    }
}
int main()
{
    f>>n;
    for(int i=1;i<=n;++i)
    {
        f>>v[i].tip;
        if(v[i].tip==1)
        {
            f>>v[i].x;
            ++cate;
            v2[cate].x=v[i].x;
        }
        else
            f>>v[i].pos;
    }
    sort(v2+1,v2+cate+1,cmp);
    for(int i=1;i<=n;++i)
    {
        if(v[i].tip==1)
        {
            int pos=cb(v[i].x);
            if(!prs[pos])
                add(pos),prs[pos]=1;
        }
        else
        {
            int b=1;
            int e=cate;
            while(b<=e)
            {
                int mid=(b+e)/2;
                int va=compute(mid);
                int vb=compute(mid-1);
                if(va==v[i].pos && vb<v[i].pos)
                {
                    g<<v2[mid].x<<'\n';
                    break;
                }
                if(va>=v[i].pos)
                    e=mid-1;
                else
                    b=mid+1;
            }
        }
    }
    return 0;
}