Description
一棵n个点的树,每个点的初始权值为1。对于这棵树有q个操作,每个操作为以下四种操作之一:
+ u v c:将u到v的路径上的点的权值都加上自然数c;
– u1 v1 u2 v2:将树中原有的边(u1,v1)删除,加入一条新边(u2,v2),保证操作完之后仍然是一棵树;
* u v c:将u到v的路径上的点的权值都乘上自然数c;
/ u v:询问u到v的路径上的点的权值和,求出答案对于51061的余数。
Input
第一行两个整数n,q
接下来n-1行每行两个正整数u,v,描述这棵树
接下来q行,每行描述一个操作
Output
对于每个/对应的答案输出一行
Sample Input
3 2
1 2
2 3
* 1 3 4
/ 1 1
Sample Output
4
HINT
数据规模和约定
10%的数据保证,1<=n,q<=2000
另外15%的数据保证,1<=n,q<=5*10^4,没有-操作,并且初始树为一条链
另外35%的数据保证,1<=n,q<=5*10^4,没有-操作
100%的数据保证,1<=n,q<=10^5,0<=c<=10^4
题解
lct
代码
#include<bits/stdc++.h>
#define ll long long
#define inf 1000000007
#define mod 51061
#define lint unsigned int
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,m,top,cnt;
int c[100005][2],fa[100005];
int size[100005],q[100005];
bool rev[100005];
lint sum[100005],val[100005],at[100005],mt[100005];
inline bool isroot(int x){return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;}
inline void update(int x)
{
int l=c[x][0],r=c[x][1];
size[x]=(size[l]+size[r]+1)%mod;
sum[x]=(sum[l]+sum[r]+val[x])%mod;
}
inline void cal(int x,int m,int a)
{
if (!x) return;
val[x]=(val[x]*m+a)%mod;
sum[x]=(sum[x]*m+a*size[x])%mod;
mt[x]=(mt[x]*m)%mod;
at[x]=(at[x]*m+a)%mod;
}
inline void pushdown(int x)
{
int l=c[x][0],r=c[x][1];
if (rev[x])
{
rev[x]^=1;rev[l]^=1;rev[r]^=1;
swap(c[x][0],c[x][1]);
}
int m=mt[x],a=at[x];
mt[x]=1;at[x]=0;
if (m!=1||a!=0)
{
cal(l,m,a);cal(r,m,a);
}
}
inline void rotate(int x)
{
int l,r,y=fa[x],z=fa[y];
if (c[y][0]==x) l=0;else l=1;r=l^1;
if (!isroot(y))
{
if (c[z][0]==y) c[z][0]=x;else c[z][1]=x;
}
fa[x]=z;fa[y]=x;
c[y][l]=c[x][r];fa[c[x][r]]=y;
c[x][r]=y;
update(y);update(x);
}
void splay(int x)
{
int top=0;
q[++top]=x;
for (int i=x;!isroot(i);i=fa[i])
q[++top]=fa[i];
while (top) pushdown(q[top--]);
while (!isroot(x))
{
int y=fa[x],z=fa[y];
if (!isroot(y))
{
if (c[y][0]==x^c[z][0]==y) rotate(x);else rotate(y);
}
rotate(x);
}
}
void access(int x)
{
int t=0;
while (x)
{
splay(x);
c[x][1]=t;
update(x);
t=x;
x=fa[x];
}
}
void makeroot(int x)
{
access(x);splay(x);rev[x]^=1;
}
void split(int u,int v)
{
makeroot(v);access(u);splay(u);
}
void link(int u,int v)
{
makeroot(u);fa[u]=v;
}
void cut(int u,int v)
{
makeroot(u);access(v);splay(v);
c[v][0]=fa[u]=0;
}
int main()
{
n=read();int Q=read();
for(int i=1;i<=n;i++)
val[i]=sum[i]=mt[i]=size[i]=1;
for (int i=1;i<n;i++)
{
int u=read(),v=read();
link(u,v);
}
char ch[10];
while (Q--)
{
scanf("%s",ch);
int u=read(),v=read();
if (ch[0]=='+')
{
int c=read();
split(u,v);cal(u,1,c);
}
else if (ch[0]=='-')
{
cut(u,v);
u=read(),v=read();
link(u,v);
}
else if (ch[0]=='*')
{
int c=read();
split(u,v);cal(u,c,0);
}
else
{
split(u,v);
printf("%d\n",sum[u]);
}
}
return 0;
}