Cod sursa(job #2783813)

Utilizator MateiAruxandeiMateiStefan MateiAruxandei Data 15 octombrie 2021 09:26:43
Problema Nums Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace __gnu_pbds;
using namespace std;

ifstream fin("nums.in");
ofstream fout("nums.out");

class cmp
{
public:
    bool operator()(string a, string b)
    {
        if(a.size() == b.size())
            return a < b;
        return a.size() < b.size();
    }
};
tree < string,  null_type,  cmp,  rb_tree_tag,  tree_order_statistics_node_update > ARB;

int main()
{
    int n;
    fin >> n;

    for(int i = 1; i <= n; ++i)
    {
        int t;
        fin >> t;
        if(t == 1)
        {
            string val;
            fin >> val;
            ARB.insert(move(val));
        }
        else {
            int val;
            fin >> val;
            string rez = *ARB.find_by_order(val - 1);
            fout << rez << '\n';
        }
    }
    return 0;
}