Cod sursa(job #1539176)

Utilizator xtreme77Patrick Sava xtreme77 Data 30 noiembrie 2015 13:49:52
Problema Nums Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.26 kb
/**
 * Code by Patrick Sava
 * "Spiru Haret" National College of Bucharest
 **/

# include "fstream"
# include "cstring"
# include "vector"
# include "queue"
# include "bitset"
# include "algorithm"
# include "map"
# include "set"
# include "unordered_map"
# include "deque"
# include "string"
# include "iomanip"
# include "cmath"
# include "stack"
# include "cassert"

const char IN [ ] =  "nums.in" ;
const char OUT [ ] = "nums.out" ;

using namespace std ;

# define pb push_back
# define mp make_pair
# define FORN( a , b , c ) for ( register int a = b ; a <= c ; ++ a )
# define FORNBACK( a , b , c ) for ( register int a = b ; a >= c ; -- a )

const int MAX = 1e5 + 14 ;

ifstream cin ( IN ) ;
ofstream cout ( OUT ) ;

unordered_map < int , int > H ;
int v [ MAX ] ;
int cate ;
vector < pair < int , int > > Q ;
int inv [ MAX ] ;

int arb [ MAX * 4 ] ;

inline void baga ( int nod , int st , int dr , int pos )
{
    if ( st == dr ) {
        arb [ nod ] ++ ;
        return ;
    }
    int mij = ( st + dr ) >> 1 ;
    if ( pos <= mij )
        baga ( nod << 1 , st , mij , pos ) ;
    else baga ( nod << 1 | 1 , mij + 1 , dr , pos ) ;
    arb [ nod ] = arb [ nod << 1 ] + arb [ nod << 1 | 1 ] ;
}

int caut ( int nod , int st , int dr , int k )
{
    if ( st == dr )
        return st ;
    int mij = ( st + dr ) >> 1 ;
    if ( arb [ nod << 1 ] >= k )
        return caut ( nod << 1 , st , mij , k ) ;
    else return caut ( nod << 1 | 1 , mij + 1 , dr , k - arb [ nod << 1 ] ) ;
}


int main()
{
    int n ;
    cin >> n ;
    while ( n -- )
    {
        int a , b ;
        cin >> a >> b ;
        Q.push_back( { a , b } )  ;
        if ( a == 1 ){
            if ( H [ b ] == 0 ) {
                H [ b ] = 1 ;
                ++ cate ;
                v [ cate ] = b ;
            }
        }
    }
    int e = 0 ;
    sort ( v + 1 , v + cate + 1 ) ;
    FORN ( i , 1 , cate ){
        H [ v [ i ] ] = ++ e ;
        inv [ e ] = v [ i ] ;
    }
    for ( auto x : Q ){
        if ( x.first == 1 ){
            int pos = H [ x.second ] ;
            baga ( 1 , 1 , e , pos ) ;
        }
        else {
            cout << inv [ caut ( 1 , 1 , e , x.second ) ] << '\n' ;
        }
    }
    return 0;
}