Cod sursa(job #444946)

Utilizator mathboyDragos-Alin Rotaru mathboy Data 22 aprilie 2010 10:43:00
Problema Nums Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.69 kb
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <iostream>


using namespace std;
using namespace __gnu_cxx;


#define nmax 100005
#define VIT vector <int> :: iterator
#define pb push_back
#define f first
#define s second
#define mp make_pair

string S [nmax];
struct cmp {
	bool operator()(const int &a, const int &b) {
		if (S [a].size () == S [b].size ())
			return S [a].compare (S [b]) < 0;
		return S [a].size () < S [b].size ();
	}
};
int aib [nmax], idx, order [nmax], i;
string str;
int opp [nmax][3];
inline void update (int x) {
	for (; x <= idx; x += x & - x) aib [x]++;
}
inline int query (int x) {
	int sum = 0;
	for (; x >= 1; x -= x & -x) sum += aib [x];
	return sum;
}
inline int cbin (int x)
{	
	int st, dr, mid, ans = 0, sol = 0;
	st = 1; dr = idx;
	while (st <= dr)
	{
		mid = (st + dr) / 2;
		ans = query (mid);
		if (ans < x)
			st = mid + 1;
		else if (ans > x)
			dr = mid - 1;
		else if (ans == x) {
			sol = mid;
			dr = mid - 1;
		}
	}
	return sol;
}
int main ()
{
	freopen ("nums.in", "r", stdin);
	freopen ("nums.out", "w", stdout);
	int m,op, qq;
	scanf ("%d\n", &m);
	for (i = 1; i <= m; i++)
	{
		scanf ("%d", &op);
		if (op == 1) 
		{	
			cin >> str;
			++idx;
			S [idx] = str;
			opp [i][0] = 1;
			//cout << S [idx] << " ";
		}
		else if (op == 0)
		{
			cin >> qq;
			opp [i][1] = qq;
		}
	}
	for (i = 1; i <= idx; i++) order [i] = i;
	random_shuffle (order + 1, order + idx + 1);
	sort (order + 1, order + idx + 1, cmp ());
	//for (i = 1; i <= idx; i++) cout << order [i] << endl;
	int a, j = 0;
	for (i = 1; i <= m; i++)
	{
		if (opp [i][0])
			update (order [++j]);
		else 
		{
			a = cbin (opp [i][1]);
			cout <<S [a] << "\n";
		}
	}
	return 0;
}