【SCOI2008】bzoj1078 斜堆

最新推荐文章于 2018-09-07 15:44:00 发布

sdfzyhx

最新推荐文章于 2018-09-07 15:44:00 发布

阅读量381

点赞数

分类专栏：其他算法 bzoj 省选真题

本文链接：https://blog.csdn.net/sdfzyhx/article/details/72785317

版权

bzoj 同时被 3 个专栏收录

251 篇文章 0 订阅

订阅专栏

其他算法

234 篇文章 0 订阅

订阅专栏

省选真题

115 篇文章 0 订阅

订阅专栏

考虑递归求解，根据两个子树的大小大力讨论一下顺序。

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=110;
int lson[maxn],rson[maxn],size[maxn],ans[maxn][maxn],n;
void dfs(int u)
{
    if (!u) return;
    size[u]=1;
    dfs(lson[u]);
    size[u]+=size[lson[u]];
    dfs(rson[u]);
    size[u]+=size[rson[u]];
    int p=0,x;
    if (size[lson[u]]==size[rson[u]])
    {
        ans[u][++p]=u;
        for (int i=1;i<=size[lson[u]];i++)
        {
            ans[u][++p]=ans[rson[u]][i];
            ans[u][++p]=ans[lson[u]][i];
        }
    }
    else if (size[lson[u]]==size[rson[u]]+1)
    {
        ans[u][++p]=u;
        for (int i=1;i<=size[rson[u]];i++)
        {
            ans[u][++p]=ans[lson[u]][i];
            ans[u][++p]=ans[rson[u]][i];
        }
        ans[u][++p]=ans[lson[u]][size[lson[u]]];
    }
    else if (size[lson[u]]>size[rson[u]])
    {
        x=size[lson[u]]-size[rson[u]];
        for (int i=1;i<=x;i++) ans[u][++p]=ans[lson[u]][i];
        ans[u][++p]=u;
        for (int i=1;i<=size[rson[u]];i++)
        {
            ans[u][++p]=ans[rson[u]][i];
            ans[u][++p]=ans[lson[u]][x+i];
        }
    }
    else
    {
        x=size[rson[u]]-size[lson[u]]+1;
        for (int i=1;i<=x;i++) ans[u][++p]=ans[rson[u]][i];
        ans[u][++p]=u;
        ans[u][++p]=ans[lson[u]][1];
        for (int i=1;i<=size[lson[u]];i++)
        {
            ans[u][++p]=ans[rson[u]][x+i];
            ans[u][++p]=ans[lson[u]][i+1];
        }
    }
}
int main()
{
    int x;
    scanf("%d",&n);
    for (int i=2;i<=n+1;i++)
    {
        scanf("%d",&x);
        if (x<100) lson[x+1]=i;
        else rson[x-99]=i;
    }
    dfs(1);
    for (int i=1;i<=n+1;i++)
        printf("%d ",ans[1][i]-1);
}