线段树求助/kk
  • 板块P2357 守墓人
  • 楼主Baiwhiter
  • 当前回复0
  • 已保存回复0
  • 发布时间2021/3/17 17:25
  • 上次更新2023/11/5 01:57:44
查看原帖
线段树求助/kk
127169
Baiwhiter楼主2021/3/17 17:25

听取wa声一片

随手打了个板子

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
typedef long long ll;
using namespace std;
const int N=200010;
int a[N],n,f;
struct segtree{
	int l,r;
	ll ls;
	ll sum;
}t[N*4];
void pushdown(int p,int lc,int rc){
	if(t[p].ls){
		t[lc].ls=t[p].ls;
		t[lc].sum+=t[p].ls*(t[lc].r-t[lc].l+1);
		t[rc].ls=t[p].ls;
		t[rc].sum+=t[p].ls*(t[rc].r-t[rc].l+1);	
		t[p].ls=0;	
	}
}
void build(int p,int l,int r){
	t[p].l=l;t[p].r=r;
	if(l==r){
	   t[p].sum=a[l];
	   t[p].ls=0;
	   return;
	}
	int mid=l+r>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
void add(int p,int l,int r,int k){
	if(t[p].l>=l && t[p].r<=r){
		t[p].sum+=k*(r-l+1);
		t[p].ls+=k;
		return;
	}
	pushdown(p,p*2,p*2+1);
	int mid=t[p].l+t[p].r>>1;
	if(mid>=l) add(p*2,l,r,k);
	if(mid<r) add(p*2+1,l,r,k);
	t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
ll query(int p,int l,int r){
	if(t[p].l>=l && t[p].r<=r){
		return t[p].sum;
	}
	int mid=t[p].l+t[p].r>>1;
	ll ans=0;
	pushdown(p,p*2,p*2+1);
	if(mid>=l){
		ans+=query(p*2,l,r);
	}
	if(mid<r){
		ans+=query(p*2+1,l,r);
	}
	return ans;
}
int main(){
	scanf("%d%d",&n,&f);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	
	build(1,1,n);
	
	int op;
	while(f--){
		scanf("%d",&op);
		if(op==1){
			int l,r,k;
			scanf("%d%d%d",&l,&r,&k);
			add(1,l,r,k);
		}
		if(op==2){
			int k;
			scanf("%d",&k);
			
			add(1,1,1,k);
		}
		if(op==3){
			int k;
			scanf("%d",&k);
			
			add(1,1,1,-k);
	    }
		if(op==4){
			int l,r;
			scanf("%d%d",&l,&r);
			cout<<query(1,l,r)<<endl;
		}
		if(op==5){
			//cout<<t[1].sum<<endl;
		    cout<<query(1,1,1)<<endl;
		}
	}
	return 0;
}
2021/3/17 17:25
加载中...