题目描述
老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。 有长为N的数列,不妨设为a1,a2,…,aN 。有如下三种操作形式: (1)把数列中的一段数全部乘一个值; (2)把数列中的一段数全部加一个值; (3)询问数列中的一段数的和,由于答案可能很大,你只需输出这个数模P的值。
输入输出格式
输入格式:
第一行两个整数N和P(1≤P≤1000000000)。 第二行含有N个非负整数,从左到右依次为a1,a2,…,aN, (0≤ai≤1000000000,1≤i≤N)。 第三行有一个整数M,表示操作总数。 从第四行开始每行描述一个操作,输入的操作有以下三种形式: 操作1:“1 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai×c (1≤t≤g≤N,0≤c≤1000000000)。 操作2:“2 t g c”(不含双引号)。表示把所有满足t≤i≤g的ai改为ai+c (1≤t≤g≤N,0≤c≤1000000000)。 操作3:“3 t g”(不含双引号)。询问所有满足t≤i≤g的ai的和模P的值 (1≤t≤g≤N)。 同一行相邻两数之间用一个空格隔开,每行开头和末尾没有多余空格。
输出格式:
对每个操作3,按照它在输入中出现的顺序,依次输出一行一个整数表示询问结果。
思路
线段树,打个乘法lazy标记即可
#include <cstdio> const int maxn = 100000 + 10; struct Seg { long long l,r,sum,add,mul; } tree[maxn*4]; long long p; long long n,m; inline void pushup(long long root) { tree[root].sum = tree[root<<1].sum+tree[root<<1|1].sum; tree[root].sum %= p; } inline void BuildTree(long long l,long long r,long long root) { tree[root].l = l; tree[root].r = r; tree[root].mul = 1; if (l == r) { scanf("%lld",&tree[root].sum); tree[root].sum %= p; return; } long long mid = l+r>>1; BuildTree(l,mid,root<<1); BuildTree(mid+1,r,root<<1|1); pushup(root); } inline void pushdown(long long root) { if (tree[root].mul != 1) { tree[root<<1].mul = tree[root<<1].mul*tree[root].mul%p; tree[root<<1|1].mul = tree[root<<1|1].mul*tree[root].mul%p; tree[root<<1].add = tree[root<<1].add*tree[root].mul%p; tree[root<<1|1].add = tree[root<<1|1].add*tree[root].mul%p; tree[root<<1].sum = tree[root<<1].sum*tree[root].mul%p; tree[root<<1|1].sum = tree[root<<1|1].sum*tree[root].mul%p; tree[root].mul = 1; } if (tree[root].add != 0) { tree[root<<1].add = (tree[root<<1].add+tree[root].add)%p; tree[root<<1|1].add = (tree[root<<1|1].add+tree[root].add)%p; tree[root<<1].sum = (tree[root<<1].sum+tree[root].add*(tree[root<<1].r-tree[root<<1].l+1))%p; tree[root<<1|1].sum = (tree[root<<1|1].sum+tree[root].add*(tree[root<<1|1].r-tree[root<<1|1].l+1))%p; tree[root].add = 0; } } inline void UpdateAdd(long long ql,long long qr,long long l,long long r,long long root,long long x) { if (ql > r || qr < l) return; if (ql <= l && qr >= r) { tree[root].add = (tree[root].add+x)%p; tree[root].sum = (tree[root].sum+x*(r-l+1))%p; return; } pushdown(root); long long mid = l+r>>1; UpdateAdd(ql,qr,l,mid,root<<1,x); UpdateAdd(ql,qr,mid+1,r,root<<1|1,x); pushup(root); } inline void UpdateMul(long long ql,long long qr,long long l,long long r,long long root,long long x) { if (ql > r || qr < l) return; if (ql <= l && qr >= r) { tree[root].add = tree[root].add*x%p; tree[root].mul = tree[root].mul*x%p; tree[root].sum = tree[root].sum*x%p; return; } pushdown(root); long long mid = l+r>>1; UpdateMul(ql,qr,l,mid,root<<1,x); UpdateMul(ql,qr,mid+1,r,root<<1|1,x); pushup(root); } inline long long Query(long long ql,long long qr,long long l,long long r,long long root) { if (ql > r || qr < l) return 0; if (ql <= l && qr >= r) return tree[root].sum; pushdown(root); long long mid = l+r>>1; return (Query(ql,qr,l,mid,root<<1)+Query(ql,qr,mid+1,r,root<<1|1))%p; } int main() { scanf("%lld%lld",&n,&p); BuildTree(1,n,1); scanf("%lld",&m); while (m--) { long long val; long long op,l,r; scanf("%lld%lld%lld",&op,&l,&r); if (op == 1) { scanf("%lld",&val); UpdateMul(l,r,1,n,1,val); } else if (op == 2) { scanf("%lld",&val); UpdateAdd(l,r,1,n,1,val); } else printf("%lld\n",Query(l,r,1,n,1)); } return 0; }