问题 A: 子数列连续和
时间限制: 1 Sec 内存限制: 128 MB
提交: 14 解决: 9
[提交] [状态] [讨论版] [命题人:admin]
题目描述
给定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<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<set>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define sca(x) scanf("%d",&x)
#define per(i,j,k) for(int i=j;i>=k;i--)
#define inf 0x3f3f3f3f
#define LL long long
#define N 100005
#define pb(x) push_back(x)
#define inf 0x3f3f3f3f
int read()
{
char c;int x=0,f=1;
c=getchar();
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
return x*f;
}
int len;
int a[N],b[N];
LL sum[N];
void add(int l,int k)
{
a[l]+=k;
sum[b[l]]+=k;
}
LL query(int l,int r)
{
LL ans=0;
for(int i=l;i<=min(b[l]*len,r);i++)ans+=a[i];
if(b[l]!=b[r])
{
for(int i=(b[r]-1)*len+1;i<=r;i++)ans+=a[i];
}
for(int i=b[l]+1;i<=b[r]-1;i++)ans+=sum[i];
return ans;
}
int main()
{
int n,m;
n=read(),m=read();
len=sqrt(n*1.0);
rep(i,1,n)a[i]=read();
rep(i,1,n)b[i]=(i-1)/len+1;
rep(i,1,n)sum[b[i]]+=a[i];
rep(i,1,m)
{
int op,l,r;
op=read(),l=read(),r=read();
if(op==1)add(l,r);
else printf("%lld\n",query(l,r));
}
}