题目描述
如题,已知一棵包含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为根节点的子树内所有节点值之和
输入输出格式
输入格式:
第一行包含4个正整数N、M、R、P,分别表示树的结点个数、操作个数、根节点序号和取模数(即所有的输出结果均对此取模)。
接下来一行包含N个非负整数,分别依次表示各个节点上初始的数值。
接下来N-1行每行包含两个整数x、y,表示点x和点y之间连有一条边(保证无环且连通)
接下来M行每行包含若干个正整数,每行表示一个操作,格式如下:
操作1: 1 x y z
操作2: 2 x y
操作3: 3 x z
操作4: 4 x
输出格式:
输出包含若干行,分别依次表示每个操作2或操作4所得的结果(对P取模)
输入输出样例
输入样例#1:
5 5 2 24
7 3 7 8 0
1 2
1 5
3 1
4 1
3 4 2
3 2 2
4 5
1 5 1 3
2 1 3
输出样例#1:
2
21
为了解决其中一个问题,于是我重拾了树链剖分。
纯手打
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<set>
#define ll long long
#define maxx 100005
#define INF 0x7f7f7f7f
using namespace std;
int n,m,r,mod;
int head[maxx],to[maxx<<1],_next[maxx<<1];
int edge;
void addEdge(int x,int y)
{
to[++edge]=y,_next[edge]=head[x],head[x]=edge;
to[++edge]=x,_next[edge]=head[y],head[y]=edge;
}
//建树
int deep[maxx];
int fa[maxx];
int son[maxx];
int tot[maxx];
int dfs1(int u,int f,int d)
{
deep[u]=d;
tot[u]=1;
int maxson=-1;
fa[u]=f;
for(int i=head[u];i;i=_next[i])
{
int v=to[i];
if(v==f)continue;
tot[u]+=dfs1(v,u,d+1);
if(tot[v]>maxson)
{
maxson=tot[v];
son[u]=v;
}
}
return tot[u];
}//处理重儿子
int id[maxx],cnt;
int a[maxx];//reflected
int b[maxx];//original
int top[maxx];
void dfs2(int u,int t)
{
id[u]=++cnt;
a[cnt]=b[u];
top[u]=t;
if(!son[u])return;
dfs2(son[u],t);
for(int i=head[u];i;i=_next[i])
{
int v=to[i];
if(!id[v])dfs2(v,v);
}
}//end...
struct Tree
{
int l,r,w,siz,f;
}T[maxx*4];
void update(int s)
{
T[s].w=(T[s<<1].w+T[s<<1|1].w)%mod;
}
void build(int s,int l,int r)
{
T[s].l=l;T[s].r=r;T[s].siz=r-l+1;
if(l==r)
{
T[s].w=a[l];
return;
}
int mid=(l+r)>>1;
build(s<<1,l,mid);
build(s<<1|1,mid+1,r);
update(s);
}
void pushDown(int s)
{
if(!T[s].f)return;
T[s<<1].w=(T[s<<1].w+T[s<<1].siz*T[s].f)%mod;
T[s<<1|1].w=(T[s<<1|1].w+T[s<<1|1].siz*T[s].f)%mod;
T[s<<1].f=(T[s<<1].f+T[s].f)%mod;
T[s<<1|1].f=(T[s<<1|1].f+T[s].f)%mod;
T[s].f=0;
}
void change(int s,int l,int r,int val)
{
if(l<=T[s].l&&T[s].r<=r)
{
T[s].w+=(ll)T[s].siz*val%mod;
T[s].w%=mod;
T[s].f=(T[s].f+val)%mod;
return;
}
pushDown(s);
int mid=(T[s].l+T[s].r)>>1;
if(l<=mid)change(s<<1,l,r,val);
if(r>mid)change(s<<1|1,l,r,val);
update(s);
}
int ask(int s,int l,int r)
{
if(l<=T[s].l&&T[s].r<=r)
return T[s].w;
int ans=0;
pushDown(s);
int mid=(T[s].l+T[s].r)>>1;
if(l<=mid)ans=(ans+ask(s<<1,l,r))%mod;
if(r>mid)ans=(ans+ask(s<<1|1,l,r))%mod;
return ans;
}//end of segment tree
int ask(int x,int y)
{
int ans=0;
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
ans=(ans+ask(1,id[top[x]],id[x]))%mod;
x=fa[top[x]];
}
if(deep[x]>deep[y])swap(x,y);
ans=(ans+ask(1,id[x],id[y]))%mod;
return ans;
}
void change(int x,int y,int val)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
change(1,id[top[x]],id[x],val);
x=fa[top[x]];
}
if(deep[x]>deep[y])swap(x,y);
//cout<<"hah"<<endl;
change(1,id[x],id[y],val);
}
int main()
{
cin>>n>>m>>r>>mod;
for(int i=1;i<=n;i++)
scanf("%d",b+i);
int x,y,z;
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
addEdge(x,y);
}
dfs1(r,0,1);
dfs2(r,r);
build(1,1,n);
int c;
while(m--)
{
scanf("%d",&c);
if(c==1)
{
scanf("%d%d%d",&x,&y,&z);
change(x,y,z);
}
else if(c==2)
{
scanf("%d%d",&x,&y);
printf("%d\n",ask(x,y));
}
else if(c==3)
{
scanf("%d%d",&x,&z);
change(1,id[x],id[x]+tot[x]-1,z);
}
else
{
scanf("%d",&x);
printf("%d\n",ask(1,id[x],id[x]+tot[x]-1));
}
}
return 0;
}```