POJ3417 Network暗的连锁 （树上差分）

树上的边差分，x++，y++，lca(x,y)-=2.

m条边可以看做将树上的一部分边覆盖，就用差分，x=1，表示x与fa(x)之间的边被覆盖一次，m次处理后跑一遍dfs统计子树和，每个节点子树和val=1，说明割去这条边后只有一种方案，val=0，说明割去后随便再割一条都行，有m中方案。

题很简单但是调了半天，发现倍增的数组开小了（开的20），以后防止这种情况开成25应该比较稳妥。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#define ll long long
using namespace std;
const int N = 1e5 + 10;
int head[N], to[N << 1], nxt[N << 1], tot;
void add(int x, int y) {
    nxt[++tot] = head[x];
    head[x] = tot;
    to[tot] = y;
}
int n, m, d[N], f[N][21], val[N];
ll ans = 0;

void dfs(int u, int fa) {
    for (int i = head[u]; i; i = nxt[i]) {
        int v = to[i];
        if (v == fa) continue;
        d[v] = d[u] + 1;
        f[v][0] = u;
        for (int j = 1; j <= 20; j++) 
            f[v][j] = f[f[v][j - 1]][j - 1];
        dfs(v,u); 
    }
}

int lca(int x, int y) {
    if (d[x] > d[y]) swap(x, y);
    for (int i = 20; i >= 0; i--)
        if (d[f[y][i]] >= d[x]) y = f[y][i];
    if (x == y) return x; 
    for (int i = 20; i >= 0; i--)
        if (f[x][i] != f[y][i]) x = f[x][i], y = f[y][i];
    return f[x][0];
}

void solve(int u) {
    for (int i = head[u]; i; i = nxt[i]) {
        int v = to[i];
        if (d[v] <= d[u]) continue;
        solve(v);
        val[u] += val[v];
    }
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i < n; i++) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }
    d[1] = 1, dfs(1,0);
    for (int i = 1; i <= m; i++) {
        int x, y; cin >> x >> y;
        int z = lca(x, y);
        val[x]++, val[y]++, val[z] -= 2;
    }
    solve(1);
    for (int i = 2;i <= n; i++) {//注意从2开始，val[2]表示2和1之间这条边被覆盖的次数
        if (val[i] == 0) ans += m;
        if (val[i] == 1) ans ++;
    }
    printf("%lld\n", ans);
    return 0;
}