BZOJ2631: tree
Time Limit: 30 Sec
Memory Limit: 128 MB
Submit: 3717
Solved: 1238
题解:
LCT,同时维护乘法和加法,在推乘法标记的时候注意先要更新加法标记
同时维护子树的和要注意必须时时更新不能等着Pushup回来
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define int unsigned int
const int N=300005;
const int MOD=51061;
int n,m,ls[N],rs[N],a[N],rev[N],sum[N],siz[N],mx[N],fa[N],w[N],plu[N],mul[N];
char o[9];
void Pushup(int k)
{
sum[k]=(sum[ls[k]]+sum[rs[k]]+a[k])%MOD;
siz[k]=siz[ls[k]]+siz[rs[k]]+1;
}
void Pudrev(int k)
{
if(!k) return;
swap(ls[k],rs[k]);
rev[k]^=1;
}
void Pudmul(int k,int v)
{
if(!k) return;
a[k]=(a[k]*v)%MOD,sum[k]=(sum[k]*v)%MOD;
mul[k]=(mul[k]*v)%MOD,plu[k]=(plu[k]*v)%MOD;
}
void Pudplu(int k,int v)
{
if(!k) return;
a[k]=(a[k]+v)%MOD,sum[k]=(sum[k]+siz[k]*v)%MOD;
plu[k]=(plu[k]+v)%MOD;
}
void Pushdown(int k)
{
if(rev[k])
{
Pudrev(ls[k]);
Pudrev(rs[k]);
rev[k]=0;
}
if(mul[k]!=1)
{
Pudmul(ls[k],mul[k]);
Pudmul(rs[k],mul[k]);
mul[k]=1;
}
if(plu[k])
{
Pudplu(ls[k],plu[k]);
Pudplu(rs[k],plu[k]);
plu[k]=0;
}
}
bool Check(int k)
{
return ls[fa[k]]!=k&&rs[fa[k]]!=k;
}
void Rotate(int k)
{
int y=fa[k],z=fa[y];
if(!Check(y))
{
if(ls[z]==y) ls[z]=k;
else rs[z]=k;
}
fa[k]=z;
fa[y]=k;
if(k==ls[y])
{
ls[y]=rs[k];
fa[rs[k]]=y;
rs[k]=y;
}
else
{
rs[y]=ls[k];
fa[ls[k]]=y;
ls[k]=y;
}
Pushup(y);
Pushup(k);
}
void Splay(int k)
{
int len=1;
w[1]=k;
for(int i=k;!Check(i);i=fa[i]) w[++len]=fa[i];
for(int i=len;i;i--) Pushdown(w[i]);
while(!Check(k))
{
int y=fa[k],z=fa[y];
if(!Check(y))
{
if(ls[y]==k^ls[z]==y) Rotate(k);
else Rotate(y);
}
Rotate(k);
}
}
void Access(int k)
{
int x=0;
while(k)
{
Splay(k);
rs[k]=x;
Pushup(k);
x=k;
k=fa[k];
}
}
void Evert(int k)
{
Access(k);
Splay(k);
rev[k]^=1;
swap(ls[k],rs[k]);
}
void EAS(int u,int v)
{
Evert(u),Access(v),Splay(v);
}
void Link(int u,int v)
{
Evert(u);
fa[u]=v;
}
void Cut(int u,int v)
{
EAS(u,v);
fa[u]=ls[v]=0;
Pushup(v);
}
int Find(int k)
{
Access(k);
Splay(k);
while(ls[k]) k=ls[k];
return k;
}
main()
{
scanf("%u%u",&n,&m);
int u,v,w,x,y;
for(int i=1;i<=n;i++) a[i]=mul[i]=siz[i]=1;
for(int j=1;j<n;j++)
{
scanf("%u%u",&u,&v);
Link(u,v);
}
while(m--)
{
scanf("%s",o);
if(o[0]=='+')
{
scanf("%u%u%u",&u,&v,&x);
EAS(u,v);
Pudplu(v,x);
Pushup(v);
}
if(o[0]=='*')
{
scanf("%u%u%u",&u,&v,&x);
EAS(u,v);
Pudmul(v,x);
Pushup(v);
}
if(o[0]=='-')
{
scanf("%u%u%u%u",&u,&v,&w,&x);
Cut(u,v),Link(w,x);
}
if(o[0]=='/')
{
scanf("%u%u",&u,&v);
EAS(u,v);
printf("%u\n",sum[v]);
}
}
}
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行,每行描述一个操作
接下来n-1行每行两个正整数u,v,描述这棵树
接下来q行,每行描述一个操作
Output
对于每个/对应的答案输出一行
Sample Input
3 2
1 2
2 3
* 1 3 4
/ 1 1
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