数状数组和线段数

 

 

 

树状数组设为e[n],原数组设为a[n] 

树状数组主要有俩个功能：

1，查找 （求区间的前缀和）  2，修改、

查找：输入x ，求a[1]+a[2]+........a[n]   时间复杂度为O(log n)

修改：输入修改的位置x，a[x]需要增加的值val，这个过程本来是直接对原数组进行修改，时间复杂度为O(1)，但是这样不利于之后的查找操作，所以要在树状数组中进行修改时间复杂度为O(log n)

树状数组主要是求lowbit（x）还有e[x]

其中lowbit(x)可以用x&-x来求得

下面我们通过例题来介绍树状数组e[x]还有它的查找和修改操作

 

 

 

1，动态求连续区间和

给定 nn 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b][a,b] 的连续和。

输入格式

第一行包含两个整数 nn 和 mm，分别表示数的个数和操作次数。

第二行包含 nn 个整数，表示完整数列。

接下来 mm 行，每行包含三个整数 k,a,bk,a,b （k=0k=0，表示求子数列[a,b][a,b]的和；k=1k=1，表示第 aa 个数加 bb）。

数列从 11 开始计数。

输出格式

输出若干行数字，表示 k=0k=0 时，对应的子数列 [a,b][a,b] 的连续和。

数据范围

1≤n≤1000001≤n≤100000,
1≤m≤1000001≤m≤100000，
1≤a≤b≤n1≤a≤b≤n

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8

输出样例：

11
30
35

 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

typedef long long ll;
const int maxn=1e5+10;
int a[maxn],tr[maxn];
int n,m;
int lowbit(int x ){
    return x&-x;
}
void add(int a ,int val ){
    for(int i=a;i<=n;i+=lowbit(i)){
        tr[i]+=val;
    }
}
ll query(int x){
    ll res=0;
    for(int i=x;i>0;i-=lowbit(i)){
        res+=tr[i];
    }
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        cin>>a[i];
        add(i,a[i]);
    }
    while(m--){
        int k,x,y;
        scanf("%d%d%d",&k,&x,&y);
        if(k==0){
            ll res=query(y)-query(x-1);
            cout<<res<<endl;
        }
        else{
            a[x]+=y;
            add(x,y);
        }   
    }
    return 0;
}

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

typedef long long ll;
const int maxn=1e5+10;
//int a[maxn],tr[maxn];
int w[maxn];
int n,m;
struct node{
	int l,r;
	int sum;
}tr[4*maxn];
void pushup(int u){
	tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void build(int u,int l,int r){
	if(l==r) tr[u]={l,r,w[r]};
	else{
		int mid=l+r>>1;
		tr[u]={l,r};
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		pushup(u);
	}
}
int query(int u,int l,int r){
  	if(tr[u].l>=l&&tr[u].r<=r) return tr[u].sum;
  	int mid =tr[u].l+tr[u].r>>1;
  	int res=0;
  	if(l<=mid) res+=query(u<<1,l,r);
  	if(r>mid) res+=query(u<<1|1,l,r);
    return res;
}
void modify(int u,int x ,int y){
	if(tr[u].l==tr[u].r) tr[u].sum+=y;
	else{
		int mid=tr[u].l+tr[u].r>>1;
		if(x<=mid) modify(u<<1,x,y);
		else modify(u<<1|1,x,y);
		pushup(u);
	}
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        cin>>w[i];
    }
    build(1,1,n);
    int k,x,y;
    while(m--){
    	scanf("%d%d%d",&k,&x,&y);
    	if(k==0){
        	int res=query(1,x,y);
        	cout<<res<<endl;
    	}
    	else{
        	w[x]+=y;
        	modify(1,x,y);
    	}	
	}
    return 0;
}