https://www.luogu.org/problemnew/show/P3384
已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作:
操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z
操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和
操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z
操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和
存一下自己的板子
#include<bits/stdc++.h>
using namespace std;
#define maxn (100000+100)
#define ll long long
int n,m,root,mod,dfs_clock,op;
vector<int> G[maxn];
int fa[maxn],son[maxn],sz[maxn],deep[maxn],id[maxn],id2[maxn],top[maxn];
ll w[maxn],ver_w[maxn];
int ql,qr,val;
ll sumv[maxn*4],addv[maxn*4];
ll _sum;
int build(int o,int l,int r)
{
int mid=(l+r)/2;
if(l==r)return sumv[o]=addv[o]=w[l]%mod;
return sumv[o]=(build(o*2,l,mid)+build(o*2+1,mid+1,r))%mod;
}
void maintain(int o,int l,int r)
{
if(l==r)sumv[o]=addv[o]%mod;
else sumv[o]=(sumv[o*2]+sumv[o*2+1]+(ll)addv[o]*(r-l+1))%mod;
}
void query(int o,int l,int r,ll add)
{
if(ql<=l&&qr>=r)_sum=(_sum+add*(r-l+1)+sumv[o])%mod;
else
{
int mid=(l+r)/2;
if(ql<=mid)query(o*2,l,mid,(add+addv[o])%mod);
if(qr>mid)query(o*2+1,mid+1,r,(add+addv[o])%mod);
}
}
void update(int o,int l,int r)
{
if(ql<=l&&qr>=r)addv[o]=(addv[o]+val)%mod;
else
{
int mid=(l+r)/2;
if(ql<=mid)update(o*2,l,mid);
if(qr>mid)update(o*2+1,mid+1,r);
}
maintain(o,l,r);
}
void dfs1(int u,int f)
{
deep[u]=deep[f]+1;
sz[u]=1;
fa[u]=f;
int maxx=0;
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(v==f)continue;
dfs1(v,u);
sz[u]+=sz[v];
if(sz[v]>maxx){maxx=sz[v];son[u]=v;}
}
}
void dfs2(int u,int up)
{
id[u]=id2[u]=++dfs_clock;
top[u]=up;
if(son[u]){dfs2(son[u],up);id2[u]=id2[son[u]];}
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(v==fa[u] || v==son[u])continue;
dfs2(v,v);
id2[u]=id2[v];
}
}
void Update1(int u,int v,int z)
{
int tpu=top[u],tpv=top[v];
while(tpu!=tpv)
{
if(deep[tpu]<deep[tpv]){swap(tpu,tpv);swap(u,v);}
ql=id[tpu],qr=id[u],val=z;
update(1,1,n);
u=fa[tpu];
tpu=top[u];
}
if(deep[u]>deep[v])swap(u,v);
ql=id[u],qr=id[v],val=z;
update(1,1,n);
}
ll Query1(int u,int v)
{
_sum=0;
int tpu=top[u],tpv=top[v];
while(tpu!=tpv)
{
if(deep[tpu]<deep[tpv]){swap(tpu,tpv);swap(u,v);}
ql=id[tpu],qr=id[u];
query(1,1,n,0);
u=fa[tpu];
tpu=top[u];
}
if(deep[u]>deep[v])swap(u,v);
ql=id[u],qr=id[v];
query(1,1,n,0);
return _sum;
}
void Update2(int u,int z)
{
ql=id[u],qr=id2[u],val=z;
update(1,1,n);
}
ll Query2(int u)
{
_sum=0;
ql=id[u],qr=id2[u];
query(1,1,n,0);
return _sum;
}
int main()
{
//freopen("input.in","r",stdin);
cin>>n>>m>>root>>mod;
int x,y,z;
for(int i=1;i<=n;i++)scanf("%d",&ver_w[i]);
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
G[x].push_back(y);
G[y].push_back(x);
}
dfs1(root,0);
dfs2(root,root);
for(int i=1;i<=n;i++)w[id[i]]=ver_w[i];
build(1,1,n);
for(int i=1;i<=m;i++)
{
scanf("%d",&op);
switch(op)
{
case 1:
scanf("%d%d%d",&x,&y,&z);
Update1(x,y,z);
break;
case 2:
scanf("%d%d",&x,&y);
printf("%lld\n",Query1(x,y));
break;
case 3:
scanf("%d%d",&x,&z);
Update2(x,z);
break;
case 4:
scanf("%d",&x);
printf("%lld\n",Query2(x));
break;
}
}
return 0;
}