Cod sursa(job #1835447)

Utilizator ade_tomiEnache Adelina ade_tomi Data 26 decembrie 2016 21:22:04
Problema Nums Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.81 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <unordered_map>
#include <cstring>

using namespace std;
const int NMAX =  100004;
int n, query[NMAX], aib[NMAX];
string s[NMAX], s2[NMAX];
unordered_map <string, int> poz;
unordered_map <string, char> used;
bool cmp (string a, string b)
{

    if (a.size() != b.size())
        return a.size() < b.size();
    return a < b;
}
int lsb (int x)
{

    return x&(-x);
}
void update(int poz, int val)
{

    while (poz <= n)
    {
        aib[poz] += val;
        poz += lsb(poz);
    }
}
string sir;
int main ()
{
    ifstream cin ("nums.in");
    ofstream cout ("nums.out");
    cin >> n;
    int nr = 0;
    for (int i = 1; i <= n; i++)
    {
        int x;
        cin >> x;
        if (x == 1)
        {

            cin >> sir;
            if (used[sir] == 0)
            {
                nr++;
                s[nr] = sir;
                s2[i] = s[nr];
                used[sir] = 1;
            }

        }
        else
        {

            cin >> query[i];
        }


    }
    sort (s + 1, s + 1 + nr, cmp);

    for (int i = 1; i <= nr; i++)
    {
        if (poz[s[i]] == 0)
            poz[s[i]] = i;


    }
    for (int i = 1; i <= n; i++)
    {

        if (query[i] == 0)
        {

            update(poz[s2[i]], 1);
           // cout << poz[s2[i]] << " " ;
        }
        else
        {
            int ans = 0;

            for (int step = 1 << 16; step > 0; step /= 2)
            {

                if (ans + step < n && aib[ans + step] < query[i])
                {

                    ans += step;
                    query[i] -= aib[ans];

                }
            }
            cout << s[ans + 1] << "\n";
        }
    }
    return 0;
}