PAT (Advanced Level) 1064. Complete Binary Search Tree (30)

因为是要构造完全二叉树，所以树的形状已经确定了。

因此只要递归确定每个节点是多少即可。

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

int a[+],ans[+];
int f[+],sz[+];
int n;

void t(int id)
{
    if(*id>n&&*id+>n)
    {
        f[id]=; sz[id]=; return;
    }

    if(*id<=n) t(*id);
    if(*id+<=n) t(*id+);

    f[id]=sz[*id];
    sz[id]=sz[*id]+sz[*id+]+;
}

void dfs(int id,int L,int R)
{
    ans[id]=a[L+f[id]];
    if(*id<=n) dfs(*id,L,L+f[id]-);
    if(*id+<=n) dfs(*id+,L+f[id]+,R);
}

int main()
{
    scanf("%d",&n);
    for(int i=;i<=n;i++) scanf("%d",&a[i]);
    sort(a+,a++n);

    memset(f,,sizeof f);
    memset(sz,,sizeof sz);
    t();

    dfs(,,n);

    for(int i=;i<=n;i++)
    {
        printf("%d",ans[i]);
        printf("%s",(i<n)?" ":"\n");
    }

    return ;
}