Cod sursa(job #1317564)

Utilizator Tester100Tester Tester100 Data 14 ianuarie 2015 23:22:16
Problema Nums Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.21 kb
#include <iostream>
#include <unordered_map>
#include <vector>
#include <set>
#include <string>
#include <algorithm>

using namespace std;
const int NMAX = 100002;
int op[NMAX], index[NMAX], ind[NMAX];
string s[NMAX];
unordered_map < string,int >M;
int v[NMAX];
struct cmp
{
    bool operator ()(const int i,const int j)
    {
        return s[i] < s[j];
    }
};
set < int,cmp >S[NMAX];
struct AIB{
    vector <int > aib;
    int n;
    AIB()
    {
        n = 0;
        aib.clear();
    }
    AIB(const int _n)
    {
        n = _n;
        aib.resize(n+1);
    }
    inline void Update(int pos)
    {
        for(;pos<=n;pos+=pos&-pos)
            aib[pos]++;
    }
    inline int Query(int pos)
    {
        int ret = 0;
        for(;pos>0;pos-=pos&-pos)
            ret += aib[pos];
        return ret;
    }
};
int main()
{
    freopen("nums.in","r",stdin);
    freopen("nums.out","w",stdout);
    cin.sync_with_stdio(false);
    int n;
    cin >> n;
    AIB A(NMAX);
    for(int i=1;i<=n;++i)
    {
        cin >> op[i];
        if(op[i]==1){
            cin >> s[i];
            int L = s[i].length();
            if(S[L].find(i)!=S[L].end())
                continue;
            S[L].insert(i);
        }
        else
            cin >> index[i];
    }
    int j = 0;
    for(int i=1;i<=100000;++i){
        for(auto x=S[i].begin();x!=S[i].end();++x){
                v[++j] = *x;
                ind[*x] = j;
            }
        S[i].clear();
    }
    for(int i=1;i<=n;++i)
        if(op[i]==1){
            cin >> s[i];
            int L = s[i].length();
            if(S[L].find(i)!=S[L].end())
                continue;
            A.Update(ind[i]);
        }
        else{
            int pos = index[i];
            int L = NMAX;
            int Left = 1;
            int Right = L-1,mid;
            while(Left<=Right)
            {
                mid = (Left+Right)/2;
                if(A.Query(mid) >= pos)
                {
                    L = mid;
                    Right = mid-1;
                }
                else
                    Left =  mid+1;
            }
            cout<<s[v[L]]<<"\n";
        }
    return 0;
}