牛客小白月赛5——I 区间（前缀和）

最新推荐文章于 2020-12-02 22:16:48 发布

vocaloid01

最新推荐文章于 2020-12-02 22:16:48 发布

阅读量346

点赞数

CC 4.0 BY-SA版权

分类专栏：杂题

本文链接：https://blog.csdn.net/vocaloid01/article/details/81162883

杂题专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种解决区间操作求和问题的方法。通过使用前缀和而非线段树或树状数组来提高效率，适用于仅有一次查询的情况。文章提供了一个具体的示例代码，展示了如何通过前缀和的方式实现区间加减操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

输出

说明

反思：

一看见区间操作就头脑发热，上去就写线段树，然后超内存错误了五六发........

树状数组倒是可以直接过，线段树其实也行，只要别开lazy数组就行（开习惯了——头巨铁）。

今天又回过头来看这个题，发现只有一次查询，那么用前缀和更快且好写。看了别人的代码发现有人把每个数处理成前一个数的差值，最后再还原回来。个人觉的这个方法麻烦而且代码又长又慢，我这里多开了一个数组，最后只需要for循环一遍[1,R]就行。

PS：以后查询只有一次的时候先想想前缀和。

代码：

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 1e6+10;

long long board[MAXN];
long long Add[MAXN];

int main(){
	
	int N,M;
	long long d;
	int a,b,c;
	while(scanf("%d %d",&N,&M) == 2){
		memset(Add,0,sizeof Add);
		for(int i=1 ; i<=N ; ++i)scanf("%lld",&board[i]);
		for(int i=1 ; i<=M ; ++i){
			scanf("%d %d %d %lld",&a,&b,&c,&d);
			if(a == 1){
				Add[b] -= d;
				Add[c+1] += d;
			}
			else {
				Add[b] += d;
				Add[c+1] -= d;
			}
		}
		long long add = 0;
		long long sum = 0;
		int L,R;
		scanf("%d %d",&L,&R);
		for(int i=1 ; i<=R ; ++i){
			add += Add[i];
			if(i>=L && i<=R)sum += add+board[i];
		}
		printf("%lld\n",sum);
	}
	
	return 0;
}