树状数组总结

单点更新区间查询 HDU 1166

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#define LL long long
#define maxn 60000
using namespace std;
int n;
LL va,sum[maxn];
char s[15];
LL getSum(int x)
{
LL ans = 0;
for(int i=x;i>0;i-=(i&(-i)))ans+=sum[i];
return ans;
}
void update(int x,LL va)
{
for(int i=x;i<=n;i+=(i&(-i)))sum[i]+=va;
}
int main()
{
int t,i1 = 1;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++)
{
scanf("%I64d",&va);
update(i,va);
}
printf("Case %d:\n",i1);
i1++;
while(scanf("%s",s))
{
int a,b;
if(s[0]=='E')break;
else{
scanf("%d%d",&a,&b);
if(s[0]=='Q')printf("%I64d\n",getSum(b)-getSum(a-1));
else if(s[0]=='S')update(a,-b);
else update(a,b);
}
}
system("pause");
}
return 0;
}

HDU 1541 键值和下标反转的题目

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define maxn 36000
using namespace std;
int sum[maxn],n,ans1[maxn];
int getSum(int x)
{
int ans = 0;
for(;x;x-=(x&(-x)))ans+=sum[x];
return ans;
}
void update(int x)
{
for(;x<=maxn;x+=(x&(-x)))sum[x]+=1;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
memset(sum,0,sizeof(sum));
memset(ans1,0,sizeof(ans1));
int x,y;
for(int i=0;i<n;i++)
{
scanf("%d %d",&x,&y);
x++;
ans1[getSum(x)]++;
update(x);
}
for(int i=0;i<n;i++)printf("%d\n",ans1[i]);
}
return 0;
}

二维的单点的更新 poj 1195

#include<cstdio>
#include<algorithm>
#include<iostream>
#define maxn 1300
#define LL long long
using namespace std;
int sum[maxn][maxn],n,a,b,c,d,flag;
void update(int x,int y,int va)
{
for(int i=x;i<=n;i+=(i&(-i)))
for(int j=y;j<=n;j+=(j&(-j)))
sum[i][j]+=va;
}
int getSum(int x,int y)
{
LL ans = 0;
for(int i=x;i;i-=(i&(-i)))
for(int j=y;j;j-=(j&(-j)))ans+=sum[i][j];
return ans;
}
int main()
{
while(scanf("%d",&flag)!=EOF)
{
if(flag==0){
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)sum[i][j] = 0;
}
else if(flag==1)
{
scanf("%d%d%d",&a,&b,&c);
a++;
b++;
update(a,b,c);
}
else if(flag==2)
{
scanf("%d%d%d%d",&a,&b,&c,&d);
a++;b++;c++;d++;
printf("%d\n",getSum(c,d)-getSum(a-1,d)-getSum(c,b-1)+getSum(a-1,b-1));
}
else break;
}
return 0;
}

hdu 2852 与二分相结合使用的树状数组查询

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define maxn 100005
using namespace std;
int sum[maxn],m,flag,cnt[maxn];
int getSum(int x)
{
int ans = 0;
for(;x;x-=(x&(-x)))ans+=sum[x];
return ans;
}
void update(int x,int va)
{
for(;x<maxn;x+=(x&(-x)))sum[x]+=va;
}
void binSearch(int L,int am)
{
int R = maxn-1,ans = - 1,ll = getSum(L-1);
while(L<=R)
{
int mid = (L+R)/2;
int pre = getSum(mid)-ll;
if(pre<am)L = mid+1;
else {ans = mid;R = mid-1;}
}
if(ans==-1)printf("Not Find!\n");
else printf("%d\n",ans);

}
int main()
{
while(scanf("%d",&m)!=EOF)
{
memset(sum,0,sizeof(sum));
memset(cnt,0,sizeof(cnt));
while(m--)
{
int a,b;
scanf("%d",&flag);
if(flag==0)
{
scanf("%d",&a);
update(a,1);
cnt[a]++;
}
else if(flag==1)
{
scanf("%d",&a);
if(!cnt[a])printf("No Elment!\n");
else {
cnt[a]--;
update(a,-1);
}
}
else{
scanf("%d %d",&a,&b);
binSearch(a+1,b);
}
}
}
return 0;
}

hdu 3584 多维的区间更新单点查询

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
int sum[105][105][105],n,m,flag,cnt[105][105][105];
int getSum(int x,int y,int z)
{
int ans = 0;
for(int i=x;i;i-=(i&-i))
for(int j=y;j;j-=(j&-j))
for(int k=z;k;k-=(k&-k))ans+=sum[i][j][k];
return ans%2;
}
void update(int x,int y,int z)
{
for(int i=x;i<=n;i+=(i&-i))
for(int j=y;j<=n;j+=(j&-j))
for(int k=z;k<=n;k+=(k&-k))sum[i][j][k]++;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
for(int k=1;k<=n;k++)
{
sum[i][j][k] = 0,
cnt[i][j][k] = 0;
}
while(m--)
{
scanf("%d",&flag);
if(flag){
int x1,y1,z1,x2,y2,z2;
scanf("%d%d%d%d%d%d",&x1,&y1,&z1,&x2,&y2,&z2);
update(x1,y1,z1);
update(x2+1,y1,z1);
update(x1,y2+1,z1);
update(x1,y1,z2+1);
update(x2+1,y2+1,z1);
update(x2+1,y1,z2+1);
update(x1,y2+1,z2+1);
update(x2+1,y2+1,z2+1);
}
else{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
printf("%d\n",getSum(x,y,z));
}
}
}
return 0;
}

多维的区间更新区间查询 设两个树状数组一个存deg[i]累加和，另一个存i*deg[i]累加和

sum[x] = (x+1)*sum1(deg[i])+sum2(i*deg[i]) 1<=i<=x;