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

Posted 2021-02-28 naruto-mzx

题目描述

维护一个数列 (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+Dldots a_r=a_r+K+(r-l) imes)。

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
",query(1,1,k));
		}
	}
}