PAT (Advanced Level) 1099. Build A Binary Search Tree (30)

预处理每个节点左子树有多少个点。

然后确定值得时候递归下去就可以了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<algorithm>
using namespace std;

struct Node
{
    int val;
    int left;
    int right;
    int numL,numR;
}s[];
int n,a[];
vector<int>ans;

void dfs(int x)
{
    if(s[x].left!=-)
    {
        dfs(s[x].left);
        s[x].numL=s[s[x].left].numL+s[s[x].left].numR+;
    }

    if(s[x].right!=-)
    {
        dfs(s[x].right);
        s[x].numR=s[s[x].right].numL+s[s[x].right].numR+;
    }
}

void Find(int x,int L,int R)
{
    s[x].val=a[L+s[x].numL];
    if(s[x].left!=-) Find(s[x].left,L,L+s[x].numL-);
    if(s[x].right!=-) Find(s[x].right,L+s[x].numL+,R);
}

void bfs()
{
    queue<int>q;
    q.push();
    while(!q.empty())
    {
        int h=q.front(); q.pop();
        ans.push_back(s[h].val);
        if(s[h].left!=-) q.push(s[h].left);
        if(s[h].right!=-) q.push(s[h].right);
    }

}
int main()
{
    scanf("%d",&n);
    for(int i=;i<n;i++) {
        scanf("%d%d",&s[i].left,&s[i].right);
        s[i].numL=s[i].numR=;
    }
    for(int i=;i<=n;i++) scanf("%d",&a[i]);
    sort(a+,a++n);
    dfs();
    Find(,,n);
    bfs();
    for(int i=;i<ans.size();i++)
    {
        printf("%d",ans[i]);
        if(i<ans.size()-) printf(" ");
        else printf("\n");
    }
    return ;
}