首页 > 代码库 > BZOJ 1078 斜堆

BZOJ 1078 斜堆

感谢MATO大神的博客 http://www.cppblog.com/MatoNo1/archive/2013/03/03/192131.html

注意细节。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#define maxn 150using namespace std;int n,x,ls[maxn],rs[maxn],fath[maxn],ans[maxn],root=1;int find(){    int now=root;    while (ls[now])    {        if (!rs[now])        {            if ((!ls[ls[now]]) && (!rs[ls[now]])) now=ls[now];            break;        }        now=ls[now];    }    if (now==root) root=ls[now];    int ret=fath[now];ls[ret]=0;fath[ls[now]]=ret;ls[ret]=ls[now];    while (ret)    {        swap(ls[ret],rs[ret]);        ret=fath[ret];    }    return now;}int main(){    scanf("%d",&n);    for (int i=1;i<=n;i++)    {        scanf("%d",&x);        if (x>=100)        {            rs[x-100+1]=i+1;            fath[i+1]=x-100+1;        }        else        {            ls[x+1]=i+1;            fath[i+1]=x+1;        }    }    for (int i=n+1;i>=1;i--)        ans[i]=find()-1;    for (int i=1;i<=n+1;i++)        printf("%d ",ans[i]);    return 0;}

 

BZOJ 1078 斜堆