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[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;
}