信奥一本通-树状数组模版题目-修改数列元素+求子数列元素和

给定n个数列，规定有两种操作，一是修改某个元素，二是求子数列[a,b]的连续和。数列的元素个数最多10万个，询问操作最多10万次。

 
输入
第一行2个整数n,m（n表示输入n个数，m表示有m个操作）
第二行输入n个数列。
接下来m行，每行有三个数k,a,b（k=0表示求子数列[a,b]的和；k=1表示第a个数加b）。

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


样例输入
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
 

 

模板题。信奥p211

记住三步操作：求lowbit、对某个元素进行加法操作、查询某个前缀和。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<cmath>
#include<iostream>
using namespace std;
const int N=1e5+10;
int n,c[N];

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

void update(int x,int k)//在第x个位置(数)上加上k，并在树状数组中修改相应元素//对某个元素进行加法操作
{
    while(x<=n)
    {
        c[x]+=k;
        x+=lowbit(x);
    }
}

int sum(int x)//查询前缀和
{
    int res=0;
    while(x>0)
    {
        res+=c[x];
        x-=lowbit(x);
    }
    return res;
}

int main()
{
    int m,k;
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&k);
        update(i,k);
    }
    for(int i=1; i<=m; i++)
    {
        int kk,a,b;
        scanf("%d %d %d",&kk,&a,&b);
        if(kk==0)//求区间和且输出
        {
            int ans=sum(b)-sum(a-1);
            printf("%d\n",ans);
        }
        else if(kk==1)//把第a个位置的元素更改为b
            update(a,b);
    }
    return 0;
}