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

链接：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

说明

 

思路：

 

使用 懒标记数组，标记询问操作，最后统一计算。

题目中问的 区间和 可以使用 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 ;
 }