Cod sursa(job #1835435)

Utilizator ade_tomiEnache Adelina ade_tomi Data 26 decembrie 2016 21:04:07
Problema Nums Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.51 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;
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);
    }
}
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)
        {
            nr++;
            cin >> s[nr];
            s2[i] = s[nr];

        }
        else
        {

            cin >> query[i];
        }


    }

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

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

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

        if (query[i] == 0)
        {

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

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

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

                    ans += step;

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