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定义,在乘法的时候强制转换一下就好了.

 #include<cstdio>
 #include<algorithm>
 #define ll long long
 #define lson 2*k
 #define rson 2*k+1

 const int maxn=1e5+;
 int n,m,mod;
 int w[maxn];
 struct node { int l,r,x,f,p; }a[*maxn];

 void build_tree(int l,int r,int k)
 {
     a[k].l=l; a[k].r=r; a[k].f=; a[k].p=;
     if(l==r)
     {
         a[k].x=w[l];
         return;
     }
     int mid=l+(r-l)/;
     build_tree(l,mid,lson);
     build_tree(mid+,r,rson);
     a[k].x=(a[lson].x+a[rson].x)%mod;
 }

 inline void cal(int k,int f,int p)
 {
     a[k].x=(int)((((ll)a[k].x*(ll)f)%mod+((ll)(a[k].r-a[k].l+)%mod)*(ll)p)%mod);
     a[k].f=(int)(((ll)a[k].f*(ll)f)%mod);
     a[k].p=(int)((((ll)a[k].p*(ll)f)%mod+p)%mod);
 }

 inline void push_down(int k)
 {
     cal(lson,a[k].f,a[k].p);
     cal(rson,a[k].f,a[k].p);
     a[k].f=;
     a[k].p=;
 }

 void update(int l,int r,int k,int f,int p)
 {
     if(a[k].l==l&&a[k].r==r)
     {
         cal(k,f,p);
         return;
     }
     if(a[k].f!=||a[k].p)
     {
         push_down(k);
     }
     int mid=a[k].l+(a[k].r-a[k].l)/;
     if(r<=mid)
     {
         update(l,r,lson,f,p);
     }
     else if(l>mid)
     {
         update(l,r,rson,f,p);
     }
     else
     {
         update(l,mid,lson,f,p);
         update(mid+,r,rson,f,p);
     }
     a[k].x=(a[lson].x+a[rson].x)%mod;
 }

 int search(int l,int r,int k)
 {
     if(a[k].l==l&&a[k].r==r)
     {
         return a[k].x;
     }
     if(a[k].f!=||a[k].p)
     {
         push_down(k);
     }
     int mid=a[k].l+(a[k].r-a[k].l)/;
     if(r<=mid)
     {
         return (search(l,r,lson)%mod);
     }
     else if(l>mid)
     {
         return (search(l,r,rson)%mod);
     }
     else
     {
         return ((search(l,mid,lson)+search(mid+,r,rson))%mod);
     }
 }

 int main()
 {
 #ifndef ONLINE_JUDGE
     freopen("star.in","r",stdin);
     freopen("star.out","w",stdout);
 #endif
     scanf("%d%d",&n,&mod);

     for(int i=;i<=n;i++)
     {
         scanf("%d",&w[i]);
     }
     build_tree(,n,);
     scanf("%d",&m);
     int qry,l,r,c;
     for(int i=;i<=m;i++)
     {
         scanf("%d%d%d",&qry,&l,&r);
         switch(qry)
         {
             case :
                 scanf("%d",&c);
                 c%=mod;
                 update(l,r,,c,);
                 break;
             case :
                 scanf("%d",&c);
                 c%=mod;
                 update(l,r,,,c);
                 break;
             case :
                 printf("%d\n",search(l,r,));
                 break;
         }
     }
 #ifndef ONLINE_JUDGE
     fclose(stdin);
     fclose(stdout);
     system("star.out");
 #endif
     return ;
 }