牛客ACM-Big Water Problem

最新推荐文章于 2024-12-06 11:24:29 发布

原创最新推荐文章于 2024-12-06 11:24:29 发布 · 181 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++ #开发语言

牛客-树状数组专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一道使用树状数组解决的更新查询问题，包括题目描述、输入输出格式及示例，并提供了完整的C++代码实现。适用于初学者了解树状数组的应用。

Big Water Problem

题目描述

给一个数列，会有多次询问，对于每一次询问，会有两种操作：
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

解题思路

一道标准的树状数组模板题，不会树状数组的朋友们请移步 Xenny-树状数组详解

代码

#include<iostream>
using namespace std;
const int N=1e6+7;

int n,m;
int a[N],c[N];

int lowbit(int x){
    return x&(-x);
}
int query(int x){
    int ret=0;
    while(x>0){
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
void add(int x,int k){
    while(x<=n){
        c[x]+=k;
        x+=lowbit(x);
    }
}
int main(){
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        add(i,a[i]);
    }
    int f,x,y;
    while(m--){
        cin>>f>>x>>y;
        if(f==1)add(x,y);
        else cout<<query(y)-query(x-1)<<endl;
    }
    return 0;
}