• POJ3468 线段树求和(线段树模板2)


    原题链接:POJ3468

    解析:本题也是线段树的一种用法,算得上裸题,我用这题来加深对线段树的理解,以及熟悉线段树的模板。郭炜老师的代码是演示指针写法,但我更喜欢用数组来表示,所以改成了我喜欢的模板。

    PS:刚刚刚刚刚才知道,这种方法是延迟标记,也就是说当对一端区间的叶子结点进行相同操作的时候,不要急着全部完成,不然递归等操作需要很多,我们可以记录对这个区间将要做什么操作,等查询的时候将所有对此区间的操作一次做完,这样就省的一次一次遍历了。举个例子,就比如对a进行+1操作,进行一千亿亿次,结果和+一千亿亿是一样的,然而效率却天差地别。

    失误点:Query函数里的第一行,tree[root].l == s && tree[root].r == e,第一次将tree[root].r 写成tree[root].l,导致答案错误。

    代码示例:

    #include<iostream>
    #include<cstdio>
    using namespace std;
    long long ans;
    struct Node{
    	int l,r;
    	long long Inc;
    	long long sum;
    	int Mid(){
    		return (l+r)/2;
    	}
    }tree[400050];
    void BuildTree(int root,int l,int r){
    	tree[root].l = l;
    	tree[root].r = r;
    	tree[root].sum = 0;
    	tree[root].Inc = 0;
    	if(l == r)	return;
    	int mid = (l+r)/2;
    	BuildTree(2*root+1,l,mid);
    	BuildTree(2*root+2,mid+1,r);
    }
    void Insert(int root,int i,int v){
    	if(tree[root].l == i && tree[root].r == i){
    		tree[root].sum = v;
    		return;
    	}
    	tree[root].sum += v;
    	if(i <= tree[root].Mid())	Insert(2*root+1,i,v);
    	else Insert(2*root+2,i,v);
    }
    void Add(int root,int s,int e,long long c){
    	if(tree[root].l == s && tree[root].r == e){
    		tree[root].Inc += c;
    		return;
    	}
    	tree[root].sum += c*(e-s+1);
    	if(e <= tree[root].Mid())	Add(2*root+1,s,e,c);
    	else if(s > tree[root].Mid())	Add(2*root+2,s,e,c);
    	else{
    		Add(2*root+1,s,tree[root].Mid(),c);
    		Add(2*root+2,tree[root].Mid()+1,e,c);
    	}
    }
    long long Query(int root,int s,int e){
    	if(tree[root].l == s && tree[root].r == e){
    		return tree[root].sum + tree[root].Inc*(e-s+1);
    	}
    	tree[root].sum += tree[root].Inc*(tree[root].r - tree[root].l+1);
    	Add(2*root+1,tree[root].l,tree[root].Mid(),tree[root].Inc);
    	Add(2*root+2,tree[root].Mid()+1,tree[root].r,tree[root].Inc);
    	tree[root].Inc = 0;
    	if(e <= tree[root].Mid())	return Query(2*root+1,s,e);
    	else if(s > tree[root].Mid())	return Query(2*root+2,s,e);
    	else{
    		return Query(2*root+1,s,tree[root].Mid()) +
    				Query(2*root+2,tree[root].Mid()+1,e); 
    	}
    }
    int main()
    {
    	int n,q,h;
    	scanf("%d%d",&n,&q);
    	BuildTree(0,1,n);
    	for(int i = 1;i <= n;i++){
    		scanf("%d",&h);
    		Insert(0,i,h);
    	}
    	char c[2];
    	for(int i = 0;i < q;i++){
    		scanf("%s",c);
    		if(c[0] == 'Q'){
    			int s,e;
    			scanf("%d%d",&s,&e);
    			printf("%lld
    ",Query(0,s,e));
    		}else{
    			int s,e,c;
    			scanf("%d%d%d",&s,&e,&c);
    			Add(0,s,e,c);	
    		}
    	}
    	return 0;
    }
  • 相关阅读:
    【C++】基础及引用
    gradle打包分编译环境
    gradle
    MediaPlayer滑动不准的问题
    python初步入门
    音频播放服务
    《深入理解Android2》读书笔记(二)
    缓存(LruCache)机制
    handler机制
    监听网络状态
  • 原文地址:https://www.cnblogs.com/long98/p/10352197.html
Copyright © 2020-2023  润新知