Cod sursa(job #2267725)

Utilizator giotoPopescu Ioan gioto Data 23 octombrie 2018 22:20:26
Problema Nums Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.74 kb
#include <bits/stdc++.h>
using namespace std;

const int MOD1 = 100007;
const int MOD2 = 100021;

int n, NR;
int a[100005], k[100005], pos[100005], rev[100005];
bool w[100005], ok[100005];
string b;
pair <string, int> s[100005];

inline bool cmp(pair <string, int> x, pair <string, int> y){
    if((x.first).size() != (y.first).size()) return (x.first).size() < (y.first).size();
    return x.first < y.first;
}

int Arb[400005];

void u(int st, int dr, int nod, int p){
    if(st == dr){++Arb[nod]; return ;}

    int mij = (st + dr) / 2;
    if(p <= mij) u(st, mij, nod * 2, p);
    else u(mij + 1, dr, nod * 2 + 1, p);
    Arb[nod] = Arb[nod * 2] + Arb[nod * 2 + 1];
}
int q(int st, int dr, int nod, int val){
    if(st == dr) return st;

    int mij = (st + dr) / 2;
    if(Arb[nod * 2] >= val) return q(st, mij, nod * 2, val);
    else return q(mij + 1, dr, nod * 2 + 1, val - Arb[nod * 2]);
}
int main()
{
    ifstream fin("nums.in");
    ofstream fout("nums.out");

    fin >> n;
    int NR = 0;
    for(int i = 1; i <= n ; ++i){
        fin >> a[i];
        if(a[i] == 1){
            fin >> b;
            ++NR;
            s[NR] = {b, NR};
            a[i] = NR;
        }
        else fin >> k[i];
    }

    sort(s + 1, s + NR + 1, cmp);
    for(int i = 1; i <= NR ; ++i){
        if(s[i].first != s[i - 1].first)
        pos[s[i].second] = i, rev[pos[s[i].second]] = i, ok[s[i].second] = 1;
    }

    for(int i = 1; i <= n ; ++i){
        if(a[i] >= 1){
            if(ok[a[i]])
            u(1, NR, 1, pos[a[i]]), w[pos[a[i]]] = 1;
        }
        else if(a[i] == 0){
            int pos = q(1, NR, 1, k[i]);
            fout << s[rev[pos]].first << "\n";
        }
    }



    return 0;
}