[洛谷 P1377] TJOI2011 树的序

问题描述

众所周知,二叉查找树的形态和键值的插入顺序密切相关。准确的讲:1、空树中加入一个键值k,则变为只有一个结点的二叉查找树,此结点的键值即为k;2、在非空树中插入一个键值k,若k小于其根的键值,则在其左子树中插入k,否则在其右子树中插入k。

我们将一棵二叉查找树的键值插入序列称为树的生成序列,现给出一个生成序列,求与其生成同样二叉查找树的所有生成序列中字典序最小的那个,其中,字典序关系是指对两个长度同为n的生成序列,先比较第一个插入键值,再比较第二个,依此类推。

输入格式

第一行,一个整数,n,表示二叉查找树的结点个数。第二行,有n个正整数,k1到kn,表示生成序列,简单起见,k1~kn为一个1到n的排列。

输出格式

一行,n个正整数,为能够生成同样二叉查找数的所有生成序列中最小的。

样例输入

4
1 3 4 2

样例输出

1 3 2 4

解析

根据题目要求,我们可以这么想:字典序最小的构法肯定是从小到大插入,但最后构出来的树会是一条链。我们可以设一个节点有两个关键值:一是本身的值,二是第几个插入。我们需要这棵树越往上第二关键字越小,即每个节点的第二关键字小于自己的两个儿子(不然不满足原树的顺序),整棵树的第一关键字又要满足二叉搜索树,这就是一棵笛卡尔树。

笛卡尔树的构造方法是:首先把所有节点按照第一关键字排序,这样构造出来的树就是一条在最右边的链。用一个栈保存最右边链的节点。对于每一个插入的节点i,都在栈中找到第一个第二关键字比自己小的点,并将i作为这个点的右儿子,而原来接的右链接到i的左儿子处,并从栈中弹出。最后中序遍历整棵树的序列即为答案。

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#define N 100002
using namespace std;
struct node{
    int dat,key;
}a[N];
int n,i,fa[N],dat[N],key[N],son[N][2],s[N],top;
int my_comp(const node &x,const node &y)
{
    return x.dat<y.dat;
}
void insert(int x,int f,int p)
{
    fa[x]=f;
    son[f][p]=x;
}
void dfs(int x)
{
    if(!x) return;
    cout<<dat[x]<<' ';
    dfs(son[x][0]);
    dfs(son[x][1]);
}
int main()
{
    cin>>n;
    for(i=1;i<=n;i++){
        cin>>a[i].dat;
        a[i].key=i;
    }
    sort(a+1,a+n+1,my_comp);
    for(i=1;i<=n;i++){
        int last=0;
        while(top>0&&key[s[top]]>a[i].key) last=top,top--;
        dat[i]=a[i].dat;
        key[i]=a[i].key;
        insert(i,s[top],1);
        insert(s[last],i,0);
        s[++top]=i;
    }
    dfs(son[0][1]);
    cout<<endl;
    return 0;
}
上一篇:小学期。。的新一天


下一篇:Java生成验证码