POJ 2182 Lost Cows 树状数组

做法：倒去确定原先的位置，这一点和这题很像。现在用的是树状数组，简短了不少，值得注意的是find函数中的判断，当经过若干次排列以后，可能会产生sum(x)==sum(y),其中have[x]!=have[y],这个bug需要在find函数中改，可以确定的是x<y，所有find中只要进行小小想改动即可

#include <iostream>
#include <cstring>
#include <cstdio>
#define LMT 8005
#define lowbit(x) x&(-x)
using namespace std;
int num[LMT],n;
bool have[LMT];
void init(void)
{
    for(int i=1;i<=n;i++)num[i]=i&(-i);
    memset(have,1,sizeof(have));
}
void update(int x,int ope)
{
    while(x<=n)
    {
        num[x]+=ope;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int ret=0;
    while(x>0)
    {
        ret+=num[x];
        x-=lowbit(x);
    }
    return ret;
}
int find(int ord)
{
    int m,s,l=1,r=n;
    while(l<=r)
    {
        m=(l+r)>>1;
        s=sum(m);
        if(s==ord&&have[m])return m;//这里
        if(s>=ord)r=m-1;
        else l=m+1;
    }
    return 0;
}
int main()
{
    int ord[LMT],ans[LMT];
    while(~scanf("%d",&n))
    {
        init();
        for(int i=2;i<=n;i++)scanf("%d",&ord[i]);
        for(int i=n;i>=2;i--)
        {
            ans[i]=find(ord[i]+1);
            update(ans[i],-1);
            have[ans[i]]=0;
        }
        for(int i=1;i<=n;i++)
        if(have[i])ans[1]=i;
        for(int i=1;i<=n;i++)printf("%d\n",ans[i]);
    }
    return 0;
}