ACM_前缀和

链接: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<=n,M<=1e6,所有输入数据都在int范围内

思路:用前缀和。不用每次修改原数组arr[],而是记录每次修改的sum[]的左右值,左值代表改变,右值代表恢复。最后求sum[]的前缀和,即全部次改变后原数组arr[]的每一列的改变值。最后把从x到y的arr[]和sum[]加起来就行了。
这里写图片描述

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <vector>
#include <string>
using namespace std;
typedef long long ll;
const int maxn=1e6+10;
ll arr[maxn];
ll sum[maxn];

int main()
{
    int n,m;
    ll q,l,r,p;
    cin>>n>>m;
    memset(sum,0,sizeof(sum));
    for(int i=1;i<=n;i++) scanf("%lld",&arr[i]);
    while(m--) 
    {
        scanf("%lld%lld%lld%lld",&q,&l,&r,&p);
        if(q==1){
            sum[l]-=p;//每次增减后的左差值 
            sum[r+1]+=p;//每次增减后的右差值(恢复) 
        }
        else{
            sum[l]+=p;
            sum[r+1]-=p;
        }
    }

    for(int i=1;i<=n;i++)//求增减的前缀和 ,即每列全部增减后的差值 
    {
        sum[i]+=sum[i-1];
    }
    int x,y;
    ll ans=0;
    cin>>x>>y;
    for(int i=x;i<=y;i++)//把从x到y的sum[i]加起来 
    {
        ans+=arr[i]+sum[i];
    }
    cout<<ans<<endl;
}

猜你喜欢

转载自blog.csdn.net/aaakkk_1996/article/details/81177513