区间 (interval)-牛客

题目描述

    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

说明

//树状数组 区间更新单点查询
#include<stdio.h>
#include<string.h>
#define lowbit(x) x&(-x)
int n;
int a[1000005];
int c[1000005];
//查询函数
int getSum(int x)
{
	int sum=0;
	for(int i=x;i<=n;i+=lowbit(i))
	{
		sum+=c[i];
	}
	return sum;
}
//更新函数
void update(int x,int v)
{
	for(int i=x;i>0;i-=lowbit(i))
	{
		c[i]+=v;
	}
}
int main()
{
	int m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	memset(c,0,sizeof(c));
	int q,l,r,p;
	while(m--)
	{
		scanf("%d%d%d%d",&q,&l,&r,&p);
		if(q==1)
		{
			update(r,-p);
			update(l-1,p);
		}
		else
		{
			update(r,p);
			update(l-1,-p);
		}
	}
	scanf("%d%d",&l,&r);
	long long s=0;
	for(int i=l;i<=r;i++)
	{
		s=s+a[i]+getSum(i);
	}
	printf("%lld",s);
 }