POJ-2201-Cartesian Tree(笛卡尔树)

Description

Input

Output

Sample Input

7
5 4
2 2
3 9
0 5
1 3
6 6
4 11

Sample Output

YES
2 3 6
0 5 1
1 0 7
5 0 0
2 4 0
1 0 0
3 0 0

Source

思路：题目中说全部的key值都不一样，因此不会存在NO的情况。

直接建树再输出就可以。

#include <cstdio>
#include <algorithm>
using namespace std;

struct S{
int id,key,val,parent,l,r;
}node[50005];

bool cmp(struct S a,struct S b)
{
    return a.key<b.key;
}

int stk[50005],top,p[50005],l[50005],r[50005];

void build(int n)//由于已经按key值升序排序了，所以后面增加的节点要么成为某个节点的右儿子。要么让根节点成为自己的左儿子
{
    int i;

    top=0;

    stk[top]=1;

    for(i=2;i<=n;i++)
    {
        while(top>=0 && node[stk[top]].val>node[i].val) top--;

        if(top>-1)//假设找到一个不大于自己的节点。就成为该节点的右儿子，还要注意让该节点原来的右儿子变成自己的左儿子(key比其大而且小于其val)
        {
            node[i].parent=stk[top];
            node[node[stk[top]].r].parent=i;
            node[i].l=node[stk[top]].r;
            node[stk[top]].r=i;
        }
        else//假设没找到不大于自己的节点。就变成新的根
        {
            node[stk[0]].parent=i;
            node[i].l=stk[0];
        }

        stk[++top]=i;
    }
}

int main()
{
    int n,i;

    while(~scanf("%d",&n))
    {
        for(i=1;i<=n;i++)
        {
            scanf("%d%d",&node[i].key,&node[i].val);

            node[i].id=i;

            node[i].parent=node[i].l=node[i].r=0;//初始化
        }

        sort(node+1,node+n+1,cmp);

        build(n);

        for(i=1;i<=n;i++)
        {
            p[node[i].id]=node[node[i].parent].id;
            l[node[i].id]=node[node[i].l].id;
            r[node[i].id]=node[node[i].r].id;
        }

        printf("YES\n");

        for(i=1;i<=n;i++) printf("%d %d %d\n",p[i],l[i],r[i]);
    }
}