PAT(A)-1099. Build A Binary Search Tree (30)

记录一个菜逼的成长。。

题目链接

题目大意：
一棵二叉排序树，有n个点，告诉你每个点的左右儿子编号；
给你n个值。
确定每个值所在的位置。输出层序序列。

先将值排好序。
从根节点开始递归建树。
每次确定当前点在当前区间排第几，也就是要知道当前点的左子树上有多少个点。搜索一遍即可。

#include <bits/stdc++.h>
using namespace std;
#define fi first 
#define se second 
typedef pair<int,int> PII;
const int maxn = 100 + 10;
int a[maxn],d[maxn],val[maxn];
PII tree[maxn];
int dfs(int x)
{
    if(x == -1)return 0;
    return dfs(tree[x].fi) + dfs(tree[x].se) + 1;
}
void build(int t,int l,int r)
{
    if(l > r)return ;
    int son = dfs(tree[t].fi);
    val[t] = a[son+l];
    build(tree[t].fi,l,l+son-1);
    build(tree[t].se,son+l+1,r);

}
void bfs(int t)
{
    queue<int>q;
    q.push(t);
    int flag = 1;
    while(!q.empty()){
        int f = q.front();q.pop();
        if(flag)printf("%d",val[f]),flag = 0;
        else printf(" %d",val[f]);
        if(tree[f].fi != -1)q.push(tree[f].fi);
        if(tree[f].se != -1)q.push(tree[f].se);
    }
    puts("");
}
int main()
{
    int n;
    while(~scanf("%d",&n)){
        for( int i = 0; i < n; i++ ){
            int u,v;
            scanf("%d%d",&u,&v);
            tree[i].fi = u;
            if(u != -1)d[u]++;
            tree[i].se = v;
            if(v != -1)d[v]++;
        }
        for( int i = 0; i < n; i++ )
            scanf("%d",a+i);
        sort(a,a+n);
        int root;
        for( int i = 0; i < n; i++ ){
            if(!d[i])root = i;
        }
        build(root,0,n-1);
        bfs(root);
    }
    return 0;
}