POJ 1442 Black Box（treap练习）

题目链接：http://poj.org/problem?id=1442

题意：给出一个数列，两种操作：（1）加入一个数字，并将其插入保持数列不下降；（2）询问第i大的数字。

思路：节点记录左侧节点个数、右侧节点个数和本身的个数（因为有一样的数字）。这个题实际用STL的优先队列写只要几行代码，我把它贴在后面了。

struct node

{

    int L,R,key,pri,cntL,cntR,cnt;

};



class treap

{

public:

    node p[N];

    int size,root;



    void init()

    {

        srand(time(0));

        size=root=-1;

    }



    void rotL(int &x)

    {

        int y=p[x].R;

        p[x].R=p[y].L;

        p[x].cntR=p[y].cntL;

        p[y].L=x;

        p[y].cntL=p[x].cntL+p[x].cntR+p[x].cnt;

        x=y;

    }



    void rotR(int &x)

    {

        int y=p[x].L;

        p[x].L=p[y].R;

        p[x].cntL=p[y].cntR;

        p[y].R=x;

        p[y].cntR=p[x].cntL+p[x].cntR+p[x].cnt;

        x=y;

    }



    void insert(int &k,int key)

    {

        if(k==-1)

        {

            k=++size;

            p[k].L=p[k].R=-1;

            p[k].cnt=1;

            p[k].cntL=p[k].cntR=0;

            p[k].key=key;

            p[k].pri=rand();

        }

        else if(key<p[k].key)

        {

            p[k].cntL++;

            insert(p[k].L,key);

            if(p[p[k].L].pri>p[k].pri) rotR(k);

        }

        else if(key==p[k].key)

        {

            p[k].cnt++;

        }

        else

        {

            p[k].cntR++;

            insert(p[k].R,key);

            if(p[p[k].R].pri>p[k].pri) rotL(k);

        }

    }





    int find(int k,int cnt)

    {

        if(cnt<=p[k].cntL)  return find(p[k].L,cnt);

        cnt-=p[k].cntL;

        if(cnt<=p[k].cnt) return p[k].key;

        cnt-=p[k].cnt;

        return find(p[k].R,cnt);

    }

};





treap a;

int n,m,b[N];



int main()

{

    Rush(n)

    {

        RD(m); a.init();

        int i;

        FOR1(i,n) RD(b[i]);

        i=1;

        int x,t=0;

        while(m--)

        {

            RD(x); t++;

            while(i<=x) a.insert(a.root,b[i++]);

            PR(a.find(a.root,t));

        }

    }

    return 0;

}

　

#include <stdio.h>

#include <queue>

using namespace std;



int a[30005],b[30005],n,m;



int main()

{

	int i,j,k;

	priority_queue<int> Qmin,Qmax;



	scanf("%d%d",&n,&m);

	for(i=1;i<=n;i++) scanf("%d",&a[i]);

	for(i=1;i<=m;i++) scanf("%d",&b[i]);

	for(i=1;i<=m;i++)

	{

		for(j=b[i-1]+1;j<=b[i];j++)

		{

			Qmin.push(a[j]);

			k=-Qmin.top();

			Qmin.pop();

			Qmax.push(k);

		}

		k=-Qmax.top();

		Qmax.pop();

		Qmin.push(k);

		printf("%d\n",Qmin.top());

	}

	return 0;

}