zoukankan      html  css  js  c++  java
  • 线段树区间加乘(模板)

    马上要进入暑假的训练了,所以复习了一下线段树的模板,感觉同最开始的理解又加深了一些,所以还是写博客总结一下,方便日后复习。

    洛谷3373是一道线段树区间加乘模板(裸),今天就又敲了一遍,有几个地方是我自己觉得很重要的。

    1.考虑是否要开Long Long。

    2.乘的标记一开始赋为1,在标记下传后也要请为1(这个很简单,但我觉得容易忘)

    3.在标记更新的时候一定要先乘后加,这个很重要。

    4.乘标记更新时会对加标记造成影响,加标记更新时不会对乘标记造成影响。

    void modify_c(int k,int L,int R,int l,int r,LL v)
    {
        if(L>=l&&R<=r)
        {
            mul[k]=mul[k]*v%p;
            add[k]=add[k]*v%p;
            sum[k]=sum[k]*v%p;
            return;
        }
        pushdown(k,L,R);
        int mid=(L+R)>>1;
        if(l<=mid) modify_c(lc[k],L,mid,l,r,v);
        if(r>mid) modify_c(rc[k],mid+1,R,l,r,v);
        sum[k]=(sum[lc[k]]+sum[rc[k]])%p;
    }
    乘标记更新

    5.pushdown中更新下面的总值的时候不把要更新的点的add和mul也弄进去,因为它会在modify的时候被更新,在L>=l&&R<=r里。(具体看代码)

    #include<bits/stdc++.h>
    #define LL long long
    #define N 100003
    using namespace std;
    int p;
    int ndnum=1;
    LL sum[N<<2],mul[N<<2],add[N<<2],lc[N<<2],rc[N<<2],a[N];
    int read()
    {
        int f=1,x=0;char s=getchar();
        while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
        while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
        return x*f;
    }
    void build(int k,int l,int r)
    {
        mul[k]=1;
        if(l==r) {sum[k]=a[l]%p;return;}
        int mid=(l+r)>>1;
        build(lc[k]=++ndnum,l,mid);
        build(rc[k]=++ndnum,mid+1,r);
        sum[k]=(sum[lc[k]]+sum[rc[k]])%p;
    }
    
    void pushdown(int k,int l,int r)
    {
        //为什么在pushdown中更新下面的总值的时候为什么不把要更新的点的mul也乘进去,
        //是因为在modify的时候我已经用他原来的mul值更新过一遍了,当然不需要乘进去
        //add同理 
        //为什么要把mul的值和add的值又更新到要更新的点的mul和add上呢?
        //这个则是因为后面又要pushdown的时候,一定要连着这个一起pushdown下去。
        mul[lc[k]]=mul[lc[k]]*mul[k]%p;
        mul[rc[k]]=mul[rc[k]]*mul[k]%p;
        
        add[lc[k]]=add[lc[k]]*mul[k]%p;
        add[rc[k]]=add[rc[k]]*mul[k]%p;
        add[lc[k]]=(add[lc[k]]+add[k])%p;
        add[rc[k]]=(add[rc[k]]+add[k])%p;
        
        int mid=(l+r)>>1;
        sum[lc[k]]=sum[lc[k]]*mul[k]%p;
        sum[rc[k]]=sum[rc[k]]*mul[k]%p;
        sum[lc[k]]=(sum[lc[k]]+add[k]*(mid-l+1)%p)%p;
        sum[rc[k]]=(sum[rc[k]]+add[k]*(r-mid)%p)%p;
        
        add[k]=0;mul[k]=1;
    }
    void modify_add(int k,int L,int R,int l,int r,LL v)
    {
        if(L>=l&&R<=r)
        {
            add[k]=(add[k]+v)%p;
            sum[k]=(sum[k]+(R-L+1)*v)%p;
            return;
        }
        pushdown(k,L,R);
        int mid=(L+R)>>1;
        if(l<=mid)modify_add(lc[k],L,mid,l,r,v);
        if(r>mid)modify_add(rc[k],mid+1,R,l,r,v);
        sum[k]=(sum[lc[k]]+sum[rc[k]])%p;
    }
    void modify_c(int k,int L,int R,int l,int r,LL v)
    {
        if(L>=l&&R<=r)
        {
            mul[k]=mul[k]*v%p;
            add[k]=add[k]*v%p;
            sum[k]=sum[k]*v%p;//就是在这里,已经用他原来的mul值更新过一遍了,所以在pushdown的时候sum[lc[k]]是乘加mul[k]和add[k] 
            return;
        }
        pushdown(k,L,R);
        int mid=(L+R)>>1;
        if(l<=mid) modify_c(lc[k],L,mid,l,r,v);
        if(r>mid) modify_c(rc[k],mid+1,R,l,r,v);
        sum[k]=(sum[lc[k]]+sum[rc[k]])%p;
    }
    LL query(int k,int L,int R,int l,int r)
    {
        if(L>=l&&R<=r) return sum[k];
        int mid=(L+R)>>1;
        pushdown(k,L,R);
        LL ans=0;
        if(l<=mid) ans=(ans+query(lc[k],L,mid,l,r))%p;
        if(r>mid) ans=(ans+query(rc[k],mid+1,R,l,r))%p;
        return ans%p;
    }
    int main()
    {
        int n=read(),m=read();p=read();
        for(int i=1;i<=n;++i) a[i]=read();
        build(1,1,n);
        for(int i=1;i<=m;++i)
        {
            int op=read(),x=read(),y=read();
            if(op==1)
            {
                LL k=read();
                modify_c(1,1,n,x,y,k);
            }
            if(op==2)
            {
                LL k=read();
                modify_add(1,1,n,x,y,k);
            }
            if(op==3) printf("%lld
    ",query(1,1,n,x,y));
        }
    }
    洛谷3373

    让我很无语的是调了好久样例过不了,最后发现是没有在main里写build。。。天,我是什么神仙。

  • 相关阅读:
    第一课:js命名空间的介绍,js对象的扩展以及js数组化
    浏览器缓存机制-社招必问知识
    2013年前端校园招聘经历
    GBDT(MART) 迭代决策树简介
    coursera 公开课 文本挖掘和分析(text mining and analytics) week 1 笔记
    Predicting purchase behavior from social media-www2013
    Recommending branded products from social media -RecSys 2013-20160422
    2016年数据挖掘,机器学习可投会议
    java 中遍历hashmap 和hashset 的方法
    NLPIR分词工具的使用(java环境下)
  • 原文地址:https://www.cnblogs.com/yyys-/p/11143231.html
Copyright © 2011-2022 走看看