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

Posted 2020-07-30 Fighting Heart

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

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

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

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


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

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

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

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

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

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

    dfs(1,1,n);

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

    return 0;
}