题目描述
给定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
思路:建立一个树状数组进行求解。树状数组板子题
代码:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll a[100005],c[100005],n,k;
ll lowbit(ll x)//
{
return x&(-x);//获取两个相关位置的下标差值
}
void update(int i, ll k)//在a[i]上加k
{
while(i<=n)
{
c[i]+=k;//相关的前缀和加k
i+=lowbit(i);//找到下一个相关位置的下标
}
}
ll sum(int i)//取得1到i的前缀和
{
ll ans=0;//累加
while(i>0)
{
ans+=c[i];
i-=lowbit(i);
}
return ans;
}
int main()
{
scanf("%d %d",&n,&k);
ll q,qq,qqq;
for(int i = 1; i <= n; i++)
{
scanf("%lld",&a[i]);
update(i,a[i]);//用函数将相关的c加上值
}
for(int i = 1; i <= k; i++)
{
scanf("%lld %lld %lld",&q,&qq,&qqq);
if(!q) printf("%lld\n",sum(qqq)-sum(qq-1));
else update(qq,qqq);
}
return 0;
}