luogu P1438 无聊的数列 |差分+线段树

标签：操作   线段树   ring   整数   space   return   ldo   void   i++   

题目描述

维护一个数列 \(a_i\)?，支持两种操作：

1 l r K D：给出一个长度等于 \(r-l+1\) 的等差数列，首项为 \(K\)，公差为 \(D\)，并将它对应加到 \([l,r]\) 范围中的每一个数上。即：令 \(a_l=a_l+K,a_{l+1}=a_{l+1}+K+D\ldots a_r=a_r+K+(r-l) \times\)。

2 p：询问序列的第 \(p\) 个数的值 \(a_p\)?。

输入格式

第一行两个整数数 \(n,m\) 表示数列长度和操作个数。

第二行 \(n\) 个整数，第 \(i\) 个数表示 \(a_i\)?。

接下来的 \(m\) 行，每行先输入一个整数 \(opt\)。

若 \(opt=1\) 则再输入四个整数 \(l\ r\ K\ D\)；

若 \(opt=2\) 则再输入一个整数 \(p\)。

输出格式

对于每个询问，一行一个整数表示答案。

差分+线段树

就没什么好说的吧

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=2e5+10;
#define int long long
int a[N],n,m;
struct Seg{
	int l,r,sum,add;
	#define l(x) tree[x].l
	#define r(x) tree[x].r
	#define sum(x) tree[x].sum
	#define add(x) tree[x].add
	#define len(x) (r(x)-l(x)+1)
}tree[N<<2];
#define ls p<<1
#define rs p<<1|1
#define mid ((l(p)+r(p))>>1)
void build(int p,int l,int r){
	l(p)=l,r(p)=r;
	if(l==r){
		sum(p)=a[l];
		return;
	}
	build(ls,l,mid);
	build(rs,mid+1,r);
	sum(p)=sum(ls)+sum(rs);
}
inline void pushdown(int p){
	sum(ls)+=add(p)*len(ls);
	sum(rs)+=add(p)*len(rs);
	add(ls)+=add(p);
	add(rs)+=add(p);
	add(p)=0;
}
void update(int p,int l,int r,int d){
	if(l<=l(p)&&r(p)<=r){
		sum(p)+=len(p)*d;
		add(p)+=d;
		return ;
	}
	if(add(p))pushdown(p);
	if(l<=mid)update(ls,l,r,d);
	if(r>mid)update(rs,l,r,d);
	sum(p)=sum(ls)+sum(rs);
}
int query(int p,int l,int r){
	if(l<=l(p)&&r(p)<=r)return sum(p);
	if(add(p))pushdown(p);
	int ans=0;
	if(l<=mid)ans+=query(ls,l,r);
	if(r>mid)ans+=query(rs,l,r);
	sum(p)=sum(ls)+sum(rs);
	return ans;
}
signed main(){
	cin>>n>>m;
	for(int i=1,x=0,y=0;i<=n;i++){
		scanf("%lld",&x);
		a[i]=x-y;
		y=x;
	}
	build(1,1,n);
	int opt,l,r,k,d;
	while(m--){
		scanf("%lld",&opt);
		if(opt==1){
			scanf("%lld%lld%lld%lld",&l,&r,&k,&d);
			update(1,l,l,k);
			if(l!=r)update(1,l+1,r,d);
			if(r<n)update(1,r+1,r+1,-k-(r-l)*d);
		}else{
			scanf("%lld",&k);
			printf("%lld\n",query(1,1,k));
		}
	}
}

luogu P1438 无聊的数列 |差分+线段树

标签：操作   线段树   ring   整数   space   return   ldo   void   i++   

原文地址：https://www.cnblogs.com/naruto-mzx/p/12722009.html