POJ 2352 Stars（treap练习）

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

题意：给出平面上一些点，每个点的level值为其左下角的点的个数（包括正下和左侧）。输出level值在[0,n-1] 的点各有多少个？

思路：由于给出的点是按照y升序然后x升序，所以将x插入treap，每次统计左子树的节点个数。

struct node

{

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

};



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+1;

        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+1;

        x=y;

    }



    void insert(int &k,int key)

    {

        if(k==-1)

        {

            k=++size;

            p[k].L=p[k].R=-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

        {

            p[k].cntR++;

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

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

        }

    }





    int find(int k,int key)

    {

        if(k==-1) return 0;

        if(key<p[k].key) return find(p[k].L,key);

        return p[k].cntL+1+find(p[k].R,key);

    }

};



struct point

{

    int x,y;



    void get()

    {

        RD(x,y);

    }

};





point p[N];

treap a;

int n,ans[N];



int main()

{

    Rush(n)

    {

        a.init(); clr(ans,0);

        int i;

        FOR0(i,n) p[i].get();

        FOR0(i,n)

        {

            a.insert(a.root,p[i].x);

            ans[a.find(a.root,p[i].x)-1]++;

        }

        FOR0(i,n) PR(ans[i]);

    }

    return 0;

}