牛客小白月赛5 I 区间 (interval) 【前缀和】

时间:2022-01-01 01:43:29

链接:https://www.nowcoder.com/acm/contest/135/I

题目描述

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]内的元素之和

输入例子:
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

-->

示例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

说明

牛客小白月赛5 I 区间 (interval) 【前缀和】
思路:
使用 懒标记数组,标记询问操作,最后统一计算。
题目中问的 区间和 可以使用 For 从头加到尾,也可在统一处理标记数组的时候求 前缀和,然后  sum[r] - sum[l-1]
AC代码:
 #include <cstdio>
#include <iostream>
using namespace std;
const int MAXN=+;
long long flag[MAXN];
int ele[MAXN];
int main(){
int n,M;
cin>>n>>M;
for(int i=;i<=n;i++)
scanf("%d",&ele[i]);
int q,L,R,P;
for(int i=;i<M;i++){
cin>>q>>L>>R>>P;
if(q==){
flag[L]-=P;
flag[R+]+=P;
}else{
flag[L]+=P;
flag[R+]-=P;
}
} int l,r;
cin>>l>>r;
long long ans=;
long long tmp=;
for(int i=;i<=n;i++){
tmp+=flag[i];
ele[i]=ele[i]+tmp;
} for(int i=l;i<=r;i++)
ans+=ele[i]; cout<<ans<<endl;
return ;
}