树链剖分(从入门到入土。)

前置知识:线段树,链式前向星,LCA,DFS序

好了就当你都会了。不会也没关系。

树链剖分通常的操作:
1.x -> y 的路径上修改 2.x -> y 的路径上查询 3. 对于 x 的子树修改 4.对于 x 的子树查询。

一般还有换根操作。树剖也也可以做LCA以及差分操作(但是树剖好像就直接修改了不需要差分)。

树链剖分有两个DFS 这两个DFS就是把一棵树变成一个序列。 然后就可以用数据结构来维护了。

第一个DFS 用来求 \(fa\)(祖先节点) \(size\)(子树大小)\(son\)(重儿子) \(d\)(深度)
重儿子指的是\(size\)较大的儿子节点。
第二个DFS 用来求\(top\)(这条链上最顶端的点) \(id\)(编号)以及其他的赋值操作。 但是重儿子要先DFS。

const int N = 1e5 + 10 ;
struct node { int v , nxt ; } ;
node e[N << 1] ;
int head[N] , cnt = 0 ;
inline void Add(int u , int v) { e[++ cnt].v = v ; e[cnt].nxt = head[u] ; head[u] = cnt ; }
int size[N] , son[N] , d[N] ;
inline void Dfs1(int u) { size[u] = 1 ;
  for(register int i = head[u] ; i ; i = e[i].nxt) {
    int v = e[i].v ; if(v == fa[u]) continue ; // 防止无限递归。
    d[v] = d[u] + 1 , fa[v] = u ,  // 记录深度 以及父亲节点
Dfs1(v) ; size[u] += size[v] ; //算出子树大小
    if(size[v] > size[son[u]]) son[u] = v ; //得出u 的 重儿子是 son[u]
  }
}
int top[N] , id[N] , tot = 0 ;
inline void Dfs2(int u , int tp) { top[u] = tp , id[u] = ++ tot ;
  if(! son[u]) return ; Dfs2(son[u] , tp) ;
  for(register int i = head[u] ; i ; i = e[i].nxt) { int v = e[i].v ;
    if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
  }
}

这段代码求的就是下图的结果。
树链剖分(从入门到入土。)_第1张图片
对于这张图 \(top\) 指的是重链上最靠上的位置。即链的最高处。
\(Question:\) 如果对于最右下的那个节点 top 是自己 那么不就死循环了吗 ?
请读者仔细思考这个问题。

假装你会线段树了。
那么我们想 怎么做树上的修改呢?
\(id\) 把这个树变成一个序列。
没错 就是跳链对于每条重链 你可以一下到顶端 那么你直接修改这条重链上的值就可以了(数据结构维护)
万一 一次重链跳不到那里怎么办?
那么说明了这个点并不是重儿子 我们想办法让它跳上去。
反复循环。 DFS1中记录了父亲节点 你将它变成它重链上的最高处的父亲节点。
因为这条重链修改过了 所以不用管重复。一直反复跳 它会跳到 链的 LCA 处 这也是为什么可以求LCA的原因
对于上面留下的问题…也是一个关于复杂度的证明
假设最坏情况 你的修改/查询的点 没有一条边在重链上 你每次都要 往上 fa 一次
但是这种情况 出现在 满多叉树上 这样的话 就是最坏的复杂度 即 \(logn\)
那么一般还要套一个线段树 所以 最坏的复杂度是 \(n log^2 n\)

写在最后:如果没有规定根的话 建议根 随机呢 万一弄个满二叉树卡你 你随机一下就可以避免被卡。
rand()%n+1
善意提醒:建议您下面的代码不要复制粘贴呢 自己一遍遍手打。。

【模板】树链剖分

#include 
using namespace std ;
inline int read() { register int res = 0 ; register char c ;
#define gc c = getchar()
    while(isspace(gc)) ;
    while(res = (res << 1) + (res << 3) + (c & 15) , isdigit(gc)) ;
    return res ;
}
int n , m , r , p ;
const int N = 1000000 + 5 ;
struct E{ int v ;int nxt ; } ;
E edge[N << 1] ;
int a[N] ; int fa[N] ; int w[N] ; int id[N] ; int son[N] ;
int cnt = 0 ; int head[N] ; int Add[N << 2] , laz[N << 2] ;
int dep[N] , siz[N] , t[N] ;
inline void Add_Edge(register int u , register int v) { edge[++ cnt].v = v ; edge[cnt].nxt = head[u] ; head[u] = cnt ; return ; }//建边。
#define l(x) x << 1
#define r(x) x << 1 | 1
inline void Push_down(register int x , register int len) {
    laz[l(x)] += laz[x] ; laz[r(x)] += laz[x] ;
    Add[l(x)] += laz[x] * (len - (len >> 1)) ; Add[r(x)] += laz[x] * (len >> 1) ;
    Add[l(x)] %= p ; Add[r(x)] %= p ;
    laz[x] = 0 ; return ;
}
inline void Build(register int l , register int r , register int rt) {//建树
    if(l == r) { Add[rt] = a[l] ;  return ; }
    register int mid = l + r >> 1 ;
    Build(l , mid , l(rt)) ; Build(mid + 1 , r , r(rt)) ;
    Add[rt] = (Add[l(rt)] + Add[r(rt)]) % p ;
}
inline void Update(register int a , register int b , register int l  , register int r , register int rt , register int k) {//正常的线段树操作
    if(a <= l and r <= b) { laz[rt] += k ; Add[rt] += k * (r - l + 1) ; }
    else {
        if(laz[rt]) Push_down(rt , r - l + 1) ;
        register int mid = l + r >> 1 ;
        if(a <= mid) Update(a , b , l , mid , l(rt) , k) ;
        if(b > mid) Update(a , b , mid + 1 , r , r(rt) , k) ;
        Add[rt] = (Add[l(rt)] + Add[r(rt)]) % p ;
    }
}
int res = 0 ;
inline void query(register int a , register int b , register int  l , register int r , register int rt) {
    if(a <= l and r <= b) { res = (res + Add[rt]) % p ; return ; }
    else {
        if(laz[rt]) Push_down(rt , r - l + 1) ;
        register int mid = l + r >> 1 ;
        if(a <= mid) query(a , b , l , mid , l(rt)) ;
        if(b > mid) query(a , b , mid + 1 , r , r(rt)) ;
    }
}
inline int Query(register int a , register int b , register int l , register int r , register int rt) {//正常的线段树操作
    res = 0 ; query(a , b , l , r , rt) ;
    return res % p ;
}
inline void Upd_Range(register int x , register int y , register int k) {//链上修改
    while(t[x] != t[y]) {
        if(dep[t[x]] < dep[t[y]]) swap(x , y) ;
         Update(id[t[x]] , id[x] , 1 , n , 1 , k) ;
         x = fa[t[x]] ;
    }
    if(dep[x] > dep[y]) swap(x , y) ;
    Update(id[x] , id[y] , 1 , n , 1 , k) ;
}
inline int Query_Range(register int x , register int y) {//链上查询
    int ans = 0 ;
    while(t[x] != t[y]) {
        if(dep[t[x]] < dep[t[y]]) swap(x , y) ;
        ans += Query(id[t[x]] , id[x] , 1 , n , 1) ;
        x = fa[t[x]] ;
    }
    if(dep[x] > dep[y]) swap(x , y) ;
    ans += Query(id[x] , id[y] , 1 , n , 1) ;
    return ans % p ;
}
inline int Qson(register int x) { return Query(id[x] , id[x] + siz[x] - 1 , 1 , n , 1) ; } // 子树查询。
inline void Updson(register int x , register int k) { Update(id[x] , id[x] + siz[x] - 1 , 1 , n , 1 , k) ; return ; }
inline void Dfs1(register int x , register int f , register int deep) {
    dep[x] = deep ; fa[x] = f ; siz[x] = 1 ;
    int max_son = -1 ;
    for(register int i = head[x] ; i ; i = edge[i].nxt) {
        register int v = edge[i].v ;
        if(v == f) continue ;
        Dfs1(v , x , deep + 1) ;siz[x] += siz[v] ;
        if(siz[v] > max_son) max_son = siz[v] , son[x] = v ;
    }
}
int tot = 0 ;
inline void Dfs2(register int x , register int tf) {
    id[x] = ++ tot ; a[tot] = w[x] ; t[x] = tf ; // a 数组重新赋值
    if(! son[x]) return ;
    Dfs2(son[x] , tf) ;
    for(register int i = head[x] ; i ; i = edge[i].nxt) {
        int v = edge[i].v ;
        if(v == fa[x] or v == son[x]) continue ;
        Dfs2(v , v) ;
    }
}
signed main() {
    n = read() ; m = read() ; r = read() ; p = read() ;
    for(register int i = 1 ; i <= n ; i ++) w[i] = read() ;
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v = read() ;
        Add_Edge(u , v) ;
        Add_Edge(v , u) ;
    }
    Dfs1(r , 0 , 1) ; Dfs2(r , r) ; Build(1 , n , 1) ;
    for( ; m -- ; ) {
        register int opt = read() ;
        if(opt == 1) {
            register int x = read() , y = read() , k = read() ;
            Upd_Range(x , y , k % p) ;
        }
        if(opt == 2) {
            register int x = read() , y = read() ;
            printf("%d\n" , Query_Range(x , y)) ;
        }
        if(opt == 3) {
            register int x = read() , y = read() ;
            Updson(x , y) ;
        }
        if(opt == 4) { printf("%d\n" , Qson(read())) ; }
    }
    return 0 ;
}

[USACO11DEC]牧草种植Grass Planting

链上修改 单点查询。
对于这个题目 要修改的是边的权值。
这里有一个技巧 是 把边的权值下传给点 然后把最上面的点忽略掉。(易证 这样是可以的。)
然后进行基本的操作就可以了。

#include
using namespace std ;
#define int long long
#define fi first
#define se second
#define pb push_back
inline int read() {
    register int x = 0 , f = 1 ;
    register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
}
template < typename T > inline bool cmax(T & x , T y) {
    return x < y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cmin(T & x , T y) {
    return x > y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cabs(T & x) {
    return x > 0 ? 1 : (x = - x) , 0 ;
}
inline int QP(int x , int y , int Mod) {
    int ans = 1 ;
    for( ; y ; y >>= 1 , x = (x * x) % Mod)
        if(y & 1) ans = (ans * x) % Mod ;
    return ans ;
}
int n , m ;
struct node {
  int v , nxt ;
} ;
const int N = 1e5 + 10 ;
node e[N << 1] ;
int head[N] , cnt = 0 ;
inline void Add(int u , int v) {
  e[++ cnt].v = v ;
  e[cnt].nxt = head[u] ;
  head[u] = cnt ;
  return ;
}
int fa[N] ;
int top[N] ;
int d[N] ;
int size[N] ;
int son[N] ;
int id[N] , idx = 0 ;
int a[N] ;
int sum[N << 2] ;
inline void build(int l , int r , int rt) {
  if(l == r) {
    sum[rt] = a[l] ;
    return ;
  }
  int mid = l + r >> 1 ;
  build(l , mid , rt << 1) ;
  build(mid + 1 , r , rt << 1 | 1) ;
}
int tag[N << 2] ;
inline void Push_down(int rt , int l , int r) {
  if(! tag[rt]) return ;
  tag[rt << 1] += tag[rt] ;
  tag[rt << 1 | 1] += tag[rt] ;
  int mid = l + r >> 1 ;
  sum[rt << 1] += tag[rt] * (mid - l + 1) ;
  sum[rt << 1 | 1] += tag[rt] * (r - mid) ;
  tag[rt] = 0 ;
  return ;
}
inline void Update(int a , int b , int l , int r , int rt) {
  if(a <= l && r <= b) { sum[rt] += (r - l + 1) ; tag[rt] ++ ; return ; }
  Push_down(rt , l , r) ;
  int mid = l + r >> 1 ;
  if(a <= mid) Update(a , b , l , mid , rt << 1) ;
  if(b > mid) Update(a , b , mid + 1 , r , rt << 1 | 1) ;
  sum[rt] = sum[rt << 1] + sum[rt << 1 | 1] ;
}
inline int Query(int a , int b , int l , int r , int rt) {
  if(a <= l && r <= b) return sum[rt] ;
  Push_down(rt , l , r) ;
  int mid = l + r >> 1 , ans = 0 ;
  if(a <= mid) ans += Query(a , b , l , mid , rt << 1) ;
  if(b > mid) ans += Query(a , b , mid + 1 , r , rt << 1 | 1) ;
  return ans ;
}
inline void Dfs(int u) {
  size[u] = 1 ;
  for(register int i = head[u] ; i ; i = e[i].nxt) {
    int v = e[i].v ;
    if(v == fa[u]) continue ;
    fa[v] = u ;
    d[v] = d[u] + 1 ;
    Dfs(v) ;
    size[u] += size[v] ;
    if(size[son[u]] < size[v]) son[u] = v ;
  }
}
inline void Dfs2(int u , int t) {
  top[u] = t ;
  a[idx] = 0 ;
  id[u] = ++ idx ;
  if(! son[u]) return ;
  Dfs2(son[u] , t) ;
  for(register int i = head[u] ; i ; i = e[i].nxt) {
    int v = e[i].v ;
    if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
  }
}
inline void Change(int x , int y) {
  int fx = top[x] ;
  int fy = top[y] ;
  while(fx ^ fy) {
    if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
    Update(id[fx] , id[x] , 1 , n , 1) ;
    x = fa[fx] ;
    fx = top[x] ;
  }
  if(d[x] > d[y]) swap(x , y) ;
  Update(id[x] + 1 , id[y] , 1 , n , 1) ;
}
inline int Query_Range(int x , int y) {
  int ans = 0 ;
  int fx = top[x] ;
  int fy = top[y] ;
  while(fx ^ fy) {
    if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
    ans += Query(id[fx] , id[x] , 1 , n , 1) ;
    x = fa[fx] ;
    fx = top[x] ;
  }
  if(d[x] > d[y]) swap(x , y) ;
  ans += Query(id[x] + 1 , id[y] , 1 , n , 1) ;
  return ans ;
}
signed main() {
  n = read() ; m = read() ;
  for(register int i = 1 ; i <= n - 1 ; i ++) {
    int u = read() , v = read() ;
    Add(u , v) ;
    Add(v , u) ;
  }
  Dfs(1) ;
  Dfs2(1 , 0) ;
  build(1 , n , 1) ;
  for(register int i = 1 ; i <= m ; i ++) {
    register char c = getchar() ;
    while(c != 'P' && c != 'Q') c = getchar() ;
    int u = read() , v = read() ;
    if(c == 'P') Change(u , v) ;
    else printf("%lld\n" , Query_Range(u , v)) ;
  }
    return 0 ;
}

Qtree1

同样的边权化成点权 然后单点修改

//Isaunoya
#include
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge
int n ;

//============================================SegTree
struct SegTree {
    int val ;
    int max ;
};
const int N = 2e5 + 10 ;
SegTree t[N << 2] ;
int a[N] ;
inline void build(int l , int r , int rt) {
    if( l == r ) {
        t[rt].max = t[rt].val = a[l] ;
        return ;
    }
    int mid = l + r >> 1 ;
    build(l , mid , rt << 1) ;
    build(mid + 1 , r , rt << 1 | 1) ;
    t[rt].max = max(t[rt << 1].max , t[rt << 1 | 1].max) ;
}
inline void Change(int x , int l , int r , int rt , int val) {
    if(l == r) {
        t[rt].max = t[rt].val = val ;
        return ;
    }
    int mid = l + r >> 1 ;
    if(x <= mid) Change(x , l , mid , rt << 1 , val) ;
    else Change(x , mid + 1 , r , rt << 1 | 1 ,val ) ;
    t[rt].max = max(t[rt << 1].max , t[rt << 1 | 1].max) ;
}
inline int Query_Max(int a , int b , int  l, int r , int rt) {
    if(a > b) return 0 ;
    if(a <= l && r <= b) return t[rt].max ;
    int ans = 0 ;
    int mid = l + r >> 1 ;
    if(a <= mid) ans = max(ans , Query_Max(a , b , l , mid , rt << 1)) ;
    if(b > mid) ans = max(ans , Query_Max(a , b , mid + 1 , r , rt << 1 | 1)) ;
    return ans ;
}
//========================================================================

struct node {
    int v ;
    int nxt ;
    int w ;
};
node e[N << 1] ;
int head[N] ;
int cnt = 0 ;
inline void Add_Edge(int u , int v , int w) {
    e[++ cnt].v = v ;
    e[cnt].nxt = head[u] ;
    e[cnt].w  = w ;
    head[u] = cnt ;
    return ;
}
int size[N] ;
int son[N] ;
int d[N] ;
int fst[N] ;
int id[N] ;
int top[N] ;
int fa[N] ;
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        d[v] = d[u] + 1 ;
        fst[v] = e[i].w ;
        fa[v] = u ;
        Dfs1(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
}
int idx = 0 ;
inline void Dfs2(int u , int t) {
    id[u] = ++ idx ;
    a[idx] = fst[u] ;
    top[u] = t ; 
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u] && v ^ son[u]) {
            Dfs2(v , v) ;
        }
    }
}
inline int Query_Range(int x , int y) {
    int fx = top[x] ;
    int fy = top[y] ;
    int ans = 0 ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        ans = max(ans , Query_Max(id[fx] , id[x] , 1 , n , 1)) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
    ans = max(ans , Query_Max(id[x] + 1 , id[y] , 1 , n , 1)) ;
    return ans ;
}
inline int getopt() {
    string s ;
    register char c = getchar() ;
    while(isspace(c)) c = getchar() ;
    while(! isspace(c)) {
        s += c ;
        c = getchar() ;
    }
    if(s == "DONE") return -1 ;
    if(s == "QUERY") return 0 ;
    return 1 ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ;
    for(register int i = 1 ; i < n ; i ++) {
        int u = read() , v = read() , w = read() ;
        Add_Edge(u , v , w) ;
        Add_Edge(v , u , w) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
    build(1 , n , 1) ;
    while(1) {
        int opt = getopt() ;
        if(opt == -1) {
            return 0 ;
        }
        if(opt == 0) {
            int x = read() , y = read() ;
            if(x == y) {
                puts("0") ;
                continue ;
            }
            else {
                write(Query_Range(x , y)) ;
            }
        }
        else {
            int x = read() , y = read() ;
            int u = e[x * 2 - 1].v ;
            int v = e[x * 2].v ;
            if(u == fa[v]) swap(u , v) ;
            Change(id[u] , 1 , n , 1 , y) ;
        }
    }
    return 0 ;
} // 这份代码会T掉一个点…

[USACO15DEC]最大流Max Flow

树剖是可以求LCA的 以这题为例。
这题是需要把压力给两个点 两个点加上1 然后这两个点的\(LCA\)减掉1 因为这个点也需要加上1 所以我们只能把这个1在\(fa_LCA\)减掉
这样递归到最底下 然后一直往上加 显然就是这棵树的权值

#include
using namespace std ;
#define int long long
#define fi first
#define se second
#define pb push_back
inline int read() {
    register int x = 0 , f = 1 ;
    register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
}
template < typename T > inline bool cmax(T & x , T y) {
    return x < y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cmin(T & x , T y) {
    return x > y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cabs(T & x) {
    return x > 0 ? 1 : (x = - x) , 0 ;
}
inline int QP(int x , int y , int Mod) {
    int ans = 1 ;
    for( ; y ; y >>= 1 , x = (x * x) % Mod)
        if(y & 1) ans = (ans * x) % Mod ;
    return ans ;
}
int n ;
struct node {
  int v , nxt ;
} ;
const int N = 50000 + 5 ;
node e[N << 1] ;
int head[N] , cnt = 0 ;
inline void Add(int u , int v) {
  e[++ cnt].v = v ;
  e[cnt].nxt = head[u] ;
  head[u] = cnt ;
  return ;
}
int fa[N] ;
int size[N] ;
int son[N] ;
int top[N] ;
int d[N] ;
inline void Dfs(int u) {
  size[u] = 1 ;
  for(register int i = head[u] ; i ; i = e[i].nxt) {
    int v = e[i].v ;
    if(v == fa[u]) continue ;
    fa[v] = u ;
    d[v] = d[u] + 1 ;
    Dfs(v) ;
    size[u] += size[v];
    if(size[v] > size[son[u]]) son[u] = v ;
  }
}
inline void Dfs2(int u , int t) {
  top[u] = t ;
  if(! son[u]) return ;
  Dfs2(son[u] , t) ;
  for(register int i = head[u] ; i ; i = e[i].nxt) {
    int v = e[i].v ;
    if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
  }
}
inline int Lca(int x , int y) {//树剖求LCA
  int fx = top[x] , fy = top[y] ;
  while(fx ^ fy) {
    if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
    x = fa[fx] ;
    fx = top[x] ;
  }
  if(d[x] > d[y]) swap(x , y) ;
  return x ;
}
int ad[N] ;
int ans = 0 ;
inline void Getans(int u , int father) { // 不断往下找 然后更新答案
  for(register int i = head[u] ; i ; i = e[i].nxt) {
    int v = e[i].v ;
    if(v == father) continue ;
    Getans(v , u) ;
    ad[u] += ad[v] ;
  }
  ans = max(ans , ad[u]) ;
}
signed main() {
  n = read() ; int k = read() ;
  for(register int i = 1 ; i <= n - 1 ;  i ++) {
    int u = read() , v = read() ;
    Add(u , v) ;
    Add(v , u) ;
  }
  Dfs(1) ;
  Dfs2(1 , 0) ;
  for(register int i = 1 ; i <= k ; i ++) {
    int u = read() , v = read() ;
    int lca = Lca(u , v) ;
    ad[u] ++ ;
    ad[v] ++ ;
    ad[lca] -- ;
    ad[fa[lca]] -- ;
  }// 树上差分
  Getans(1 , 0) ;
  printf("%lld" , ans) ;
    return 0 ;
}

[HEOI2016/TJOI2016]树

技巧题
这题是可以在链上二分的 但是主流做法好像不是这个?
链上二分要注意是反过来的。。

#include
using namespace std ;
#define int long long
#define fi first
#define se second
#define pb push_back
inline int read() {
    register int x = 0 , f = 1 ;
    register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
}
template < typename T > inline bool cmax(T & x , T y) {
    return x < y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cmin(T & x , T y) {
    return x > y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cabs(T & x) {
    return x > 0 ? 1 : (x = - x) , 0 ;
}
inline int QP(int x , int y , int Mod) {
    int ans = 1 ;
    for( ; y ; y >>= 1 , x = (x * x) % Mod)
        if(y & 1) ans = (ans * x) % Mod ;
    return ans ;
}
int n , m ;
struct node {
    int v , nxt ;
} ;
const int N = 1e5 + 10 ;
node e[N << 1] ;
int head[N] , cnt = 0 ;
inline void Add(int u , int v) { e[++ cnt].v = v , e[cnt].nxt = head[u] , head[u] = cnt ; }
int fa[N] , size[N] , son[N] , d[N] ;
inline void Dfs(int u) { size[u] = 1 ; for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ; if(v == fa[u]) continue ;
        d[v] = d[u] + 1 , fa[v] = u , Dfs(v) , size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
}
int top[N] , id[N] , idx = 0 , f[N] ;
inline void Dfs2(int u , int t) { id[u] = ++ idx , top[u] = t , f[idx] = u ;
    if(! son[u]) return ; Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ; if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
    }
}
int sum[N << 2] ;
inline void build(int l , int r , int rt){
    if(l == r) { sum[rt] = 0 ; return ; }
    int mid = l + r >> 1 ; build(l , mid , rt << 1) , build(mid + 1 , r , rt << 1 | 1) ;
}
inline void upd(int x , int l , int r , int rt) {
    if(l == r) { sum[rt] = 1 ; return ; }
    int mid = l + r >> 1 ;
    if(x <= mid) upd(x , l , mid , rt << 1) ; else upd(x , mid + 1 , r , rt << 1 | 1) ;
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1] ;
}
inline int query(int a , int b , int l , int r , int rt) {
    if(a <= l && r <= b) { return sum[rt] ; }
    int mid = l + r >> 1 , ans = 0 ;
    if(a <= mid) ans += query(a , b , l , mid , rt << 1) ;
    if(b > mid) ans += query(a , b ,  mid + 1 , r , rt << 1 | 1) ;
    return ans ;
}
inline int chk(int l , int r) { // 二分找答案
    if(l == r) return l ;
    int mid = l + r >> 1 ;
    if(query(mid + 1 , r , 1 , n , 1)) return chk(mid + 1 , r) ; // 反过来。
    else return chk(l , mid) ;
}
inline int Find(int x , int y = 1) { // 求答案 因为最近的点一定是在 1 ~ x 的这条路径上。
    int fx = top[x] , fy = top[y] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        int q = query(id[fx] , id[x] , 1 , n , 1) ;
        if(q) return chk(id[fx] , id[x]) ;
        else { x = fa[fx] , fx = top[x] ; continue ; }
    }
    if(d[x] > d[y]) swap(x , y) ;
    return chk(id[x] , id[y]) ;
}
signed main() {
    n = read() ; m = read() ;
    for(register int i = 1 ; i <= n - 1 ; i ++) { int u = read() , v = read() ; Add(u , v) , Add(v , u) ;}
    Dfs(1) , Dfs2(1 , 0) , build(1 , n , 1) ;
    for(register int i = 1 ; i <= m ; i ++) {
        register char c = getchar() ;
        while(c != 'C' && c != 'Q') c = getchar() ;
        int x = read() ;
        if(c == 'C') upd(id[x] , 1 , n , 1) ;
        if(c == 'Q') printf("%lld\n" , f[Find(x)]) ;
    }
    return 0 ;
}

[国家集训队]旅游
sol

//Isaunoya
#include
using namespace std ;
inline int read() {
    register int x = 0 ;
    register int f = 1 ;
    register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
}
int st[105] ;
template < typename T > inline void write(T x , char c = '\n') {
    int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge
//#define int long long
#define swap(x , y) x ^= y ^= x ^= y
int n ;
const static int N = 200000 + 5 ;
int a[N] ;
namespace SegTree {
    struct Node {
        int mn ; // the min
        int mx ; // the max
        int add ; // the sum
        int lazy ; // the sign
    };
    Node t[N << 2] ;
    inline void Push_down(int rt) {
        if(t[rt].lazy) {
            t[rt << 1].add = - t[rt << 1].add ;
            t[rt << 1 | 1].add = - t[rt << 1 | 1].add ;
            t[rt << 1].lazy ^= 1 ;
            t[rt << 1 | 1].lazy ^= 1 ;
            swap(t[rt << 1].mx , t[rt << 1].mn) ;
            swap(t[rt << 1 | 1].mx , t[rt << 1 | 1].mn) ;
            t[rt << 1].mx = - t[rt << 1].mx ;
            t[rt << 1].mn = - t[rt << 1].mn ;
            t[rt << 1 | 1].mx = - t[rt << 1 | 1].mx ;
            t[rt << 1 | 1].mn = - t[rt << 1 | 1].mn ;
            t[rt].lazy = 0 ;
        }
        return ;
    }
    //==============================================push_down
    inline void Push_Up(int rt) {
        t[rt].add = t[rt << 1].add + t[rt << 1 | 1].add ;
        t[rt].mx = max(t[rt << 1].mx , t[rt << 1 | 1].mx) ;
        t[rt].mn = min(t[rt << 1].mn , t[rt << 1 | 1].mn) ;
        return ;
    }
    //==============================================push_up
    inline void build(int l , int r , int rt) {
        if(l == r) {
            t[rt].add = t[rt].mn = t[rt].mx = a[l] ;
            return ;
        }
        int mid = l + r >> 1 ;
        build(l , mid , rt << 1) ;
        build(mid + 1 , r , rt << 1 | 1) ;
        Push_Up(rt) ;
    }
    //==============================================build
    inline void Add(int x , int l , int r , int rt , int val) {
        if(l == r) {
            t[rt].add = t[rt].mn = t[rt].mx = val ;
            return ;
        }
        int mid = l + r >> 1 ;
        Push_down(rt) ;
        if(x <= mid) Add(x , l , mid , rt << 1 , val) ;
        else Add(x , mid + 1 , r , rt << 1 | 1 , val) ;
        Push_Up(rt) ;
    }
    //==============================================change x - > val
    inline void Change(int a , int b , int l , int r , int rt) {
        if(a > r || b < l) return ;
        if(a <= l && r <= b) {
            t[rt].add = - t[rt].add ;
            t[rt].lazy ^= 1 ;
            swap(t[rt].mx , t[rt].mn) ;
            t[rt].mx = - t[rt].mx ;
            t[rt].mn = - t[rt].mn ;
            return ;
        }
        int mid = l + r >> 1 ;
        Push_down(rt) ;
        Change(a , b , l , mid , rt << 1) ;
        Change(a , b , mid + 1 , r , rt << 1 | 1) ;
        Push_Up(rt) ;
    }
    //===============================================change x - > -x
    inline int Sum(int a , int b , int l , int r , int rt) {
        if(a > r || b < l) return 0 ;
        if(a <= l && r <= b) return t[rt].add ;
        int mid = l + r >> 1 ;
        Push_down(rt) ;
        int ans = 0 ;
        ans += Sum(a , b , l , mid , rt << 1 ) ;
        ans += Sum(a , b , mid + 1 , r , rt << 1 | 1) ;
        Push_Up(rt) ;
        return ans ;
    }
    //====================================================== a - > b sum
    inline int Min(int L , int R , int l , int r , int rt) {
        if(L > r || R < l) return INT_MAX ;
        if(L <= l && r <= R) return t[rt].mn ;
        int ans = INT_MAX ;
        int mid = l + r >> 1 ;
        Push_down(rt) ;
        if(L <= mid) ans = min(ans , Min(L , R , l , mid , rt << 1)) ;
        if(R > mid) ans = min(ans , Min(L , R, mid + 1 , r , rt << 1 | 1)) ;
        Push_Up(rt) ;
        return ans ;
    }
    //====================================================== a - > b min
    inline int Max(int L , int R , int l , int r , int rt) {
        if(L > r || R < l) return INT_MIN ;
        if(L <= l && r <= R) return t[rt].mx ;
        int ans = INT_MIN ;
        int mid = l + r >> 1 ;
        Push_down(rt) ;
        if(L <= mid) ans = max(ans , Max(L , R , l , mid , rt << 1)) ;
        if(R > mid) ans = max(ans , Max(L , R, mid + 1 , r , rt << 1 | 1)) ;
        Push_Up(rt) ;
        return ans ;
    }
    //====================================================== a - > b max
}
//===========================================================SegTree
namespace SLPF {
    struct node {
        int v ;
        int nxt ;
        int w ;
    };
    int fa[N] ; int id[N] ; int son[N] ;
    int size[N] ; int d[N] ; int top[N] ;
    int fst[N] ;
    node e[N << 1] ;
    int tot = 0 ;
    int head[N] ; int cnt = 0 ;
    inline void Add_Edge(int u , int v , int w) {
        e[++ cnt].v = v ;
        e[cnt].nxt = head[u] ;
        e[cnt].w = w ;
        head[u] = cnt ;
        return ;
    }
    inline void Dfs1(int u) {
        size[u] = 1 ;
        for(register int i = head[u] ; i ; i = e[i].nxt) {
            int v = e[i].v ;
            if(v ^ fa[u]) {
                d[v] = d[u] + 1 ;
                fa[v] = u ;
                fst[v] = e[i].w ;
                Dfs1(v) ;
                size[u] += size[v] ;
                if(size[v] > size[son[u]]) son[u] = v ;
            }
        }
    }
    inline void Dfs2(int u , int t) {
        id[u] = ++ tot ;
        top[u] = t ;
        a[tot] = fst[u] ;
        if(son[u]) Dfs2(son[u] , t) ;
        for(register int i = head[u] ; i ; i = e[i].nxt) {
            int v = e[i].v ;
            if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
        }
    }
    //========================================================Dfs1 && Dfs2

    inline void Change_Range(int x , int y) {
        int fx = top[x] ;
        int fy = top[y] ;
        while(fx ^ fy) {
            if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
            SegTree::Change(id[fx] , id[x] , 1 , tot , 1) ;
            x = fa[fx] ;
            fx = top[x] ;
        }
        if(id[x] > id[y]) swap(x , y) ;
        SegTree::Change(id[x] + 1 , id[y] , 1 , tot , 1) ;
    }
    inline int Query_Sum(int x , int y) {
        int fx = top[x] ;
        int fy = top[y] ;
        int ans = 0 ;
        while(fx ^ fy) {
            if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
            ans += SegTree::Sum(id[fx] , id[x] , 1 , tot , 1) ;
            x = fa[fx] ;
            fx = top[x] ;
        }
        if(id[x] > id[y]) swap(x , y) ;
        ans += SegTree::Sum(id[x] + 1 , id[y] , 1 , tot , 1) ;
        return ans ;
    }
    inline int Query_Min(int x , int y) {
        int fx = top[x] ;
        int fy = top[y] ;
        int ans = INT_MAX ;
        while(fx ^ fy) {
            if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
            ans = min(ans , SegTree::Min(id[fx] , id[x] , 1 , tot , 1)) ;
            x = fa[fx] ;
            fx = top[x] ;
        }
        if(id[x] > id[y]) swap(x , y) ;
        ans = min(ans , SegTree::Min(id[x] + 1 , id[y] , 1 , tot , 1)) ;
        return ans ;
    }
    inline int Query_Max(int x , int y) {
        int fx = top[x] ;
        int fy = top[y] ;
        int ans = INT_MIN ;
        while(fx ^ fy) {
            if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
            ans = max(ans , SegTree::Max(id[fx] , id[x] , 1 , tot , 1)) ;
            x = fa[fx] ;
            fx = top[x] ;
        }
        if(id[x] > id[y]) swap(x , y) ;
        ans = max(ans , SegTree::Max(id[x] + 1 , id[y] , 1 , tot , 1)) ;
        return ans ;
    }
}
using namespace SLPF ;
inline int getopt() {
    string s = "" ;
    register char c = getchar() ;
    while(isspace(c)) c = getchar() ;
    while(! isspace(c)) {
        s += c ;
        c = getchar() ;
    }
    if(s == "C") return 0 ;
    if(s == "N") return 1 ;
    if(s == "SUM") return 2 ;
    if(s == "MAX") return 3 ;
    if(s == "MIN") return 4 ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ;
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v = read() , w = read() ;
        u ++ , v ++ ;
        Add_Edge(u , v , w) ;
        Add_Edge(v , u , w) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 0) ;
    SegTree::build(1 , n , 1) ;
    for(register int t = read() ; t -- ; ) {
        int opt = getopt() ;
//      write(opt) ;
        if(opt == 0) {
            int x = read() , y = read() ;
            x ++ ;
            SegTree::Add(id[x] , 1 , n , 1 , y) ;
        }
        if(opt == 1) {
            int x = read() , y = read() ;
            x ++ , y ++ ;
            Change_Range(x , y) ;
        }
        if(opt == 2) {
            int x = read() , y = read() ;
            x ++ , y ++ ;
            write(Query_Sum(x , y)) ;
        }
        if(opt == 3) {
            int x = read() , y = read() ;
            x ++ , y ++ ;
            write(Query_Max(x , y)) ;
        }
        if(opt == 4) {
            int x = read() , y = read() ;
            x ++ , y  ++ ;
            write(Query_Min(x , y)) ;
        }
    }
    return 0 ;
}

[ZJOI2008]树的统计

与模板题差不多。。

//Isaunoya
#include 
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
int n , m , r , p ;
const int N = 10000000 + 5 ;
struct E{ int v ;int nxt ; } ;
E edge[N << 1] ;
int a[N] ; int fa[N] ; int w[N] ; int id[N] ; int son[N] ;
int cnt = 0 ; int head[N] ; int Add[N << 2] , laz[N << 2] ;
int mx[N << 2] ;
int dep[N] , siz[N] , t[N] ;
inline void Add_Edge(register int u , register int v) { edge[++ cnt].v = v ; edge[cnt].nxt = head[u] ; head[u] = cnt ; return ; }
#define l(x) x << 1
#define r(x) x << 1 | 1
inline void Push_down(register int x , register int len) {
    laz[l(x)] += laz[x] ; laz[r(x)] += laz[x] ;
    Add[l(x)] += laz[x] * (len - (len >> 1)) ; Add[r(x)] += laz[x] * (len >> 1) ;
    Add[l(x)] %= p ; Add[r(x)] %= p ;
    laz[x] = 0 ; return ;
}
inline void Build(register int l , register int r , register int rt) {
    if(l == r) { mx[rt] = Add[rt] = a[l] ;  return ; }
    register int mid = l + r >> 1 ;
    Build(l , mid , l(rt)) ; Build(mid + 1 , r , r(rt)) ;
    Add[rt] = (Add[l(rt)] + Add[r(rt)])  ;
    mx[rt] = max(mx[rt << 1] , mx[rt << 1 | 1]) ;
}
inline void Update(register int a , register int b , register int l  , register int r , register int rt , register int k) {
    if(a <= l and r <= b) { laz[rt] += k ; Add[rt] += k * (r - l + 1) ; }
    else {
        if(laz[rt]) Push_down(rt , r - l + 1) ;
        register int mid = l + r >> 1 ;
        if(a <= mid) Update(a , b , l , mid , l(rt) , k) ;
        if(b > mid) Update(a , b , mid + 1 , r , r(rt) , k) ;
        Add[rt] = (Add[l(rt)] + Add[r(rt)])  ;
        mx[rt] = max(mx[rt << 1] , mx[rt << 1 | 1]) ;
    }
}
inline void upd(int x , int l , int r , int rt , int v) {
    if(l == r) {
        mx[rt] = Add[rt] = v ;
        return ;
    }
    int mid = l + r >> 1 ;
    if(x <= mid) upd(x , l , mid , rt << 1 , v) ;
    else upd(x , mid + 1 , r , rt << 1 | 1 , v) ;
    Add[rt] = (Add[l(rt)] + Add[r(rt)])  ;
    mx[rt] = max(mx[rt << 1] , mx[rt << 1 | 1]) ;
}
int res = 0 ;
inline void query(register int a , register int b , register int  l , register int r , register int rt) {
    if(a <= l and r <= b) { res = (res + Add[rt])  ; return ; }
    else {
        if(laz[rt]) Push_down(rt , r - l + 1) ;
        register int mid = l + r >> 1 ;
        if(a <= mid) query(a , b , l , mid , l(rt)) ;
        if(b > mid) query(a , b , mid + 1 , r , r(rt)) ;
    }
}
inline int Query(register int a , register int b , register int l , register int r , register int rt) {
    res = 0 ; query(a , b , l , r , rt) ;
    return res  ;
}
inline int Query_Max(int a , int b , int l , int r , int rt) {
    if(a <= l && r <= b) return mx[rt] ;
    int mid = l + r >> 1 ;
    int ans = INT_MIN ;
    if(a <= mid) ans = max(ans , Query_Max(a , b , l , mid , rt << 1)) ;
    if(b > mid) ans = max(ans , Query_Max(a , b , mid + 1 , r , rt << 1 | 1)) ;
    return ans ;
}
inline void Upd_Range(register int x , register int y , register int k) {
    while(t[x] != t[y]) {
        if(dep[t[x]] < dep[t[y]]) swap(x , y) ;
         Update(id[t[x]] , id[x] , 1 , n , 1 , k) ;
         x = fa[t[x]] ;
    }
    if(dep[x] > dep[y]) swap(x , y) ;
    Update(id[x] , id[y] , 1 , n , 1 , k) ;
}
inline int Query_Range(register int x , register int y) {
    int ans = 0 ;
    while(t[x] != t[y]) {
        if(dep[t[x]] < dep[t[y]]) swap(x , y) ;
        ans += Query(id[t[x]] , id[x] , 1 , n , 1) ;
        x = fa[t[x]] ;
    }
    if(dep[x] > dep[y]) swap(x , y) ;
    ans += Query(id[x] , id[y] , 1 , n , 1) ;
    return ans  ;
}
inline int Query_Max_Range(register int x , register int y) {
    int ans = INT_MIN ;
    while(t[x] != t[y]) {
        if(dep[t[x]] < dep[t[y]]) swap(x , y) ;
        ans = max(ans , Query_Max(id[t[x]] , id[x] , 1 , n , 1)) ;
        x = fa[t[x]] ;
    }
    if(dep[x] > dep[y]) swap(x , y) ;
    ans = max(ans , Query_Max(id[x] , id[y] , 1 , n , 1)) ;
    return ans  ;
}
inline int Qson(register int x) { return Query(id[x] , id[x] + siz[x] - 1 , 1 , n , 1) ; }
inline void Updson(register int x , register int k) { Update(id[x] , id[x] + siz[x] - 1 , 1 , n , 1 , k) ; return ; }
inline void Dfs1(register int x , register int f , register int deep) {
    dep[x] = deep ; fa[x] = f ; siz[x] = 1 ;
    int max_son = -1 ;
    for(register int i = head[x] ; i ; i = edge[i].nxt) {
        register int v = edge[i].v ;
        if(v == f) continue ;
        Dfs1(v , x , deep + 1) ;siz[x] += siz[v] ;
        if(siz[v] > max_son) max_son = siz[v] , son[x] = v ;
    }
}
int tot = 0 ;
inline void Dfs2(register int x , register int tf) {
    id[x] = ++ tot ; a[tot] = w[x] ; t[x] = tf ;
    if(! son[x]) return ;
    Dfs2(son[x] , tf) ;
    for(register int i = head[x] ; i ; i = edge[i].nxt) {
        int v = edge[i].v ;
        if(v == fa[x] or v == son[x]) continue ;
        Dfs2(v , v) ;
    }
}
inline int getopt() {
    string s = "" ; register char c = getchar() ;
    while(isspace(c)) c = getchar() ;
    while(! isspace(c)) {
        s += c ;
        c = getchar() ;
    }
    if(s == "CHANGE") return 0 ;
    if(s == "QMAX") return 1 ;
    if(s == "QSUM") return 2 ; 
}
signed main() {
    n = read() ; 
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v = read() ;
        Add_Edge(u , v) ;
        Add_Edge(v , u) ;
    }for(register int i = 1 ; i <= n ; i ++) w[i] = read() ;
    m = read() ; r = 1 ;
    Dfs1(r , 0 , 1) ; Dfs2(r , r) ; Build(1 , n , 1) ;
    for( ; m -- ; ) {
        int opt = getopt() ;
//        cout  << opt ;
        int u = read() , v = read() ;
        if(opt == 0) {
            upd(id[u] , 1 , n , 1 , v) ;
        }
        if(opt == 1) {
            write(Query_Max_Range(u , v)) ;
        }
        if(opt == 2) {
            write(Query_Range(u , v)) ;
        }
    }
    return 0 ;
}

[SHOI2012]魔法树

//Isaunoya
#include
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge

#define int long long
//==============================================================edge
struct node {
    int v ;
    int nxt ;
};
int n ;
const int N = 100000 + 5 ;
node e[N << 1] ;
int cnt = 0 ; int head[N] ;
inline void Add(int u , int v) {
    e[++ cnt].v = v ;
    e[cnt].nxt = head[u] ;
    head[u] = cnt ;
    return ;
}
int size[N] ;
int son[N] ; int id[N] ;
int idx = 0 ; int d[N] ;
int top[N] ; int fa[N] ;
//================================================================slpf
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        fa[v] = u ;
        d[v] = d[u] + 1 ;
        Dfs1(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
}

inline void Dfs2(int u , int t) {
    id[u] = ++ idx ;
    top[u] = t ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
    }
}
//=======================================================Segment_Tree
struct Seg {
    int sum ;
    int tag ;
};
Seg t[N << 2] ;
inline void Push_down(int rt , int l , int r) {
    if(t[rt].tag) {
        t[rt << 1].tag += t[rt].tag ;
        t[rt << 1 | 1].tag += t[rt].tag ;
        int mid = l + r >> 1 ;
        t[rt << 1].sum += (mid - l + 1) * t[rt].tag ;
        t[rt << 1 | 1].sum += (r - mid) * t[rt].tag ;
        t[rt].tag = 0 ;
        return ;
    }
}
inline void Push_up(int rt) {
    t[rt].sum = t[rt << 1].sum + t[rt << 1 | 1].sum ;
}
inline void build(int l , int r , int rt) {
    if(l == r) {
        t[rt].sum = 0 ;
        return ;
    }
    int mid = l + r >> 1 ;
    build(l , mid , rt << 1) ;
    build(mid + 1 , r , rt << 1 | 1) ;
    Push_up(rt) ;
}
inline void Change(int a , int b , int l , int r , int rt , int val) {
    if(a <= l && r <= b) {
        t[rt].sum += (r - l + 1) * val ;
        t[rt].tag += val ;
        return ;
    }
    Push_down(rt , l , r) ;
    int mid = l + r >> 1 ;
    if(a <= mid) Change(a , b , l , mid , rt << 1 , val) ;
    if(b > mid) Change(a , b , mid + 1 , r , rt << 1 | 1 , val) ;   
    Push_up(rt) ;
    return ;
}
inline int Query(int a , int b , int l , int r , int rt) {
    if(a <= l && r <= b) return t[rt].sum ;
    Push_down(rt , l , r) ;
    int mid = l + r >> 1 ;
    int ans = 0 ;
    if(a <= mid) ans += Query(a , b , l , mid , rt << 1) ;
    if(b > mid) ans += Query(a , b , mid + 1 , r , rt << 1 | 1) ;
    return ans ;
}

inline void Change_Range(int x , int y , int val) {
    int fx = top[x] ;
    int fy = top[y] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        Change(id[fx] , id[x] , 1 ,n , 1 , val) ;
        x = fa[fx] ; 
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
    Change(id[x] , id[y] , 1 , n , 1 , val) ;
    return ;
}
//inline int Query_Range(int x , int y) {
//  int fx = top[x] ;
//  int fy = top[y] ;
//  int ans = 0 ;
//  while(fx  ^ fy) {
//      if(d[fx] < d[fy]) swap(x ,y) , swap(fx , fy) ;
//      ans += Query(id[fx] , id[x] , 1 , n , 1) ;
//      x = fa[fx] ;
//      fx = top[x] ;
//  }
//  if(id[x] > id[y]) swap(x , y) ;
//  ans += Query(id[x] , id[y] , 1 , n , 1) ;
//  return ans ;
//}
inline int Query_Son(int x) { return Query(id[x] , id[x] + size[x] - 1 , 1 , n , 1) ; }
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ;
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v= read() ;
        u ++ ; v ++ ;
        Add(u , v) ;
        Add(v , u) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
    build(1 , n , 1) ;
    for(register int q = read() ; q -- ; ) {
        register char c = getchar() ;
        for( ; c != 'Q' && c != 'A' ; c = getchar()) ;
        if(c == 'A') {
            int x = read() , y = read() ;
            x ++ , y ++ ;
            int val = read() ;
            Change_Range(x , y , val) ;
        } 
        if(c == 'Q') {
            int x = read() ;
            x ++ ;
            write(Query_Son(x)) ;
        } 
    }
    return 0 ;
}

APIO[部落冲突]
sol

//Isaunoya
#include
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge
int n ;
vector < pair < int , int > > v ;
const int N = 3e5 + 10 ;
int a[N] ;
struct node {
    int v ;
    int nxt ;
};
node e[N << 1] ; int head[N] ;
int cnt = 0 ;
inline void Add(int u , int v) {
    e[++ cnt].v = v ;
    e[cnt].nxt = head[u] ;
    head[u] = cnt ;
    return ;
} 
int size[N] ;
int top[N] , son[N] ;
int fa[N] ; int id[N] ; int d[N] ;
int idx = 0 ;
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u]) {
            fa[v] = u ;
            d[v] = d[u] + 1 ;
            Dfs1(v) ;
            size[u] += size[v] ;
            if(size[son[u]] < size[v]) son[u] = v ;
        }
    }
}
inline void Dfs2(int u , int t) {
    id[u] = ++ idx ;
    top[u] = t ;
    a[idx] = 0 ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u] && v ^ son[u]) {
            Dfs2(v , v) ;
        }
    }
}
int sum[N << 2] ;
int tag[N << 2] ;
inline void build(int l , int r , int rt) {
    if(l == r) {
        sum[rt] = a[l] ; 
        tag[rt] = 0 ;
        return ;
    }
    int mid = l + r >> 1 ;
    build(l , mid , rt << 1) ;
    build(mid + 1 , r , rt << 1 | 1) ;
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1] ;
    return ;
}
inline void Push_down(int rt , int l , int r) {
    if(tag[rt]) {
        tag[rt << 1] += tag[rt] ;
        tag[rt << 1 | 1] += tag[rt] ;
        int mid = l + r >> 1 ;
        sum[rt << 1] += tag[rt] * (mid - l + 1) ;
        sum[rt << 1 | 1] += tag[rt] * (r - mid) ;
        tag[rt] = 0 ;
        return ;
    }
}
inline void Change(int a , int b , int l , int r , int rt , int val) {
    if(a <= l && r <= b) {
        sum[rt] += val * (r - l + 1) ;
        tag[rt] += val ;
        return ;
    }
    int mid = l + r >> 1 ;
    Push_down(rt , l , r) ;
    if(a <= mid) Change(a , b , l , mid , rt << 1 , val) ;
    if(b > mid) Change(a , b , mid + 1 , r , rt << 1 | 1 , val) ;
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1] ;
    return ;
}
inline int Query(int a , int b , int l , int r , int rt) {
    if(a <= l && r <= b) return sum[rt] ;
    int mid = l + r >> 1 ;
    int ans = 0 ;
    Push_down(rt , l , r) ;
    if(a <= mid) ans += Query(a , b , l , mid , rt << 1) ;
    if(b > mid) ans += Query(a , b , mid + 1 , r , rt << 1 | 1) ;
    return ans ;
}
inline int Query_Range(int x , int y) {
    int fx = top[x] ;
    int fy = top[y] ;
    int ans = 0 ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        ans += Query(id[fx] , id[x] , 1 , n , 1) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
    ans += Query(id[x] + 1 , id[y] , 1 , n , 1) ;
    return ans ;
}
inline void Change_Range(int x , int y , int val) {
    int fx = top[x] ;
    int fy = top[y] ;
    int ans = 0 ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        Change(id[fx] , id[x] , 1 , n , 1 , val) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
    Change(id[x] + 1 , id[y] , 1 , n , 1 , val) ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ; int q = read() ;
    for(register int  i = 1 ; i <= n - 1 ; i ++) {
        int u = read() ;
        int v = read() ;
        Add(u , v) ;
        Add(v , u) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
    build(1 , n , 1) ;
    for(register int i = 1 ; i <= q ; i ++) {
        register char c = getchar() ;
        for( ; c != 'C' && c != 'Q' && c != 'U' ; c = getchar()) ;
        if(c == 'Q') {
            int x = read() , y = read() ;
            if(Query_Range(x , y)) puts("No") ;
            else puts("Yes") ;
        }
        if(c == 'C') {
            int x = read() , y = read() ;
            Change_Range(x , y , 1) ;
            v.push_back(make_pair(x , y)) ;
        }
        if(c == 'U') {
            int num = read() - 1 ;
            int x = v[num].first ;
            int y = v[num].second ;
            Change_Range(x , y , - 1) ;
        }
    }
    return 0 ;
}

[HAOI2015]树上操作

  • 操作 1 :把某个节点 x 的点权增加 a 。
  • 操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。
  • 操作 3 :询问某个节点 x 到根的路径中所有点的点权和。

很显然是个板子题啊。

// Isaunoya
#include 
using namespace std ;
inline int read() { register int res = 0 ; register int f = 1 ; register char c ;
#define gc c = getchar()
    while(isspace(gc)) ; if(c == '-') f = -1 , gc ;
    while(res = (res << 1) + (res << 3) + (c & 15) , isdigit(gc)) ;
    return res * f ;
}
#define int long long
int n , m , r ;
const int N = 1000000 + 5 ;
struct E{ int v ;int nxt ; } ;
E edge[N << 1] ;
int a[N] ; int fa[N] ; int w[N] ; int id[N] ; int son[N] ;
int cnt = 0 ; int head[N] ; int Add[N << 2] , laz[N << 2] ;
int dep[N] , siz[N] , t[N] ;
inline void Add_Edge(register int u , register int v) { edge[++ cnt].v = v ; edge[cnt].nxt = head[u] ; head[u] = cnt ; return ; }
#define l(x) x << 1
#define r(x) x << 1 | 1
inline void Push_down(register int x , register int len) {
    laz[l(x)] += laz[x] ; laz[r(x)] += laz[x] ;
    Add[l(x)] += laz[x] * (len - (len >> 1)) ; Add[r(x)] += laz[x] * (len >> 1) ;
    laz[x] = 0 ; return ;
}
inline void Build(register int l , register int r , register int rt) {
    if(l == r) { Add[rt] = a[l] ;  return ; }
    register int mid = l + r >> 1 ;
    Build(l , mid , l(rt)) ; Build(mid + 1 , r , r(rt)) ;
    Add[rt] = (Add[l(rt)] + Add[r(rt)])  ;
}
inline void Update(register int a , register int b , register int l  , register int r , register int rt , register int k) {
    if(a <= l and r <= b) { laz[rt] += k ; Add[rt] += k * (r - l + 1) ; }
    else {
        if(laz[rt]) Push_down(rt , r - l + 1) ;
        register int mid = l + r >> 1 ;
        if(a <= mid) Update(a , b , l , mid , l(rt) , k) ;
        if(b > mid) Update(a , b , mid + 1 , r , r(rt) , k) ;
        Add[rt] = (Add[l(rt)] + Add[r(rt)])  ;
    }
}
int res = 0 ;
inline void query(register int a , register int b , register int  l , register int r , register int rt) {
    if(a <= l and r <= b) { res = (res + Add[rt])  ; return ; }
    else {
        if(laz[rt]) Push_down(rt , r - l + 1) ;
        register int mid = l + r >> 1 ;
        if(a <= mid) query(a , b , l , mid , l(rt)) ;
        if(b > mid) query(a , b , mid + 1 , r , r(rt)) ;
    }
}
inline int Query(register int a , register int b , register int l , register int r , register int rt) {
    res = 0 ; query(a , b , l , r , rt) ;
    return res  ;
}
inline void Upd_Range(register int x , register int y , register int k) {
    while(t[x] != t[y]) {
        if(dep[t[x]] < dep[t[y]]) swap(x , y) ;
         Update(id[t[x]] , id[x] , 1 , n , 1 , k) ;
         x = fa[t[x]] ;
    }
    if(dep[x] > dep[y]) swap(x , y) ;
    Update(id[x] , id[y] , 1 , n , 1 , k) ;
}
inline int Query_Range(register int x , register int y) {
    int ans = 0 ;
    while(t[x] != t[y]) {
        if(dep[t[x]] < dep[t[y]]) swap(x , y) ;
        ans += Query(id[t[x]] , id[x] , 1 , n , 1) ;
        x = fa[t[x]] ;
    }
    if(dep[x] > dep[y]) swap(x , y) ;
    ans += Query(id[x] , id[y] , 1 , n , 1) ;
    return ans ;
}
inline int Qson(register int x) { return Query(id[x] , id[x] + siz[x] - 1 , 1 , n , 1) ; }
inline void Updson(register int x , register int k) { Update(id[x] , id[x] + siz[x] - 1 , 1 , n , 1 , k) ; return ; }
inline void Dfs1(register int x , register int f , register int deep) {
    dep[x] = deep ; fa[x] = f ; siz[x] = 1 ;
    int max_son = -1 ;
    for(register int i = head[x] ; i ; i = edge[i].nxt) {
        register int v = edge[i].v ;
        if(v == f) continue ;
        Dfs1(v , x , deep + 1) ;siz[x] += siz[v] ;
        if(siz[v] > max_son) max_son = siz[v] , son[x] = v ;
    }
}
int tot = 0 ;
inline void Dfs2(register int x , register int tf) {
    id[x] = ++ tot ; a[tot] = w[x] ; t[x] = tf ;
    if(! son[x]) return ;
    Dfs2(son[x] , tf) ;
    for(register int i = head[x] ; i ; i = edge[i].nxt) {
        int v = edge[i].v ;
        if(v == fa[x] or v == son[x]) continue ;
        Dfs2(v , v) ;
    }
}
signed main() {
    n = read() ; m = read() ; r = 1 ;
    for(register int i = 1 ; i <= n ; i ++) w[i] = read() ;
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v = read() ;
        Add_Edge(u , v) ;
        Add_Edge(v , u) ;
    }
    Dfs1(r , 0 , 1) ; Dfs2(r , r) ; Build(1 , n , 1) ;
    for( ; m -- ; ) {
        register int opt = read() ;
        if(opt == 1) {
            int x = read() , a = read() ;
            Upd_Range(x , x , a) ;
        }
        if(opt == 2) {
            int x = read() , z = read() ;
            Updson(x , z) ;
        }
        if(opt == 3) {
            int x = read() ;
            printf("%lld\n" , Query_Range(x , r)) ;
        }
    }
    return 0 ;
}

上面都是用线段树维护的序列 其实树剖不一定要用 线段树 只要是数据结构都可以
比如说ODT 什么的 如果不会ODT也没什么关系 你会用线段树当然更好 ODT 是可以卡掉的 但是在树上问题 通常不容易卡掉。

CF343D Water Tree

//Isaunoya
#include
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge

int n ;
struct ODT {
    int l , r ;
    mutable int val ;
    bool operator <(const ODT & x) const{
        return l < x.l ;
    }
};
set < ODT > s ;
typedef set < ODT > :: iterator IT ;
inline IT Split(int pos) {
    IT it(-- s.upper_bound({pos , 0 , 0})) ;
    if(it -> l == pos) return it ;
    int L(it -> l) ;
    int R(it -> r) ;
    int Val(it -> val) ;
    s.erase(it) ;
    s.insert({L , pos - 1 , Val}) ;
    return s.insert({pos , R , Val}).first ;
}
inline void Assign(int l , int r , int val) {
    IT itr(Split(r + 1)) ;
    IT itl(Split(l)) ;
    s.erase(itl , itr) ;
    s.insert({l , r , val}) ;
    return ;
}
//===========================================ODT
struct node {
    int v ;
    int nxt ;
} ;
const int N = 500000 + 5 ;
node e[N << 1] ;
int head[N] ;
int cnt = 0 ;
inline void Add(int u ,int v) {
    e[++ cnt].v = v ;
    e[cnt].nxt = head[u] ;
    head[u] = cnt ;
    return ;
}
int size[N] ; int d[N] ;
int top[N] ; int idx = 0 ;
int id[N] ; int fa[N] ;
int son[N] ;
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        d[v] = d[u] + 1 ;
        fa[v] = u ;
        Dfs1(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    } 
}
inline void Dfs2(int u , int t) {
    id[u] = ++ idx ;
    top[u] = t ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
    }
}
inline void Change_Range(int x) {
    int fx = top[x] ;
    while(fx ^ 1) {
        Assign(id[fx] , id[x] , 0) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    Assign(id[1] , id[x] , 0) ;
}
inline void Query(int x) {
    IT pos(Split(x)) ;
    write(pos -> val) ;
}
signed main() {
#ifndef Online_Judge
//  freopen("testdata.in" , "r" , stdin) ;
//  freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ;
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v = read() ;
        Add(u , v) ;
        Add(v , u) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
    s.insert({0 , n + 1 , 0}) ;
    for(register int m = read() ; m -- ; ) {
        int opt = read() ;
        if(opt == 1) {
            int v = read() ;
            Assign(id[v] , id[v] + size[v] - 1 , 1) ;
        }
        if(opt == 2) {
            int v = read() ;
            Change_Range(v) ;
        }
        if(opt == 3) {
            int v = read() ;
            Query(id[v]) ;
        }
    }
    return 0 ;
}

[NOI2015]软件包管理器
因为 软件之间是有依赖关系的

  • 每次安装软件 就把根节点到x软件路径上的值全部变为1
  • 同理 每次卸载软件 就把x以及它的子树的值变为0
//Isaunoya
#include
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge

struct node {
    int l ; int r ;
    mutable int val ;
    bool operator < (const node & x) const {
        return l < x.l ;
    }
};
set < node > s ;

#define slt set < node > :: iterator
inline slt Split(int pos) {
    slt it = s.lower_bound((node) {pos}) ;
    if(it != s.end() && it -> l == pos) return it ;
    -- it ;
    int l = it -> l ;
    int r = it -> r ;
    int val = it -> val ;
    s.erase(it) ;
    s.insert({l , pos - 1 , val}) ;
    return s.insert({pos , r , val}).first ;
}
inline int Assign(int l , int r , int val) {
    slt it2 = Split(r + 1) ;
    slt it1 = Split(l) ;
    int sum = 0 ;
    int sum2 = (r - l + 1) * val  ;
    for(slt it = it1 ; it != it2 ; it ++) sum += (it -> r - it -> l + 1) * it -> val ;
    s.erase(it1 , it2) ;
    s.insert({l , r , val}) ;
    return abs(sum - sum2) ;
}
int n ;
struct Node {
    int v ;
    int nxt ;
};
const int N = 1e5 + 10 ;
Node e[N << 1] ;
int cnt = 0 ;
int head[N] ;
inline void Add(int u , int v) {
    e[++ cnt].v = v ;
    e[cnt].nxt = head[u] ;
    head[u] = cnt ;
    return ;
}
int top[N] ;
int id[N] ; int size[N] ;
int d[N] ; int idx = 0 ;
int fa[N] ; int son[N] ;
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt ) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        d[v] = d[u] + 1 ;
        fa[v] = u ;
        Dfs1(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
    return ;
}
inline void Dfs2(int u , int t) {
    id[u] = ++ idx ;
    top[u] = t ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if((v ^ fa[u]) && (v ^ son[u])) Dfs2(v , v) ;
    }
}
inline int getopt() { string s = "" ;
    register char c = getchar() ;
    while(isspace(c)) c = getchar() ;
    while(! isspace(c)) {
        s += c ;
        c = getchar() ;
    }
    if(s == "install") return 1 ;
    if(s == "uninstall") return 0 ;
}
inline int Change_Range(int x , int y) {
    int fx = top[x] ;
    int fy = top[y] ;
    int ans = 0 ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        ans += Assign(id[fx] , id[x] , 1) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
    ans += Assign(id[x] , id[y] , 1) ;
    return ans ;
}
inline int Uninstall(int x) {
    return Assign(id[x] , id[x] + size[x] - 1 , 0) ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ;
    s.insert({1 , n + 1 , 0}) ;
    for(register int i = 2 ; i <= n ; i ++) {
        int u = read() ; u ++ ;
        Add(u , i) ;
        Add(i , u) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
    for(register int t = read() ; t -- ; ) {
        int opt = getopt() ;
        if(opt == 1) {
            int x = read() ; x ++ ;
            write(Change_Range(x , 1)) ;
        }
        if(opt == 0) {
            int x = read() ; x ++ ;
            write(Uninstall(x)) ;
        }
    }
    return 0 ;
}

[SDOI2011]染色

这题也是个ODT
不过不同的是 这题是区间的颜色数量 连续的颜色算一次 这样的话 线段树还是很难维护的。(应该也可以。)

//Isaunoya
#include
using namespace std ;
inline int read() {
    register int x = 0 ;
    register int f = 1 ;
    register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
}
int st[105] ;
template < typename T > inline void write(T x , char c = '\n') {
    int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge

struct node {
    int l ;
    int r ;
    mutable int val ;
    bool operator < (const node & x) const {
        return l < x.l ;
    }
};
set < node > s ;

#define slt set < node > :: iterator
inline slt Split(int pos) {
    slt it (-- s.upper_bound({pos})) ;
    if(it -> l == pos) return it ;
    int l = it -> l ;
    int r = it -> r ;
    int val = it -> val ;
    s.erase(it) ;
    s.insert({l , pos - 1 , val}) ;
    return s.insert({pos , r , val}).first ;
}
inline void Assign(int l , int r , int val) {
    slt itr = Split(r + 1) ;
    slt itl = Split(l) ;
    s.erase(itl , itr) ;
    s.insert({l , r , val}) ;
    return ;
}
int n ;
struct Node {
    int v ;
    int nxt ;
};
const int N = 1e5 + 10 ;
int a[N] ;
Node e[N << 1] ;
int cnt = 0 ;
int head[N] ;
inline void Add(int u , int v) {
    e[++ cnt].v = v ;
    e[cnt].nxt = head[u] ;
    head[u] = cnt ;
    return ;
}
int top[N] ;
int id[N] ;
int size[N] ;
int d[N] ;
int idx = 0 ;
int fa[N] ;
int son[N] ;
int fst[N] ;
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt ) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        d[v] = d[u] + 1 ;
        fa[v] = u ;
        Dfs1(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
    return ;
}
inline void Dfs2(int u , int t) {
    id[u] = ++ idx ;
    top[u] = t ;
    a[idx] = fst[u] ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if((v ^ fa[u]) && (v ^ son[u])) Dfs2(v , v) ;
    }
}
inline int getopt() {
    register char c = getchar() ;
    while(isspace(c)) c = getchar() ;
    return c == 'Q' ;
}
inline void Change_Range(int x , int y , int val) {
    int fx = top[x] ;
    int fy = top[y] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        Assign(id[fx] , id[x] , val) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
    Assign(id[x] , id[y] , val) ;
}
#define swap(x , y) x ^= y ^= x ^= y
inline int Query_Range(int x , int y) {
    int ans = 0 ;
    int lasta = 0 ;
    int lastb = 0 ;
    int fx = top[x] ;
    int fy = top[y] ;
    slt itl , itr ;
    while(fx ^ fy) {
        if(d[fx] > d[fy]) {
            itr = Split(id[x] + 1) , itl = Split(id[fx]) ;
            for(-- itr ; ; --itr) {
                if(itr -> val ^ lasta)
                    lasta = itr -> val , ++ ans ;
                if(itr == itl) break;
            }
            x = fa[fx] ;
            fx = top[x] ;
        } else {
//          swap(x , y) , swap(fx , fy) ;
            itr = Split(id[y] + 1) , itl = Split(id[fy]) ;
            for(-- itr ; ; --itr) {
                if(itr -> val ^ lastb)
                    lastb = itr -> val , ++ ans ;
                if(itr == itl) break;
            }
            y = fa[fy] ;
            fy = top[y] ;
        }
    }
    if(id[x] > id[y]) {
        itr = Split(id[x] + 1) , itl = Split(id[y]) ;
        for(-- itr ; ; --itr) {
            if(itr -> val ^ lasta)
                lasta = itr -> val , ++ ans ;
            if(itr == itl) break;
        }
    } else {
        itr = Split(id[y] + 1) , itl = Split(id[x]) ;
        for(-- itr ; ; --itr) {
            if(itr -> val ^ lastb)
                lastb = itr -> val , ++ ans ;
            if(itr == itl) break;
        }
    }
    return ans - (lasta == lastb) ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ;
    int t = read() ;
    for(register int i = 1 ; i <= n ; i ++) fst[i] = read() ;
    for(register int i = 2 ; i <= n ; i ++) {
        int u = read() ;
        int v = read() ;
        Add(u , v) ;
        Add(v , u) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
    for(register int i = 1 ; i <= n ; i ++) s.insert({i , i , a[i]}) ;
    for( ; t -- ; ) {
        int opt = getopt() ;
        if(opt == 1) {
            int x = read() , y = read() ;
            write(Query_Range(x , y));
        }
        if(opt == 0) {
            int x = read() , y = read() , val = read() ;
            Change_Range(x , y , val) ;
        }
    }
    return 0 ;
}

[SDOI2014]旅行
动态开点。

  • “CC x c“:城市x的居民全体改信了c教;
  • “CW x w“:城市x的评级调整为w;
  • “QS x y“:一位旅行者从城市x出发,到城市y,并记下了途中留宿过的城市的评级总和;
  • “QM x y“:一位旅行者从城市x出发,到城市y,并记下了途中留宿过的城市的评级最大值。
    基本操作还是一样。
    如果没有 第一个 操作 是可以树剖+线段树做的 但是有了第一个操作 就只能用动态开点做了。
//Isaunoya
#include
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge
int n ;
const int N = 5e5 + 5 ;

int val[N] ;
int cl[N] ;
struct node {
    int v ;
    int nxt ;
};
node e[N << 1] ;
int L[N << 2] , R[N << 2] ;
int head[N] ; int cnt = 0 ;
inline void Add(int u ,int v) {
    e[++ cnt].v = v ;
    e[cnt].nxt = head[u] ; 
    head[u] = cnt ;
    return ;
}
int a[N] ;
int fa[N] ;
int top[N] ;
int id[N] ;
int size[N] ;
int son[N] ;
int d[N] ; int RT[N] ;
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        fa[v] = u ;
        d[v] = d[u] + 1 ;
        Dfs1(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
}
int idx = 0 ;
inline void Dfs2(int u , int t) {
    top[u] = t ;
    id[u] = ++ idx ;
    a[idx] = u ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
    }
}
int mx[N << 2] ;
int sum[N << 2];

inline void Push_Up(int rt) {
    mx[rt] = max(mx[L[rt]] , mx[R[rt]]) ;
    sum[rt] = sum[L[rt]] + sum[R[rt]] ;
    return ;
}
int c = 0 ;

inline void Modify(int & p , int l , int r , int k , int val) {//开点。
    if(! p) p = ++ c ;
    if(l >= r) {
        mx[p] = sum[p] = val ;
        return ;
    } 
    int mid = l + r >> 1 ;
    if(k <= mid) Modify(L[p] , l , mid , k , val) ;
    else Modify(R[p] , mid + 1 , r , k , val) ;
    Push_Up(p) ;
    if(! sum[p]) p = 0 ;
}
inline int Query_Max(int p , int a , int b , int l , int r) { // 查询
    if(! p) return -1 ;
    if(a <= l && r <= b) return mx[p] ;
    int mid = l + r >> 1 ;
    int ans = - 1 ;
    if(a <= mid) ans = max(ans , Query_Max(L[p] , a , b,  l , mid)) ;
    if(b > mid) ans = max(ans , Query_Max(R[p] , a , b, mid + 1 , r)) ;
    return ans ;
}
inline int Query_Sum(int p , int a , int b , int  l , int r){//查询
    if(! p) return 0 ;
    if(a <= l && r <= b) return sum[p] ;
    int mid = l + r >> 1 ;
    int ans = 0 ;
    if(a <= mid) ans += Query_Sum(L[p] , a , b , l , mid) ;
    if(b > mid) ans += Query_Sum(R[p] , a , b , mid + 1 , r) ;
    return ans ;
}
inline int Max_Range(int x , int y) {
    int fx = top[x] ;
    int fy = top[y] ;
    int ans = 0 ;
    int t = cl[x] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        ans = max(ans , Query_Max(RT[t] , id[fx] , id[x] , 1 , idx)) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
    ans = max(ans , Query_Max(RT[t] , id[x] , id[y] , 1 , idx)) ;
    return ans ;
}
inline int Query_Range(int x , int y) {
    int fx = top[x] ;
    int fy = top[y] ;
    int ans = 0 ;
    int t = cl[x] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        ans += Query_Sum(RT[t] , id[fx] , id[x] , 1 , idx) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(id[x] > id[y]) swap(x , y) ;
//  cout << ans << endl ;
    ans += Query_Sum(RT[t] , id[x] , id[y] , 1 , idx) ;
//  cout << ans << endl ;
    return ans ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ; int q = read() ;
    for(register int i = 1 ; i <= n ;i ++) {
        val[i] = read() ;
        cl[i] = read() ;
    }
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v = read() ;
        Add(u , v) ;
        Add(v , u) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
//  puts("1") ;
    for(register int i = 1 ; i <= n ; i ++) Modify(RT[cl[a[i]]] , 1 , idx ,  i , val[a[i]]) ;
    for(register int i = 1 ; i <= q ; i ++) {
        register char c = getchar() ;
        for( ; c != 'C' && c != 'Q' ; c = getchar()) ;
        register char c2 = getchar() ;
        if(c == 'C' && c2 == 'C') {
            int c = read() , x = read() ;
            Modify(RT[cl[c]] , 1 , idx , id[c] , 0) ;//删点
            cl[c] = x ;
            Modify(RT[cl[c]] , 1 , idx , id[c] , val[c]) ;//开点
        }
        if(c == 'C' && c2 == 'W') {
            int c = read() , x = read() ;
            val[c] = x ;
            Modify(RT[cl[c]] , 1 , idx , id[c] , val[c]) ; // 修改
        }
        if(c == 'Q' && c2 == 'S') {
            int x = read() , y = read() ;
            write(Query_Range(x , y)) ;
        }
        if(c == 'Q' && c2 == 'M') {
            int x = read() , y  =read() ;
            write(Max_Range(x , y)) ;
        }
    }
    return 0 ;
}

Count on a tree

给定一棵N个节点的树,每个点有一个权值,对于M个询问(u,v,k),你需要回答u xor lastans和v这两个节点间第K小的点权。其中lastans是上一个询问的答案,初始为0,即第一个询问的u是明文。
强制在线,那么考虑静态主席树求区间第k小。。 还是一样的套路。。 如果不会主席树 建议 -> HERE

//Isaunoya
#include
using namespace std ;
inline int read() { register int x = 0 ; register int f = 1 ; register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
} int st[105] ;
template < typename T > inline void write(T x , char c = '\n') { int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge
const int N = 1e5 + 10 ;
const int M = N << 5 ;
int n , q ; int m ;
int sum[M] , L[M] , R[M] ;
int a[N] , b[N] , rt[N] ;
struct node{
    int v ;
    int nxt ;
};
node e[N << 1] ;
int head[N] ;
int fa[N] , size[N] , d[N] , son[N] , top[N] ;
int cnt =  0 ; 
inline void Add(int u , int v) {
    e[ ++ cnt] . v = v ;
    e[ cnt] . nxt = head[u] ;
    head[u] = cnt ;
    return ;
}
int tot = 0 ;
inline void Update(int last , int & now , int l , int r , int x) { // 动态开点 主席树基操
    sum[now = ++ tot] = sum[last] + 1 ;
    if(l == r) return ;
    int mid = l + r >> 1 ;
    if(x <= mid) R[now] = R[last] , Update(L[last] , L[now] , l , mid , x) ;
    else L[now] = L[last] , Update(R[last] , R[now] , mid + 1 , r , x) ;
    return ;
}
inline int Query(int a , int b , int lca ,int lca_fa , int l , int r , int k) { // 求 k 小…
    if(l >= r) return l ;
    int x = sum[L[a]] + sum[L[b]] - sum[L[lca]] - sum[L[lca_fa]] ;
    int mid = l + r >> 1 ;
    if(x >= k) return Query(L[a] , L[b] , L[lca] , L[lca_fa] , l , mid , k) ;
    else return Query(R[a] , R[b] , R[lca] , R[lca_fa] , mid + 1 , r , k - x) ;
}

inline void Dfs(int u) {
    size[u] = 1 ;
    Update(rt[fa[u]] , rt[u] , 1 , m , a[u]) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        fa[v] = u  ;
        d[v] = d[u] + 1 ;
        Dfs(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
}
inline void Dfs2(int u , int t) {
    top[u] = t ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
    }
}
inline int Lca(int x , int y) {
    int fx = top[x] ;
    int fy = top[y] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(d[x] > d[y]) swap(x , y) ;
    return x ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() , q = read() ;
    for(register int i = 1 ; i <= n ; i ++) a[i] = read() ;
    for(register int i = 1 ; i <= n ; i ++) b[i] = a[i] ;
    sort(b + 1 , b + n + 1) ;
    m = unique(b + 1 , b + 1 + n) - b - 1 ;
    for(register int i = 1 ; i <= n ; i ++) 
        a[i] = lower_bound(b + 1 , b + m + 1 , a[i]) - b ;
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u = read() , v = read() ;
        Add(u , v) ;
        Add(v , u) ;
    }
    Dfs(1) ;
    Dfs2(1 , 1) ;
    int ans = 0 ;
    for(register int i = 1 ; i <= q ; i ++) {
        int x = read() ^ ans , y = read() ;
        int lca = Lca(x , y) ;
        int z = read() ;
        write(ans = b[Query(rt[x] , rt[y] , rt[lca] , rt[fa[lca]] , 1 , m , z)]) ;
    }
    return 0 ;
}

[Vani有约会]雨天的尾巴

对于这题是一个线段树合并的问题
线段树合并例题 -> sol
递归求解 合并线段树的信息。

//Isaunoya
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize("Ofast")
#include
using namespace std ;
inline int read() {
    register int x = 0 ;
    register int f = 1 ;
    register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
}
int st[105] ;
template < typename T > inline void write(T x , char c = '\n') {
    int tp = 0 ;
    if(x == 0) return (void) puts("0") ;
    if(x < 0) putchar('-') , x = -x ;
    for( ; x ; x /= 10) st[++ tp] = x % 10 ;
    for( ; tp ; tp --) putchar(st[tp] + '0') ;
    putchar(c) ;
}
//#define Online_Judge
int n , m ;
const int N = 1e5 + 10 ;
int lc[N << 6] ;
int rc[N << 6] ;
int rt[N << 6] ;
int mx[N << 6] , id[N << 6] ;
int cnt = 0 ;
inline void Push_up(int rt) {
    if(mx[lc[rt]] >= mx[rc[rt]]) {
        mx[rt] = mx[lc[rt]] ;
        id[rt] = id[lc[rt]] ;
    } else {
        mx[rt] = mx[rc[rt]] ;
        id[rt] = id[rc[rt]] ;
    }
}
inline int Merge(int a , int b , int  l , int r) {
    if(! a || ! b) return a | b ;
    if(l == r) {
        mx[a] += mx[b] , id[a] = l ;
        return a ;
    }
    int mid = l + r >> 1 ;
    lc[a] = Merge(lc[a] , lc[b] , l , mid) ;
    rc[a] = Merge(rc[a] , rc[b] , mid + 1 , r) ;
    return Push_up(a) , a ;
}
inline void Insert(int & p , int l , int r , int pos , int v) {
    if(! p) p = ++ cnt ;
    if(l == r) {
        id[p] = l ;
        mx[p] += v ;
        return ;
    }
    int mid = l + r >> 1 ;
    if(pos <= mid) Insert(lc[p] , l , mid , pos , v) ;
    else Insert(rc[p] , mid + 1 , r , pos , v) ;
    Push_up(p) ;
    return ;
}
struct node {
    int v ;
    int nxt ;
};
int tot = 0 ;
int head[N] ;
node e[N << 1] ;
inline void Add(int u , int v) {
    e[++ tot].v = v ;
    e[tot].nxt = head[u] ;
    head[u] = tot ;
    return ;
}
int top[N] , fa[N] ;
int d[N] , idx[N] ;
int size[N] , son[N] ;
int Idx = 0 ;
inline void Dfs1(int u) {
    size[u] = 1 ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v == fa[u]) continue ;
        fa[v] = u ;
        d[v] = d[u] + 1 ;
        Dfs1(v) ;
        size[u] += size[v] ;
        if(size[v] > size[son[u]]) son[u] = v ;
    }
}
inline void Dfs2(int u , int t) {
    idx[u] = ++ Idx ;
    top[u] = t ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt) {
        int v = e[i].v ;
        if(v ^ fa[u] && v ^ son[u]) Dfs2(v , v) ;
    }
}
inline int GetLca(int x , int y) {
    int fx = top[x] ;
    int fy = top[y] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(fx , fy) , swap(x , y) ;
        x = fa[fx] ;
        fx = top[x] ;
    }
    if(d[x] > d[y]) swap(x , y) ;
    return x ;
}
int ans[N] ;
inline void GetAns(int u , int fa) {
    for(register int i = head[u ] ; i ; i  = e[i].nxt) {
        int v = e[i].v ;
        if(v == fa) continue ;
        GetAns(v , u) ;
        rt[u] = Merge(rt[u] , rt[v] , 1 , N) ;
    }
    ans[u] = id[rt[u]] ;
    if(mx[rt[u]] == 0) ans[u] = 0 ;
}
signed main() {
#ifdef Online_Judge
    freopen("testdata.in" , "r" , stdin) ;
    freopen("testdata2.out" , "w" , stdout) ;
#endif
    n = read() ;
    m = read() ;
    for(register int i = 1 ; i <= n - 1 ; i ++) {
        int u , v ;
        u = read() , v = read() ;
        Add(u , v) ;
        Add(v , u) ;
    }
    Dfs1(1) ;
    Dfs2(1 , 1) ;
    for(register int i = 1 ; i <= m ; i ++) {
        int x = read() , y = read() , z = read() ;
        int lca = GetLca(x , y) ;
        Insert(rt[lca] , 1 , N , z , - 1) ;
        Insert(rt[fa[lca]] , 1 , N , z , - 1) ;
        Insert(rt[x] , 1 , N , z , 1) ;
        Insert(rt[y] , 1 , N , z , 1) ;
    }
    GetAns(1 , 0) ;
    for(register int i = 1 ; i <= n ; i ++) write(ans[i]) ;
    return 0 ;
}
  • 这里还有一个重要操作:换根
  • 换根是一个比较重要的操作 "换根"不一定要重新DFS以及建树 只需要根据根所在的位置 和 查询的位置 分类讨论就可以了。
  • 但是你需要求一个 k - 1 级祖先(也就是最上面的那个的儿子节点) 所以需要一种倍增的想法

[BZOJ3083][遥远的国度]

//Isaunoya
#include
using namespace std ;
#define int long long
#define fi first
#define se second
#define pb push_back
inline int read() {
    register int x = 0 , f = 1 ;
    register char c = getchar() ;
    for( ; ! isdigit(c) ; c = getchar()) if(c == '-') f = -1 ;
    for( ; isdigit(c) ; c = getchar()) x = (x << 1) + (x << 3) + (c & 15) ;
    return x * f ;
}
template < typename T > inline bool cmax(T & x , T y) {
    return x < y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cmin(T & x , T y) {
    return x > y ? (x = y) , 1 : 0 ;
}
template < typename T > inline bool cabs(T & x) {
    return x > 0 ? 1 : (x = - x) , 0 ;
}
inline int QP(int x , int y , int Mod) {
    int ans = 1 ;
    for( ; y ; y >>= 1 , x = (x * x) % Mod)
        if(y & 1) ans = (ans * x) % Mod ;
    return ans ;
}
struct node {
    int v , nxt ;
} ;
int n , m ;
const int N = 100000 + 5 ;
int a[N] ;
node e[N<<1];int head[N],cnt=0;
inline void Add(int u , int v){
    e[++cnt].v = v ;
    e[cnt].nxt = head[u] ;
    head[u] = cnt ;
}
int fa[N][22], size[N] , son[N] , d[N] ;
inline void Dfs(int u) {
    size[u] = 1 ;
    for(register int i = 1 ; i <= 20 ; i ++) { // 倍增预处理
        fa[u][i] = fa[fa[u][i - 1]][i - 1] ;
        if(! fa[u][i]) break ;
    }
    for(register int i = head[u] ; i ; i = e[i].nxt){
        int v = e[i].v ;
        if(v == fa[u][0]) continue ;
        fa[v][0] = u ;
        d[v] = d[u] + 1 ;
        Dfs(v) ;
        size[u] += size[v] ;
        if(size[son[u]] < size[v]) son[u] = v ;
    }
}
int id[N] , idx = 0 ;
int fst[N] , top[N] ;
inline void Dfs2(int u , int t) {
    id[u] = ++ idx ;
    fst[idx] = a[u] ;
    top[u] = t ;
    if(! son[u]) return ;
    Dfs2(son[u] , t) ;
    for(register int i = head[u] ; i ; i = e[i].nxt){
        int v = e[i].v ;
        if((v ^ fa[u][0]) && (v ^ son[u])) Dfs2(v , v) ;
    }
    return ;
}
int mn[N << 2] ;
int tag[N << 2] ;
inline void build(int l , int r , int rt) {
    tag[rt] = 0 ;
    if(l == r) { mn[rt] = fst[l] ; return ; }
    int mid = l + r >> 1 ;
    build(l , mid , rt << 1) ;
    build(mid + 1 , r , rt << 1 | 1) ;
    mn[rt] = min(mn[rt << 1] , mn[rt << 1 | 1]) ;
}
inline void Push_down(int rt) {
    if(! tag[rt]) return ;
    mn[rt << 1] = mn[rt << 1 | 1] = tag[rt << 1] = tag[rt << 1 | 1] = tag[rt] ;
    return (void)(tag[rt]=0) ;
}
inline void Change(int a , int b , int  l, int r , int rt , int val) {
    if(a <= l && r <= b) {
        mn[rt] = tag[rt] = val ;
        return ;
    } Push_down(rt) ;
    int mid = l + r >> 1 ;
    if(a <= mid) Change(a , b , l , mid , rt << 1 , val) ;
    if(b > mid) Change(a , b , mid + 1 , r , rt << 1 | 1 , val) ;
    mn[rt] = min(mn[rt << 1] , mn[rt << 1 | 1]) ;
}
inline int Query(int a , int b , int  l, int r , int rt) {
    if(a <= l && r <= b) return mn[rt] ;
    Push_down(rt) ;
    int mid = l + r >> 1 ;
    int ans = INT_MAX ;
    if(a <= mid) cmin(ans , Query(a , b , l , mid , rt << 1 )) ;
    if(b > mid) cmin(ans , Query(a , b , mid + 1 , r , rt << 1 | 1)) ;
    return ans ;
}
inline void Change_Range(int x , int y , int val) {
    int fx = top[x] , fy = top[y] ;
    while(fx ^ fy) {
        if(d[fx] < d[fy]) swap(x , y) , swap(fx , fy) ;
        Change(id[fx] , id[x] , 1 , n , 1 , val) ;
        x = fa[fx][0] ;
        fx = top[x] ;
    }
    if(d[x] > d[y]) swap(x , y) ;
    Change(id[x] , id[y] , 1 , n , 1 , val) ;
}
inline int getfa(int x , int k) { // k 级祖先
    for(register int i = 20 ; i >= 0 ; i --)
        if(k >= (1 << i)) k ^= (1 << i) , x = fa[x][i] ;
    return x ;
}
int root = 1 ;
inline void getans(int x) { // 分类讨论
    if(x == root) {//与根相同就是全局最大值即mn[1]
        printf("%lld\n" , mn[1]) ;
        return ;
    }
    int y = 0 ;
    int p , q ;
    if(d[x] < d[root] && fa[y = getfa(root , d[root] - d[x] - 1)][0] == x) {
        p = Query(1 , id[y] - 1 , 1 , n , 1) ;
        if(id[y] + size[y] <= n) q = Query(id[y] + size[y] , n , 1 , n , 1) ;
        else q = INT_MAX ;
        printf("%lld\n" , min(p , q)) ;
    }
    else printf("%lld\n" , Query(id[x] , id[x] + size[x] - 1 , 1 , n , 1)) ; // 子树中
}
signed main() {
    n = read() ; m = read() ;
    for(register int i = 1 ; i < n ; i ++) {
        int u = read() , v = read() ;
        Add(u , v) ;
        Add(v , u) ;
    }
    for(register int i = 1 ; i <= n ; i ++) a[i] = read() ;
    Dfs(1) ;
    Dfs2(1 , 0) ;
    root = read() ;
    build(1 , n , 1) ;
    for(register int i = 1 ; i <= m ; i ++) {
        int opt = read() ;
        if(opt == 1) {
            root = read() ;
        }
        if(opt == 2) {
            int x = read() , y = read() , val = read() ;
            Change_Range(x , y , val) ;
        }
        if(opt == 3) {
            int x = read() ;
            getans(x) ;
        }
    }
    return 0 ;
}

你可能感兴趣的:(树链剖分(从入门到入土。))