【我可能学了假线段树】【线段树】BZOJ1798 [AHOI2009]Seq 维护序列seq

    xiaoxiao2021-04-14  73

    题面在这里

    题解

    之前写的线段树都是数组版的,代码巨长…… 今天试着链式储存线段树,发现极其好用! pushdown,pushup,添加标记等等的操作都可以写成成员函数,代码十分简洁!

    而且Insert和Query操作都借鉴了ZH神犇的写法,非常省代码 所以说,我以前可能学了假的线段树

    另外,本题有一个注意点,就是如果一个节点同时拥有两种标记,会有优先级的问题 你不知道到底是先加还是先乘 我们可以强行先乘后加,但是如果得到标记的顺序是先加后乘呢? 没有关系,我们可以在进行乘操作时,对加标记也乘一下,就没有问题了

    附上代码:

    #include<cstdio> #define LL long long const int maxn=100005,INF=0x3f3f3f3f; int n,tt,q,a[maxn]; struct node{ node *l,*r; int L,R,s,add,mul; node(node *son=NULL,int x=0,int y=0) {l=r=son;L=x;R=y;add=0;mul=1;} void pushup() {s=(l->s+r->s)%tt;} void tag_add(LL w) {s=((R-L+1)*w%tt+s)%tt; add=(add+w)%tt;} void tag_mul(LL w) {s=s*w%tt; add=add*w%tt; mul=mul*w%tt;} void pushdown(){ if (L==R) return; if (mul!=1) {l->tag_mul(mul); r->tag_mul(mul); mul=1;} if (add!=0) {l->tag_add(add); r->tag_add(add); add=0;} } }nil; typedef node* P_node; P_node null,rot; void Seg_T_init(){ rot=null=&nil; null->l=null->r=null;null->s=0; } P_node buildtree(int l,int r){ P_node x=new node(null,l,r); if (l==r) {x->s=a[l]; return x;} int mid=l+r>>1; x->l=buildtree(l,mid);x->r=buildtree(mid+1,r); x->pushup();return x; } void ist_add(P_node x,int l,int r,int w){ if (r<x->L||x->R<l) return; x->pushdown(); if (l<=x->L&&x->R<=r) {x->tag_add(w);return;} ist_add(x->l,l,r,w);ist_add(x->r,l,r,w); x->pushup(); } void ist_mul(P_node x,int l,int r,int w){ if (r<x->L||x->R<l) return; x->pushdown(); if (l<=x->L&&x->R<=r) {x->tag_mul(w);return;} ist_mul(x->l,l,r,w);ist_mul(x->r,l,r,w); x->pushup(); } int qry(P_node x,int l,int r){ if (r<x->L||x->R<l) return 0; x->pushdown(); if (l<=x->L&&x->R<=r) return x->s; return (qry(x->l,l,r)+qry(x->r,l,r))%tt; } inline int red(){ int tot=0,f=1;char ch=getchar(); while (ch<'0'||'9'<ch) {if (ch=='-') f=-f;ch=getchar();} while ('0'<=ch&&ch<='9') tot=tot*10+ch-48,ch=getchar(); return tot*f; } int main(){ Seg_T_init(); n=red(),tt=red(); for (int i=1;i<=n;i++) a[i]=red(); rot=buildtree(1,n);q=red(); while (q--){ int c=red(); if (c==1){ int l=red(),r=red(),w=red(); ist_mul(rot,l,r,w); }else if (c==2){ int l=red(),r=red(),w=red(); ist_add(rot,l,r,w); }else{ int l=red(),r=red(); printf("%d\n",qry(rot,l,r)); } } return 0; }
    转载请注明原文地址: https://ju.6miu.com/read-670312.html

    最新回复(0)