【线段树】区间求和

题目描述

给定一数列，规定有两种操作，一是修改某个元素，二是求区间的连续和。

输入

输入数据第一行包含两个正整数n,m(n<=100000,m<=500000),以下是m行，

输出

每行有三个正整数k,a,b(k=0或1, a,b<=n).k=0时表示将a处数字加上b,k=1时表示询问区间[a,b]内所有数的和。对于每个询问输出对应的答案。

样例输入

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

样例输出

10
6
0
6
16
6
24
14
50
41

 

思路：单点修改区间查询的线段树板子题，但是要注意输入输出不能用cin、cout，否则会超时（嘤

 

代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e5+10;
ll sum[maxn*4];
int n,m;
void build(int l,int r,int root)
{
    //赋根节点的sum初值
    sum[root]=0;
    //退出递归的条件
    if(l==r) return;
    //找两个子节点
    int mid = (l+r)/2;
    build(l,mid,root*2);
    build(mid+1,r,root*2+1);
    //父结点等于两个子节点之和
    sum[root]=sum[root*2]+sum[root*2+1];
    return;
}
void update(int pos,int num,int l, int r,int root)
{
    if(l==r)//找到了节点
    {
        sum[root]+=num;
        return;
    }
    int mid = (l+r)/2;
    if(pos<=mid) update(pos,num,l,mid,root*2);
    else update(pos,num,mid+1,r,root*2+1);
    sum[root]=sum[root*2]+sum[root*2+1];
}
ll query(int x,int y,int l,int r,int root)
{
    //如果整个区间都是需要的，则直接返回这个区间的和，不必再细分
    if(x<=l&&y>=r)return sum[root];
    ll ans=0;
    int mid = (l+r)/2;
    //如果左边的节点有自己所需
    if(x<=mid) ans+=query(x,y,l,mid,root*2);
    //如果左边的节点有自己所需
    if(y>mid) ans+=query(x,y,mid+1,r,root*2+1);
    return ans;
}
int main()
{
    int k,a,b;
    //cin >> n>> m;
    scanf("%d %d",&n,&m);
    build(1,n,1);
    while(m--)
    {
        scanf("%d %d %d",&k,&a,&b);
        //cin>>k>>a>>b;
        //加数
        if(k==0)update(a,b,1,n,1);
        //查询
        else printf("%lld\n",query(a,b,1,n,1));
            //cout<<query(a,b,1,n,1)<<endl;
    }

    return 0;
}