zoukankan      html  css  js  c++  java
  • hihoCoder#1080 (线段树)

    题目大意:线段树的区间更改与查询,但是涉及到两种区间修改方式,一是给区间中的数全部加上一个数,二是将一个区间全部置为同一个数,然后询问整个区间和。

    题目分析:处理好set操作和add操作的先后顺序就OK了。

    代码如下:

    # include<iostream>
    # include<cstdio>
    # include<cstring>
    # include<algorithm>
    using namespace std;
    
    const int N=100001;
    
    int n,m;
    int tr[N*4+5];
    int lazy_set[N*4+5];
    int lazy_add[N*4+5];
    
    void pushDown(int rt,int l,int r)
    {
    	int mid=l+(r-l)/2;
    	if(lazy_set[rt]!=0){
    		lazy_set[rt<<1]=lazy_set[rt<<1|1]=lazy_set[rt];
    		lazy_add[rt<<1]=lazy_add[rt<<1|1]=0;
    		tr[rt<<1]=lazy_set[rt]*(mid-l+1);
    		tr[rt<<1|1]=lazy_set[rt]*(r-mid);
    		lazy_set[rt]=0;
    	}
    	if(lazy_add[rt]!=0){
    		lazy_add[rt<<1]+=lazy_add[rt];
    		lazy_add[rt<<1|1]+=lazy_add[rt];
    		tr[rt<<1]+=lazy_add[rt]*(mid-l+1);
    		tr[rt<<1|1]+=lazy_add[rt]*(r-mid);
    		lazy_add[rt]=0;
    	}
    }
    
    void makeTree(int rt,int l,int r)
    {
    	if(l==r){
    		scanf("%d",&tr[rt]);
    	}else{
    		int mid=l+(r-l)/2;
    		makeTree(rt<<1,l,mid);
    		makeTree(rt<<1|1,mid+1,r);
    		tr[rt]=tr[rt<<1]+tr[rt<<1|1];
    	}
    }
    
    void update(int rt,int l,int r,int L,int R,int x,int flag)
    {
    	if(L<=l&&r<=R){
    		if(flag){
    			lazy_set[rt]=x;
    			lazy_add[rt]=0;
    			tr[rt]=x*(r-l+1);
    		}else{
    			lazy_add[rt]+=x;
    			tr[rt]+=x*(r-l+1);
    		}
    	}else{
    		pushDown(rt,l,r);
    		int mid=l+(r-l)/2;
    		if(L<=mid)
    			update(rt<<1,l,mid,L,R,x,flag);
    		if(R>mid)
    			update(rt<<1|1,mid+1,r,L,R,x,flag);
    		tr[rt]=tr[rt<<1]+tr[rt<<1|1];
    	}
    }
    
    int main()
    {
    	while(~scanf("%d%d",&n,&m))
    	{
    		memset(lazy_set,0,sizeof(lazy_set));
    		memset(lazy_add,0,sizeof(lazy_add));
    		makeTree(1,0,n);
    		int flag,l,r,v;
    		while(m--)
    		{
    			scanf("%d%d%d%d",&flag,&l,&r,&v);
    			update(1,0,n,l,r,v,flag);
    			printf("%d
    ",tr[1]);
    		}
    	}
    	return 0;
    }
    

      

  • 相关阅读:
    参考文献
    Redis安装以及常见错误
    Linux下搭建Mysql主从遇到的问题
    Linux中创建虚拟环境安装问题
    python 二分查找算法
    python 递归函数
    python 内置函数
    python 装饰器
    初识正则表达式
    内置函数***
  • 原文地址:https://www.cnblogs.com/20143605--pcx/p/5497359.html
Copyright © 2011-2022 走看看