天天看点

【nowcoder】Big Water Problem(线段树,单点更新,区间求和)

链接:https://www.nowcoder.com/acm/contest/77/B

来源:牛客网

时间限制:C/C++ 1秒,其他语言2秒

空间限制:C/C++ 131072K,其他语言262144K

64bit IO Format: %lld

题目描述

给一个数列,会有多次询问,对于每一次询问,会有两种操作:

1:给定两个整数x, y, 然后在原数组的第x位置上加y;

2:给定两个整数l,r,然后输出数组从第l位加到第r位数字的和并换行

输入描述:

第一行有两个整数n, m(1 <= n, m <= 100000)代表数列的长度和询问的次数

第二行n个数字,对于第i个数字a[i],(0<=a[i]<=100000)。

接下来m行,每一行有三个整数f, x, y。第一个整数f是1或者是2,代表操作类型,如果是1,接下来两个数x,y代表第x的位置上加y,如果是2,则求x到y的和,保证数据合法。

输出描述:

输出每次求和的结果并换行

示例1

输入

10 2

1 2 3 4 5 6 7 8 9 10

1 1 9

2 1 10

输出

64

分析:单点更新,区间求和

#include <bits/stdc++.h>
using namespace std;

#define mem(a,n) memset(a,n,sizeof(a))
#define memc(a,b) memcpy(a,b,sizeof(b))
#define rep(i,a,n) for(int i=a;i<n;i++) ///[a,n)
#define dec(i,n,a) for(int i=n;i>=a;i--)///[n,a]
#define pb push_back
#define fi first
#define se second
#define IO ios::sync_with_stdio(false)
#define fre freopen("in.txt","r",stdin)
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
typedef unsigned long long ull;
const double PI=acos(-);
const double E=;
const double eps=;
const int INF=;
const int MOD=;
const int N=+;
const ll maxn=+;
const int dir[][]= {-,,,,,-,,};
int sum[N<<],a[N];
void pushup(int rt)
{
    sum[rt]=sum[rt<<]+sum[rt<<|];
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%d",&sum[rt]);
        return ;
    }
    int m=(l+r)>>;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int p,int c,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]+=c;
        return ;
    }
    int m=(l+r)>>;
    if(p<=m)
        update(p,c,lson);
    else
        update(p,c,rson);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        return sum[rt];
    }
    int m=(l+r)>>;
    int ret=;
    if(L<=m)
        ret+=query(L,R,lson);
    if(R>m)
        ret+=query(L,R,rson);
    return ret;
}
int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    build(,n,);
    int op,a,b;
    while(q--)
    {
        scanf("%d%d%d",&op,&a,&b);
        if(op==)
            update(a,b,,n,);
        else
        {
            int ans=query(a,b,,n,);
            printf("%d\n",ans);
        }
    }
    return ;
}