1798: [Ahoi2009]Seq 维护序列seq
Time Limit: 30 Sec Memory Limit: 64 MBDescription
老师交给小可可一个维护数列的任务,现在小可可希望你来帮他完成。 有长为N的数列,不妨设为a1,a2,…,aN 。有如下三种操作形式: (1)把数列中的一段数全部乘一个值; (2)把数列中的一段数全部加一个值; (3)询问数列中的一段数的和,由于答案可能很大,你只需输出这个数模P的值。
Input
第一行两个整数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)。 同一行相邻两数之间用一个空格隔开,每行开头和末尾没有多余空格。
Output
对每个操作3,按照它在输入中出现的顺序,依次输出一行一个整数表示询问结果。
Sample Input
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
Sample Output
2 35 8
HINT
【样例说明】
初始时数列为(1,2,3,4,5,6,7)。经过第1次操作后,数列为(1,10,15,20,25,6,7)。对第2次操作,和为10+15+20=45,模43的结果是2。经过第3次操作后,数列为(1,10,24,29,34,15,16}对第4次操作,和为1+10+24=35,模43的结果是35。对第5次操作,和为29+34+15+16=94,模43的结果是8。测试数据规模如下表所示数据编号 1 2 3 4 5 6 7 8 9 10N= 10 1000 1000 10000 60000 70000 80000 90000 100000 100000M= 10 1000 1000 10000 60000 70000 80000 90000 100000 100000Source
裸线段树,之前做过,不过WA了一下午,终于找到原因了。。 每个询问读t,g,c的时候,用cin就RE用scanf就AC。。求解QAQ..
1 #include2 #include 3 #include 4 #include 5 #include 6 #include 7 #include 8 #include 9 using namespace std;10 const int N = 201000;11 typedef long long lld;12 #define Ch1 (i<<1) 13 #define Ch2 ((i<<1)|1)14 #define For(i,n) for(int i=1;i<=n;++i)15 #define Rep(i,l,r) for(int i=l;i<=r;++i)16 17 struct TREE{18 int l,r,mid;19 lld mul,plus,sum;20 }T[N<<1];21 22 int n,Mod,m;23 24 void Build(int l,int r,int i){ 25 T[i].l = l; T[i].r = r;T[i].mid = (l+r)/2;26 T[i].mul = 1;T[i].plus = 0;27 if(l==r) {28 scanf("%lld",&T[i].sum);29 return;30 }31 Build(l,T[i].mid,Ch1);32 Build(T[i].mid+1,r,Ch2);33 T[i].sum = (T[Ch1].sum + T[Ch2].sum) % Mod;34 }35 36 void Pushdown(int i){37 T[Ch1].mul= ( T[Ch1].mul * T[i].mul ) %Mod;38 T[Ch2].mul= ( T[Ch2].mul * T[i].mul ) %Mod;39 T[Ch1].plus=( T[Ch1].plus * T[i].mul + T[i].plus ) % Mod;40 T[Ch2].plus=( T[Ch2].plus * T[i].mul + T[i].plus ) % Mod; 41 T[Ch1].sum= ( T[Ch1].sum * T[i].mul ) %Mod;42 T[Ch2].sum= ( T[Ch2].sum * T[i].mul ) %Mod;43 T[Ch1].sum+=((T[Ch1].r-T[Ch1].l+1)*T[i].plus) % Mod;44 T[Ch2].sum+=((T[Ch2].r-T[Ch2].l+1)*T[i].plus) % Mod; 45 T[i].plus = 0;T[i].mul = 1;46 }47 48 void Modify(int l,int r,lld M,lld P,int i){49 if(l>r) return;50 if(l==T[i].l&&T[i].r==r){51 T[i].sum = ( T[i].sum * M + P * (T[i].r-T[i].l+1) ) % Mod;52 T[i].mul = ( T[i].mul * M ) %Mod;53 T[i].plus =( T[i].plus * M + P ) %Mod;54 return;55 }56 Pushdown(i);57 Modify(l,min(T[i].mid,r),M,P,Ch1);58 Modify(max(T[i].mid+1,l),r,M,P,Ch2);59 T[i].sum = (T[Ch1].sum + T[Ch2].sum)%Mod; 60 }61 62 lld query(int l,int r,int i){63 if(l>r) return 0;64 if(T[i].l==l&&T[i].r==r) return T[i].sum % Mod;65 Pushdown(i);66 return query(l,min(r,T[i].mid),Ch1)%Mod + query(max(T[i].mid+1,l),r,Ch2)%Mod;67 }68 69 int main(){70 scanf("%d%lld",&n,&Mod);71 Build(1,n,1);72 scanf("%d",&m);73 For(i,m){74 int op,t,g;75 lld c,M = 1 , P = 0;76 scanf("%d",&op);77 if(op==3) scanf("%d%d",&t,&g);78 else scanf("%d%d%lld",&t,&g,&c); 79 if(op==1) M = c;80 else P = c;81 if(op==3) printf("%lld\n",query(t,g,1)%Mod);82 else Modify(t,g,M,P,1);83 }84 return 0;85 }