Poj 3468 A Simple Problem with Integers

最新推荐文章于 2023-10-02 10:26:46 发布

原创最新推荐文章于 2023-10-02 10:26:46 发布 · 552 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM #大一

Data Structure 专栏收录该内容

8 篇文章

订阅专栏

本文深入探讨了使用懒惰更新思想优化线段树算法的关键点，包括加速求和操作和插入操作的方法。通过增量更新和维护区间总和，作者成功解决了TLE和WA的问题，提供了一个实用的模板。文章详细解释了如何避免朴素算法的不足，并通过多次尝试最终实现了性能优化。

需要lazy思想的线段树

原本都关机睡觉了,来了灵感,双重维护

听说过lazy的想法,实现方式是自己想到的,增量+子区间总和

3*TLE

1*WA

这个可以当做自己的模板了吧~~

TLE的关键点在于

insert和sum的结束条件都不能是T[w].l==T[w].r这样比O(n)的朴素算法还慢..貌似

想要加快sum,需要维护下段总和

想要加快insert,需要维护区间上统一增量

二者缺一不可

在多次实现中和刷牙+思考终于想出...

满足感满满

#include<stdio.h>
#define LL long long
struct line{
	int l,r;
	LL x,s;
	}T[1<<18];
void build(int s,int t,int w){
	T[w].l=s;T[w].r=t;
	T[w].x=0;T[w].s=0;
	if(s==t)return;
	int mid=(s+t)/2;
	build(s,mid,w*2);
	build(mid+1,t,w*2+1);
	}
void insert(int s,int t,int w,LL val){
	if(T[w].l==s&&T[w].r==t){T[w].x+=val;return;}
	T[w].s+=(t+1-s)*val;
	int mid=(T[w].l+T[w].r)/2;
	if(s>mid)insert(s,t,w*2+1,val);
	else if(t<=mid)insert(s,t,w*2,val);
	else {
		insert(s,mid,w*2,val);
		insert(mid+1,t,w*2+1,val);
		}
	}
LL sum(int s,int t,int w){
	LL tp=(t+1-s)*T[w].x; 
	if(T[w].l==s&&T[w].r==t)return tp+T[w].s;
	int mid=(T[w].l+T[w].r)/2;
	if(s>mid)return tp+sum(s,t,w*2+1);
	else if(t<=mid)return tp+sum(s,t,w*2);
	else {
		return tp+sum(s,mid,w*2)+sum(mid+1,t,w*2+1);
		}	
	}
int main(){
	int N,Q;
	while(scanf("%d%d",&N,&Q)!=EOF){
		build(1,N,1);
		int i,j,s,t;
		LL x;
		char c;
		for(i=1;i<=N;i++){
			scanf("%lld%*c",&x);
			insert(i,i,1,x);
			}
		for(i=1;i<=Q;i++){
			scanf("%c",&c);
			if(c=='Q'){
				scanf("%d %d%*c",&s,&t);
				printf("%lld\n",sum(s,t,1)); 
				continue;
				}
			scanf("%d %d %lld%*c",&s,&t,&x);
			insert(s,t,1,x);
			}
		} 
	return 0;
	}