链接:https://www.nowcoder.com/acm/contest/135/I
来源:牛客网
区间 (interval)
时间限制:C/C++ 2秒,其他语言4秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
题目描述
Apojacsleam喜欢数组。
他现在有一个n个元素的数组a,而他要对a[L]-a[R]进行M次操作:
操作一:将a[L]-a[R]内的元素都加上P
操作二:将a[L]-a[R]内的元素都减去P
最后询问a[l]-a[r]内的元素之和?
请认真看题干及输入描述。
输入描述:
输入共M+3行: 第一行两个数,n,M,意义如“题目描述” 第二行n个数,描述数组。 第3-M+2行,共M行,每行四个数,q,L,R,P,若q为1则表示执行操作2,否则为执行操作1 第4行,两个正整数l,r
输出描述:
一个正整数,为a[l]-a[r]内的元素之和
示例1
输入
10 5 1 2 3 4 5 6 7 8 9 10 1 1 5 5 1 2 3 6 0 2 5 5 0 2 5 8 1 4 9 6 2 7
输出
23
说明
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<set>
const int MAXN=1e6+10;
using namespace std;
int lowbit(int x){return x&(-x);}
int n,Q,opt,x,y;
long long A[MAXN],C[MAXN],sum[MAXN],z;
void modifyA(int x,int val){
while(x<=n){
A[x]+=val;
x+=lowbit(x);
}
}
void modifyC(int x,long long val){
while(x<=n){
C[x]+=val;
x+=lowbit(x);
}
}
void update(int from,int to,long long val){
modifyA(from,val);
modifyA(to+1,-val);
modifyC(from,val*from);
modifyC(to+1,-val*(to+1));
}
long long queryA(int loc){
long long ans=0;
while(loc){
ans+=A[loc];
loc-=lowbit(loc);
}
return ans;
}
long long queryC(int loc){
long long ans=0;
while(loc){
ans+=C[loc];
loc-=lowbit(loc);
}
return ans;
}
long long query(int loc)
{
return sum[loc]+(loc+1)*queryA(loc)-queryC(loc);
}
long long query(int from,int to)
{
return query(to)-query(from-1);
}
int m;
int main()
{
scanf("%d%d",&n,&m);
for(register int i=1;i<=n;i++)scanf("%lld",&sum[i]),sum[i]+=sum[i-1];
while(m--){
scanf("%d%d%d",&opt,&x,&y);
if(opt==1)
{
scanf("%lld",&z);
update(x,y,-z);
}
else
{
scanf("%lld",&z);
update(x,y,z);
}
}
int L,R;
scanf("%d%d",&L,&R);
printf("%lld\n",query(L,R));
return 0;
}