USACO 2015 December Contest, Platinum Problem Max Flow【树链剖分】

　　题意比较难理解，就是给你n个点的树，然后给你m个修改操作，每一次修改包括一个点对(x, y)，意味着将x到y所有的点权值加一，最后问你整个树上的点权最大是多少。

　　

　　比较裸的树链剖分了，感谢Haild的讲解。

　　首先第一遍dfs预处理出size，son(重儿子)。

　　第二遍dfs重编号。

　　然后线段树就可以了。

 

　　感觉就是把一棵树弄成一条一条的链，新奇的hash方法。

  1 #include <bits/stdc++.h>
  2 #define rep(i, a, b) for (int i = a; i <= b; i++)
  3 #define drep(i, a, b) for (int i = a; i >= b; i--)
  4 #define REP(i, a, b) for (int i = a; i < b; i++)
  5 #define pb push_back
  6 #define mp make_pair
  7 #define clr(x) memset(x, 0, sizeof(x))
  8 #define xx first
  9 #define yy second
 10 using namespace std;
 11 typedef long long i64;
 12 typedef pair<int, int> pii;
 13 const int inf = ~0U >> 1;
 14 const i64 INF = ~0ULL >> 1;
 15 //*******************************
 16 
 17 template <typename T> void MAX(T &a, T &b) { if (a < b) a = b; }
 18 template <typename T> void MIN(T &a, T &b) { if (a > b) a = b; }
 19 
 20 const int maxn = 50005;
 21 
 22 struct Ed {
 23     int u, v, nx; Ed() {}
 24     Ed(int _u, int _v, int _nx) :
 25         u(_u), v(_v), nx(_nx) {}
 26 } E[maxn << 1];
 27 int G[maxn], edtot;
 28 void addedge(int u, int v) {
 29     E[++edtot] = Ed(u, v, G[u]);
 30     G[u] = edtot;
 31     E[++edtot] = Ed(v, u, G[v]);
 32     G[v] = edtot;
 33 }
 34 
 35 struct Seg_Tree {
 36     int lazy[maxn << 2], Max[maxn << 2];
 37     void Push_down(int o) {
 38         if (!lazy[o]) return;
 39         lazy[o << 1] += lazy[o], lazy[o << 1 | 1] += lazy[o];
 40         Max[o << 1] += lazy[o], Max[o << 1 | 1] += lazy[o];
 41         lazy[o] = 0;
 42     }
 43     void Push_up(int o) { Max[o] = max(Max[o << 1], Max[o << 1 | 1]); }
 44     void update(int o, int l, int r, int ql, int qr, int v) {
 45         if (ql <= l && r <= qr) {
 46             lazy[o] += v;
 47             Max[o] += v;
 48             return;
 49         }
 50         Push_down(o);
 51         int mid = l + r >> 1;
 52         if (ql <= mid) update(o << 1, l, mid, ql, qr, v);
 53         if (qr > mid) update(o << 1 | 1, mid + 1, r, ql, qr, v);
 54         Push_up(o);
 55     }
 56 } T;
 57 
 58 int size[maxn], pre[maxn], son[maxn], dep[maxn];
 59 int dfs_size(int x, int fa) {
 60     size[x] = 1; int haha = -inf; pre[x] = fa; dep[x] = dep[fa] + 1;
 61     for (int i = G[x]; i; i = E[i].nx) if (E[i].v != fa) {
 62         size[x] += dfs_size(E[i].v, x);
 63         if (size[E[i].v] > haha) haha = size[E[i].v], son[x] = E[i].v;
 64     }
 65     return size[x];
 66 }
 67 int ndtot, pos[maxn], top[maxn];
 68 void repos(int x, int tp) {
 69     pos[x] = ++ndtot;
 70     top[x] = tp;
 71     if (son[x]) repos(son[x], tp);
 72     for (int i = G[x]; i; i = E[i].nx) if (E[i].v != pre[x] && E[i].v != son[x]) repos(E[i].v, E[i].v);
 73 }
 74 
 75 int n;
 76 void update(int x, int y) {
 77     while (top[x] != top[y]) {
 78         if (dep[top[x]] < dep[top[y]]) swap(x, y);
 79         T.update(1, 1, n, pos[top[x]], pos[x], 1); x = pre[top[x]];
 80     }
 81     T.update(1, 1, n, min(pos[x], pos[y]), max(pos[x], pos[y]), 1);
 82 }
 83 
 84 int main() {
 85     freopen("maxflow.in", "r", stdin);
 86     freopen("maxflow.out", "w", stdout);
 87     int m; scanf("%d%d", &n, &m);
 88     REP(i, 1, n) {
 89         int x, y; scanf("%d%d", &x, &y);
 90         addedge(x, y);
 91     }
 92     dfs_size(1, 1);
 93     repos(1, 1);
 94     while (m--) {
 95         int x, y; scanf("%d%d", &x, &y);
 96         update(x, y);
 97     }
 98     printf("%d\n", T.Max[1]);
 99     return 0;
100 }

View Code