深入浅出(进阶版)是不是线段树过不了
查看原帖
深入浅出(进阶版)是不是线段树过不了
1037997
_TGWQ12_楼主2024/1/3 21:49
#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
const int N=1e5+5;
int n=10,m,mn;
ll a[N],w[N*2],tag[N*2];
bool xij(int l,int r,int L,int R)
{
	return (l>R)||(r<L);
} 
bool bah(int l,int r,int L,int R)
{
	return L<=l&&r<=R; 
}
void up(int v)
{
	w[v]=w[v*2]+w[v*2+1];
}
void make(int v,int len,ll wi)
{
	tag[v]+=wi;
	w[v]+=len*wi;
}
void down(int v,int l,int r)
{
	int M=(l+r)>>1;
	make(v*2,M-l+1,tag[v]);
	make(v*2+1,r-M,tag[v]);
	 tag[v]=0;
}
void zao(int v,int l,int r)
{
	if(l==r)
	{
		w[v]=a[l];
		return ;
	}
	zao(v*2,l,(l+r)>>1); 
	zao(v*2+1,(l+r)/2+1,r);
	up(v);
	mn=max(mn,v);
}
void jia(int v,int l,int r,int L,int R,ll wi)
{
	if(v>=n*2) return ;
	if(bah(l,r,L,R))
		make(v,r-l+1,wi);
	if(!xij(l,r,L,R))
	{
		int M=(l+r)>>1;
		down(v,l,r);
		jia(v*2,l,M,L,R,wi);
		jia(v*2+1,M+1,r,L,R,wi);
		up(v);
	}
}
ll he(int v,int l,int r,int L , int R)
{
	if(bah(l,r,L,R))
		return w[v];
	if(!xij(l,r,L,R))
	{
		int M=(l+r)>>1;
		down(v,l,r);
		return he(v*2,l,M,L,R)+he(v*2+1,M+1,r,L,R); 
	}
	return 0;
}
void dfs(int v)
{
	if(v>mn) return  ;
	cout<<w[v]<<endl;
	dfs(v*2);
	dfs(v*2+1);
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	zao(1,1,n);
//	dfs(1); 
	int t,L,R,wi;
	while(m--)
	{
		cin>>t;
		if(t==1)
		{
			cin>>L>>R>>wi;
			jia(1,1,n,L,R,wi);
		}
		else if(t==2)
		{
			cin>>L>>R;
			cout<<he(1,1,n,L,R)<<endl; 
		}
	}
	return 0;
} 

本人一蒟蒻不知是太蒟了,还是这本就有问题。我已经改的和书上的没太大区别了

2024/1/3 21:49
加载中...