刷题记录:牛客NC19889[AHOI2009]SEQ 维护序列SEQ

传送门:牛客

题目描述:

老师交给小可可一个维护数列的任务，现在小可可希望你来帮他完成。 有长为N的数列，不妨设为a1,a2,…,aN 。
有如下三种操作形式： 
(1)把数列中的一段数全部乘一个值; 
(2)把数列中的一段数全部加一个值; 
(3)询问数列中的一段数的和，由于答案可能很大，你只需输出这个数模P的值。
输入:
7 43
1 2 3 4 5 6 7
5
1 2 5 5
3 2 4
2 3 7 9
3 1 3
3 4 7
输出:
2
35
8

线段树维护区间乘修改+区间加修改的题目,是这道题的简化版,具体操作在那道题讲的很清楚了,可以参考那道题的题解,此处就不在赘述了

下面是具体的代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define int long long
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
struct Segment_tree{
	int l,r,sum,mul,add;
}tree[maxn*4];
int n,p;int a[maxn];int m;
void pushup(int rt) {
	tree[rt].sum=(tree[ls].sum+tree[rs].sum)%p;
}
void build(int l,int r,int rt) {
	tree[rt].l=l;tree[rt].r=r;tree[rt].mul=1;
	if(l==r) {
		tree[rt].sum=a[l];
		return ;
	}
	int mid=(l+r)>>1;
	build(lson);build(rson);
	pushup(rt);
}
void pushdown(int rt) {
	int lenls=tree[ls].r-tree[ls].l+1;
	int lenrs=tree[rs].r-tree[rs].l+1;
	tree[ls].add=(tree[ls].add*tree[rt].mul%p+tree[rt].add)%p;
	tree[rs].add=(tree[rs].add*tree[rt].mul%p+tree[rt].add)%p;
	tree[ls].mul=tree[ls].mul*tree[rt].mul%p;
	tree[rs].mul=tree[rs].mul*tree[rt].mul%p;
	tree[ls].sum=(tree[ls].sum*tree[rt].mul%p+tree[rt].add*lenls%p)%p;
	tree[rs].sum=(tree[rs].sum*tree[rt].mul%p+tree[rt].add*lenrs%p)%p;
	tree[rt].add=0;tree[rt].mul=1;
}
void update(int l,int r,int rt,int mul,int add) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		tree[rt].add=(tree[rt].add*mul%p+add)%p;
		tree[rt].mul=tree[rt].mul*mul%p;
		tree[rt].sum=(tree[rt].sum*mul%p+add*(r-l+1)%p)%p;
		return ;
	}
	if(tree[rt].mul!=1||tree[rt].add) pushdown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) update(l,r,ls,mul,add);
	else if(l>mid) update(l,r,rs,mul,add);
	else update(l,mid,ls,mul,add),update(mid+1,r,rs,mul,add);
	pushup(rt);
}
int query(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		return tree[rt].sum;
	}
	if(tree[rt].mul!=1||tree[rt].add) pushdown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) return query(l,r,ls);
	else if(l>mid) return query(l,r,rs);
	else return (query(l,mid,ls)+query(mid+1,r,rs))%p;
}
signed main() {
	n=read();p=read();
	for(int i=1;i<=n;i++) a[i]=read();
	build(root);
	m=read();
	for(int i=1;i<=m;i++) {
		int opt=read();
		if(opt==1) {
			int l=read(),r=read(),c=read();
			update(l,r,1,c,0);
		}
		else if(opt==2) {
			int l=read(),r=read(),c=read();
			update(l,r,1,1,c);
		}
		else {
			int l=read(),r=read();
			printf("%lld\n",query(l,r,1));
		}
	}
	return 0;	
}