vijos1512

一维和二维树状数组的使用

1.一维树状数组的使用

c[i]:以i结尾的树状数组规律计数和

注：下标均从1开始

int low_bit(int x){//k是x的二进制中从右往左数0的个数，返回2^k
    return x&(-x);
}

int Sum(int n){//求a[1...n]的和
    int sum=0;
    while(n){
        sum+=c[n];
        n-=low_bit(n);
    }
    return sum;
}

void update(int i,int x){//更新a[i]=x
    while(i<=n){
        c[i]+=x;
        i+=low_bit(i);
    }
}

 

2.二维树状数组 修改某个点的权值；统计矩形x1,y1,x2,y2内的点权值之和

c[i][j]表示以坐标(i,j)结尾的树状数组规律计数和

#include<stdio.h>
#include<iostream>
#include<string.h>
#define maxn 1030
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;

int n,q,ans;
int c[maxn][maxn];

inline int lowbit(int x){
    return ((x)&(-x));
}

void update(int x,int y,int k){
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=n;j+=lowbit(j))
            c[i][j]+=k;
}

int query(int x,int y){//求矩形1,1,x,y内的权值之和（两边均为闭区间）
    int sum=0;
    for(int i=x;i>0;i-=lowbit(i))
        for(int j=y;j>0;j-=lowbit(j))
            sum+=c[i][j];
    return sum;
}

int main(){
    scanf("%d",&n);
    while(scanf("%d",&q)!=EOF){
        if(q==3) break;
        if(q==1){
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            update(x+1,y+1,k);
        }
        if(q==2){
            int x1,y1,x2,y2;
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            ans=query(x2+1,y2+1)-query(x1,y2+1)-query(x2+1,y1)+query(x1,y1);
            printf("%d\n",ans);
        }
    }
    return 0;
}