牛客小白月赛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]内的元素之和

示例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=1000000+5;
long long flag[MAXN];
int ele[MAXN];
int main(){
    int n,M;
    cin>>n>>M;
    for(int i=1;i<=n;i++)
        scanf("%d",&ele[i]);
    int q,L,R,P;
    for(int i=0;i<M;i++){
        cin>>q>>L>>R>>P;
        if(q==1){
            flag[L]-=P;
            flag[R+1]+=P;
        }else{
            flag[L]+=P;
            flag[R+1]-=P;
        }
    }
 
 
    int l,r;
    cin>>l>>r;
    long long ans=0;
    long long tmp=0;
    for(int i=1;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 0;
}