POJ2182 Lost Cows 树状数组

一群牛，编号为1到n，但是编号乱了，已知每只牛的前面有多少只编号比其小，求出牛的编号。

插点问段。

这道题要从后面反推回来，比如最后的一只牛，知道有a只编号比它小，则它的编号为a+1。

update  更新已经确定的编号，

sum(i)   查询已经确定的编号中（后面的牛的编号都确定了），有多少个比i小。

#include<cstdio>
#include<algorithm>
#include<cstring>
const int MAXN=8000+5;
int c[MAXN];
int a[MAXN];
int ans[MAXN];
int tot;
int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int num,int n)
{
    while(x<=n){
        c[x]+=num;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int temp=0;
    while(x>0){
        temp+=c[x];
        x-=lowbit(x);
    }
    return temp;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF){
        memset(c,0,sizeof(c));
        memset(ans,0,sizeof(ans));
        tot=n;
        a[0]=a[1]=0;
        for(int i=2;i<=n;i++)
            scanf("%d",&a[i]);
        for(int i=n;i>0;i--){
            for(int j=a[i]+1;j<=n;j++){
                if(1+sum(j)+a[i]==j){
                    ans[tot]=j;
                    break;
                }
            }
            update(ans[tot],1,n);
            tot--;
        }
        for(int i=1;i<=n;i++){
            printf("%d
",ans[i]);
        }
    }
    return 0;
}