【bzoj2631】tree link-cut-tree

2016-06-01 08:50:36

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=2631

注意加和乘的标记下传问题。

还有就是split后，要分清x和y哪个是祖先。

pushup在access和rotate后都要进行。

这题还卡常数，开ll就会T，开unsigned int即可。

#include<bits/stdc++.h>
#define inf 1000000000
#define uint unsigned int
#define N 100005
#define mod 51061
using namespace std;
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,q;
char ch[5];
namespace LCT{
  int ch[N][2],fa[N],sz[N];bool rev[N];
  uint cj[N],add[N],val[N],sum[N];
  inline bool isroot(int x){
    if(ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x)return 1;return 0;
  }
  void change(int x,int c,int a){
    if(!x)return;
    val[x]=(val[x]*c+a)%mod;
    sum[x]=(sum[x]*c+a*sz[x])%mod;
    add[x]=(add[x]*c+a)%mod;
    cj[x]=(cj[x]*c)%mod;
  }
  void pushdown(int x){
    if(rev[x]){
      rev[ch[x][0]]^=1;rev[ch[x][1]]^=1;rev[x]^=1;
      swap(ch[x][0],ch[x][1]);
    }
    if(cj[x]!=1||add[x]!=0){
      change(ch[x][0],cj[x],add[x]);
      change(ch[x][1],cj[x],add[x]);
    }
    cj[x]=1;add[x]=0;
  }
  void Pushdown(int x){
    if(!isroot(x))Pushdown(fa[x]);
    pushdown(x);
  }
  void pushup(int x){
    sum[x]=(sum[ch[x][0]]+sum[ch[x][1]]+val[x])%mod;
    sz[x]=(sz[ch[x][0]]+sz[ch[x][1]]+1)%mod;
  }
  void rotate(int x){
    int y=fa[x],k=ch[y][1]==x;
    if(!isroot(y))ch[fa[y]][ch[fa[y]][1]==y]=x;
    fa[x]=fa[y];
    fa[ch[x][!k]]=y;
    fa[y]=x;
    ch[y][k]=ch[x][!k];
    ch[x][!k]=y;
    pushup(y);pushup(x);
  }
  void splay(int x){
    Pushdown(x);
    for(int y=fa[x];!isroot(x);y=fa[x]){
      if(!isroot(y)){
        if((ch[fa[y]][ch[fa[y]][1]==y])!=(ch[y][1]==x))rotate(x);
        else rotate(y);
      }rotate(x);
    }
  }
  inline void access(int x){
    int y=0;
    while(x){
      splay(x);
      ch[x][1]=y;pushup(x);
      x=fa[y=x];
    }
  }
  inline void moveroot(int x){
    access(x);splay(x);rev[x]^=1;
  }
  inline void link(int x,int y){
    moveroot(x);fa[x]=y;splay(x);
  }
  inline void cut(int x,int y){
    moveroot(x);access(y);splay(y);ch[y][0]=fa[x]=0;
  }
  inline void split(int x,int y){
    moveroot(y);access(x);splay(x);
  }
}
int main(){
  n=read();q=read();
  for(int i=1;i<=n;i++)LCT::val[i]=LCT::sum[i]=LCT::cj[i]=LCT::sz[i]=1;
  for(int i=1;i<n;i++)LCT::link(read(),read());
  while(q--){
    scanf("%s",ch);
    int u=read(),v=read();
    if(ch[0]=='+')LCT::split(u,v),LCT::change(u,1,read());
    else if(ch[0]=='-'){
      LCT::cut(u,v);u=read();v=read();LCT::link(u,v);
    }
    else if(ch[0]=='*')LCT::split(u,v),LCT::change(u,read(),0);
    else LCT::split(u,v),printf("%d\n",LCT::sum[u]%mod);
  }
  return 0;
}

2631: tree

Time Limit: 30 Sec  Memory Limit: 128 MB
Submit: 3385  Solved: 1137
[Submit][Status][Discuss]

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