WA 0pts 树链剖分LCA 求调
查看原帖
WA 0pts 树链剖分LCA 求调
779970
lunjiahao楼主2024/2/28 21:44
#include<bits/stdc++.h>
#define int long long
#define ls u<<1
#define rs u<<1|1 
using namespace std;
const int N=1e5+5;
int n,m,t[N<<2],tag[N<<2],a[N],b[N];
int cnt,siz[N],fa[N],dep[N],hs[N],w[N],top[N];
vector <int> g[N];
void dfs1(int u,int fath)
{
	fa[u]=fath;
	siz[u]=1;
	hs[u]=-1;
	dep[u]=dep[fath]+1;
	for(auto v:g[u])
	{
		if(v==fath) continue;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(hs[u]==-1||siz[v]>siz[hs[u]]) hs[u]=v;
	}
}
void dfs2(int u,int t)
{
	top[u]=t;
	b[++cnt]=a[u];
	w[u]=cnt;
	if(hs[u]==-1) return;
	dfs2(hs[u],t);
	for(auto v:g[u])
		if(v!=fa[u]&&hs[u]!=v)
			dfs2(v,v);
}
void push_up(int u)
{
	t[u]=t[ls]+t[rs];
}
void build(int u,int l,int r)
{
	if(l==r)
	{
		t[u]=b[l];
		return;
	}
	int mid=(l+r)>>1;
	build(ls,l,mid);
	build(rs,mid+1,r);
	push_up(u);
}
void push_down(int u,int l,int r)
{
	if(!tag[u]) return;
	tag[ls]+=tag[u];
	tag[rs]+=tag[u];
	int mid=(l+r)>>1;
	t[ls]+=tag[u]*(mid-l+1);
	t[rs]+=tag[u]*(r-mid);
	tag[u]=0;
}
void update(int u,int l,int r,int x,int y,int k)
{
	if(y<l||r<x) return;
	if(x<=l&&r<=y)
	{
		tag[u]+=k;
		t[u]+=k*(r-l+1);
		return;
	}
	push_down(u,l,r);
	int mid=(l+r)>>1;
	if(x<=mid) update(ls,l,mid,x,y,k);
	if(y>mid) update(rs,mid+1,r,x,y,k);
	push_up(u);
}
int query(int u,int l,int r,int x,int y)
{
	if(y<l||r<x) return 0;
	if(x<=l&&r<=y) return t[u];
	push_down(u,l,r);
	int mid=(l+r)>>1,res=0;
	if(x<=mid) res+=query(ls,l,mid,x,y);
	if(y>mid) res+=query(rs,mid+1,r,x,y);
	return res;
}
int modify(int u,int v)
{
	int ans=0;
	while(top[u]^top[v])
	{
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		ans+=query(1,1,n,w[top[u]],w[u]);
		u=fa[top[u]];
	}
	if(u==v) return ans;
	if(dep[u]<dep[v]) swap(u,v);
	ans+=query(1,1,n,w[hs[u]],w[u]);
	return ans;
}
signed main()
{
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%lld%lld",&u,&v);
		g[u].emplace_back(v);
		g[v].emplace_back(u);
	}
	dfs1(1,0);
	dfs2(1,1);
	build(1,1,n);
	while(m--)
	{
		int opt,x,y;
		scanf("%lld%lld",&opt,&x);
		if(opt==1)
		{
			scanf("%lld",&y);
			update(1,1,n,w[x],w[x],y);
		}
		if(opt==2)
		{
			scanf("%lld",&y);
			update(1,1,n,w[x],w[x]+siz[x]-1,y);
		}
		if(opt==3)
			printf("%lld\n",modify(1,x));
	}
	return 0;
}
2024/2/28 21:44
加载中...