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\n",ans[i]);
}
}
return 0;
}

poj2182