牛客网小白月赛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

说明

 
 

题意:

给你一个数组,对数组区间进行N次操作,同加P,或者是同减P

问l到r区间的元素和

思路:

当时做题的时候写了线段树,

直接暴力模拟会T掉,

后来发现直接求前缀和就可以

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int maxn=1e6+86;

int data[maxn];
int add[maxn];

int main()
{
    int n,m;
    int a,b,c,d;
    int l,r;
    long long ans;
    scanf("%d %d",&n,&m);
    ans=0;
    memset(data,0,sizeof(data));
    memset(add,0,sizeof(add));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&data[i]);
    }
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d %d",&a,&b,&c,&d);
        if(a==1)
        {
            add[b]-=d;
            add[c+1]+=d;
        }
        else
        {
            add[b]+=d;
            add[c+1]-=d;
        }
    }
    for(int i=1;i<=n;i++)
    {
        add[i]+=add[i-1];
    }
    scanf("%d %d",&l,&r);
    for(int i=l;i<=r;i++)
    {
        ans+=data[i];
        ans+=add[i];
    }
    printf("%lld\n",ans);
}

猜你喜欢

转载自blog.csdn.net/leper_gnome/article/details/81173996
今日推荐