• 【codevs4927】线段树练习


    题目大意:维护一个序列,支持区间加、区间染色、区间最值查询、区间和查询。

    题解:对于区间赋值操作来说,维护一个赋值标记,注意,这里不能直接用赋值的值直接维护,因为不像加法标记,0 表示不用处理,这里 0 会表示成将区间赋值为 0,在这里卡了一小会QAQ。另外,关于多标记线段树下传的规则,可以考虑两种不同的下传方式,将序列最终的值表示出来,在额外考虑加入优先级高的标记,如果对优先级低的标记产生了不好的影响,则优先级关系应该相反。

    代码如下

    #include <bits/stdc++.h>
    using namespace std;
    const int maxn=1e5+10;
    
    inline long long read(){
    	long long x=0,f=1;char ch;
    	do{ch=getchar();if(ch=='-')f=-1;}while(!isdigit(ch));
    	do{x=x*10+ch-'0';ch=getchar();}while(isdigit(ch));
    	return f*x;
    }
    
    int n,m;
    long long a[maxn];
    char op[10];
    struct node{
    	#define ls(x) t[x].lc
    	#define rs(x) t[x].rc
    	int lc,rc;
    	long long sum,mx,mi,stag,atag;bool s;
    }t[maxn<<1];
    int tot,root;
    inline void pushup(int o){
    	t[o].mi=min(t[ls(o)].mi,t[rs(o)].mi);
    	t[o].mx=max(t[ls(o)].mx,t[rs(o)].mx);
    	t[o].sum=t[ls(o)].sum+t[rs(o)].sum;
    }
    inline void pushdown(int o,int l,int r){
    	int mid=l+r>>1;
    	if(t[o].s){
    		t[ls(o)].stag=t[o].stag,t[ls(o)].atag=0,t[ls(o)].sum=(mid-l+1)*t[o].stag,t[ls(o)].mi=t[ls(o)].mx=t[o].stag,t[ls(o)].s=1;
    		t[rs(o)].stag=t[o].stag,t[rs(o)].atag=0,t[rs(o)].sum=(r-mid)*t[o].stag,t[rs(o)].mi=t[rs(o)].mx=t[o].stag,t[rs(o)].s=1;
    		t[o].s=0;
    	}
    	if(t[o].atag){
    		t[ls(o)].atag+=t[o].atag,t[ls(o)].sum+=(mid-l+1)*t[o].atag,t[ls(o)].mx+=t[o].atag,t[ls(o)].mi+=t[o].atag;
    		t[rs(o)].atag+=t[o].atag,t[rs(o)].sum+=(r-mid)*t[o].atag,t[rs(o)].mx+=t[o].atag,t[rs(o)].mi+=t[o].atag;
    		t[o].atag=0;
    	}
    }
    int build(int l,int r){
    	int o=++tot;
    	if(l==r){t[o].sum=t[o].mi=t[o].mx=a[l];return o;}
    	int mid=l+r>>1;
    	ls(o)=build(l,mid),rs(o)=build(mid+1,r);
    	return pushup(o),o;
    }
    void modify(int o,int l,int r,int x,int y,long long val,int opt){//1 -> set, 2 -> add
    	if(l==x&&r==y){
    		if(opt==1)t[o].s=1,t[o].stag=val,t[o].atag=0,t[o].sum=(r-l+1)*val,t[o].mi=t[o].mx=val;
    		else t[o].atag+=val,t[o].sum+=(r-l+1)*val,t[o].mi+=val,t[o].mx+=val;
    		return;
    	}
    	int mid=l+r>>1;
    	pushdown(o,l,r);
    	if(y<=mid)modify(ls(o),l,mid,x,y,val,opt);
    	else if(x>mid)modify(rs(o),mid+1,r,x,y,val,opt);
    	else modify(ls(o),l,mid,x,mid,val,opt),modify(rs(o),mid+1,r,mid+1,y,val,opt);
    	pushup(o);
    }
    long long query(int o,int l,int r,int x,int y,int opt){// 1 -> sum, 2 -> min, 3 -> max
    	if(l==x&&r==y){
    		if(opt==1)return t[o].sum;
    		else if(opt==2)return t[o].mi;
    		else return t[o].mx;
    	}
    	int mid=l+r>>1;
    	pushdown(o,l,r);
    	if(y<=mid)return query(ls(o),l,mid,x,y,opt);
    	else if(x>mid)return query(rs(o),mid+1,r,x,y,opt);
    	else{
    		long long resl=query(ls(o),l,mid,x,mid,opt);
    		long long resr=query(rs(o),mid+1,r,mid+1,y,opt);
    		if(opt==1)return resl+resr;
    		else if(opt==2)return min(resl,resr);
    		else return max(resl,resr);
    	}
    }
    
    int main(){
    	n=read(),m=read();
    	for(int i=1;i<=n;i++)a[i]=read();
    	root=build(1,n);
    	while(m--){
    		scanf("%s",op+1);
    		if(op[1]=='a'){
    			int l=read(),r=read(),val=read();
    			modify(root,1,n,l,r,val,2);
    		}else if(op[2]=='e'){
    			int l=read(),r=read(),val=read();
    			modify(root,1,n,l,r,val,1);
    		}else if(op[2]=='u'){
    			int l=read(),r=read();
    			printf("%lld
    ",query(root,1,n,l,r,1));
    		}else if(op[3]=='x'){
    			int l=read(),r=read();
    			printf("%lld
    ",query(root,1,n,l,r,3));
    		}else{
    			int l=read(),r=read();
    			printf("%lld
    ",query(root,1,n,l,r,2));
    		}
    	}
    	return 0;
    } 
    
  • 相关阅读:
    2019年第九周作业
    2019年春第八周作业
    2019年春第七周作业
    2019春第六周作业
    2019年春季学期第五周作业
    2019年春季学期第四周作业
    2019年春季学期第三周作业 ----PTA作业
    2019年春季学期第三周作业
    2019春第七周作业
    第五周作业
  • 原文地址:https://www.cnblogs.com/wzj-xhjbk/p/10557917.html
Copyright © 2020-2023  润新知