树状数组2的关键是lowbit()函数,lowbit(x)=x&-x,功能是找到x的二进制的最后一“1”(不得不感叹这个精妙之处),其原理是利用负数的补码,补码是源码取反加一,例如x=6=000110(2),-x=x补=111010(2),那么lowbit(x)=2;
大致上树状数组是运用三个函数来实现lowbit(),add(),sum(),
lowbit(x)引出一个tree[ ]数组,所有的计算都是围绕tree[ ]进行,
#define lowbit(x) ((x)& -(x))
void add(int x,int d){//单点修改
while(x<=n){
tree[x]+=d;
x+=lowbit(x);
}
}
int sum(int x){//求区间1-x的和
int sum=0;
while(x>0){
sum+=tree[x];
x-=lowbit(x);
}
return sum;
}
已poj 2182为例
(1)在n个位置上,每个位置上都有一头牛,及a1=a2=…=an=1,不过并不需要直接定义和使用数组a[ ]
(2)tree[ ]数组的初始化,这个题目比较特殊,不需要add初始化,因为lowbit()就是tree[i];
(3)程序所要做的,就是对每一个x[i]+1,用find_cow()找出对应的x,就是第x头牛之后,ax=0,方法就是add()更新数组tree[ ],及执行add(x,-1),
下面的程序完全套模板
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<string.h>
#define ll long long
#include<queue>
#include<map>
#include<set>
#define inf 0x3f3f3f3f
#define lowbit(x) ((x)& -(x))
using namespace std;
int n,tree[100005],x[100005];
int ans[100005];
void add(int x,int d){
while(x<=n){
tree[x]+=d;
x+=lowbit(x);
}
}
int sum(int x){
int sum=0;
while(x>0){
sum+=tree[x];
x-=lowbit(x);
}
return sum;
}
int find_cow(int x){
int l=1,r=n;
while(l<r){
int mid=(l+r)>>1;
if(sum(mid)<x)l=mid+1;
else r=mid;
}
return l;
}
int main(){
cin>>n;
x[1]=0;
for(int i=2;i<=n;i++){
scanf("%d",&x[i]);
}
for(int i=1;i<=n;i++){
tree[i]=lowbit(i);
}
for(int i=n;i>0;i--){
int r=find_cow(x[i]+1);
add(r,-1);
ans[i]=r;
}
for(int i=1;i<=n;i++){
printf("%d\n",ans[i]);
}
}