网上关于树状数组的介绍很多,这里就不详细介绍,主要是总结模板
首先lowbit部分
int lowbit(int x) {return x&(-x);}
树状数组的标准用法是单点修改求前缀和
普通的用法如下(复杂度均为logn)
单点修改` 区间查询(求前缀和)初始化的话执行n次add操作
void add(int x,int k){
while(x<=n){
c[x]+=k;
x+=lowbit(x);
}
}
int query(int x){
int sum=0;
while(x>0){
sum+=c[x];
x-=lowbit(x);
}
return sum;
}
然后是区间更新,单点查询 这里使用差分的思想,维护差分数组c[i]
因为对一段区间都加上k ,他们他们之间的差值是不会变的,只有c[l] 和c[r+1]发生了变化,而数组c[i]的前缀和正好是a[i]
代码类似,只是修改的时候应该是add(l,k)和add(r+1,-k)
最后就是区间修改和区间查询 和上面的思路类似
void add(int x,int k){
int p=x;
while(x<=n){
sum1[x]+=k
sum2[x]+=k*(p-1); //注意是p
x+=lowbit(x);
}
}
void range_add(int l,int r,int k){
add(l,k);
add(r+1,-k);
}
int query(int x){
int p = x;
int sum=0;
while(x>0){
sum+=p*sum1[x]-sum2[x];
x-=lowbit(x);
}
return sum;
}
int range_query(int l,int r){
return query(r)-query(l-1); // 包括当前下标
}
poj 2299 树状数组求逆序对模板
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
const int MAX=5e5+5;
struct node{
int v;
int i;
}a[MAX];
int c[MAX];
int n;
bool cmp(const struct node a,const struct node b){
if(a.v==b.v) return a.i<b.i;
return a.v<b.v;
}
int lowbit(int x){return x&(-x);}
void add(int x,int k){
while(x<=n){
c[x]+=k;
x+=lowbit(x);
}
}
int query(int x){
int sum=0;
while(x>0){
sum+=c[x];
x-=lowbit(x);
}
return sum;
}
int main (){
//int n;
while(cin>>n&&n){
memset(c,0,sizeof(c));
for(int i=1;i<=n;i++){
cin>>a[i].v;
a[i].i=i;
}
sort(a+1,a+n+1,cmp);
long long ans=0;
for(int i=1;i<=n;i++){
add(a[i].i,1);
ans+=i-query(a[i].i);
}
cout<<ans<<endl;
}
return 0;
}