BZOJ_1798__Codevs_2216_[AHOI_2009]_行星序列_(线段树)

描述

BZOJ: http://www.lydsy.com/JudgeOnline/problem.php?id=1798

Codevs: http://codevs.cn/problem/2216/

给出n和行星的质量,进行m次操作:

1.将[l,r]区间内所有行星质量*c.

2.将[l,r]区间内所有行星质量+c.

3.询问[l,r]区间内行星质量和.

分析

双标记线段树,多加一个乘法的标记.a[k]位置的标记表示的是要传给a[k]的子节点的值,乘法位置为a[k].f,加法为a[k].p,表示的是先乘后加.按照先乘后加的规定,当跟新值时,ax+b变为(ax+b)*c1+c2=axc1+bc1+c2=(axc1)+(bc1+c2),也就是标记向下传递(push_down)的时候,乘法位置要乘,加法位置要先乘后加.

之前一直wa,最后才发现数组没开够...

感觉好像把乘法转化为加法也能做...但没尝试.

注意:

1.最好在确保扔给函数的参数都是取过模的,这样不容易出错.

2.可以不用long long定义,在乘法的时候强制转换一下就好了.

  1 #include<cstdio>
  2 #include<algorithm>
  3 #define ll long long
  4 #define lson 2*k
  5 #define rson 2*k+1
  6 
  7 const int maxn=1e5+5;
  8 int n,m,mod;
  9 int w[maxn];
 10 struct node { int l,r,x,f,p; }a[3*maxn];
 11 
 12 void build_tree(int l,int r,int k)
 13 {
 14     a[k].l=l; a[k].r=r; a[k].f=1; a[k].p=0;
 15     if(l==r)
 16     {
 17         a[k].x=w[l]; 
 18         return; 
 19     }    
 20     int mid=l+(r-l)/2;
 21     build_tree(l,mid,lson);
 22     build_tree(mid+1,r,rson);
 23     a[k].x=(a[lson].x+a[rson].x)%mod;
 24 }
 25 
 26 inline void cal(int k,int f,int p)
 27 {
 28     a[k].x=(int)((((ll)a[k].x*(ll)f)%mod+((ll)(a[k].r-a[k].l+1)%mod)*(ll)p)%mod);
 29     a[k].f=(int)(((ll)a[k].f*(ll)f)%mod);
 30     a[k].p=(int)((((ll)a[k].p*(ll)f)%mod+p)%mod);
 31 }
 32 
 33 inline void push_down(int k)
 34 {
 35     cal(lson,a[k].f,a[k].p);
 36     cal(rson,a[k].f,a[k].p);
 37     a[k].f=1;
 38     a[k].p=0;
 39 }
 40 
 41 void update(int l,int r,int k,int f,int p)
 42 {
 43     if(a[k].l==l&&a[k].r==r)
 44     {
 45         cal(k,f,p);
 46         return;
 47     }
 48     if(a[k].f!=1||a[k].p)
 49     {
 50         push_down(k);
 51     }
 52     int mid=a[k].l+(a[k].r-a[k].l)/2;
 53     if(r<=mid)
 54     {
 55         update(l,r,lson,f,p);
 56     }
 57     else if(l>mid)
 58     {
 59         update(l,r,rson,f,p);
 60     }
 61     else
 62     {
 63         update(l,mid,lson,f,p);
 64         update(mid+1,r,rson,f,p);
 65     }
 66     a[k].x=(a[lson].x+a[rson].x)%mod;
 67 }
 68 
 69 int search(int l,int r,int k)
 70 {
 71     if(a[k].l==l&&a[k].r==r)
 72     {
 73         return a[k].x;
 74     }
 75     if(a[k].f!=1||a[k].p)
 76     {
 77         push_down(k);
 78     }
 79     int mid=a[k].l+(a[k].r-a[k].l)/2;
 80     if(r<=mid)
 81     {
 82         return (search(l,r,lson)%mod);
 83     }
 84     else if(l>mid)
 85     {
 86         return (search(l,r,rson)%mod);
 87     }
 88     else
 89     {
 90         return ((search(l,mid,lson)+search(mid+1,r,rson))%mod);
 91     }
 92 }
 93 
 94 int main()
 95 {
 96 #ifndef ONLINE_JUDGE
 97     freopen("star.in","r",stdin);
 98     freopen("star.out","w",stdout);
 99 #endif
100     scanf("%d%d",&n,&mod);
101     
102     for(int i=1;i<=n;i++)
103     {
104         scanf("%d",&w[i]);
105     }
106     build_tree(1,n,1);
107     scanf("%d",&m);
108     int qry,l,r,c;
109     for(int i=1;i<=m;i++)
110     {
111         scanf("%d%d%d",&qry,&l,&r);
112         switch(qry)
113         {
114             case 1:
115                 scanf("%d",&c);
116                 c%=mod;
117                 update(l,r,1,c,0);
118                 break;
119             case 2:
120                 scanf("%d",&c);
121                 c%=mod;
122                 update(l,r,1,1,c);
123                 break;
124             case 3:
125                 printf("%d\n",search(l,r,1));
126                 break;
127         }
128     }
129 #ifndef ONLINE_JUDGE
130     fclose(stdin);
131     fclose(stdout);
132     system("star.out");
133 #endif
134     return 0;
135 }