闇の連鎖

题意

给定一棵$n$个节点的图，其中有$n-1$条树边，$m$条非树边。第一次可以删掉一条树边，第二次可以删掉一条非树边，使得这个图最后不连通。求方案数。

数据范围

$1\le n\le 100000$
$1\le m\le 200000$

思路

我们考察所有所有树边，如果切掉这条边的话，可以切哪些非树边，使得最后图不连通。如果这条树边在一个环中，那么只需要把构成这个环的那条非树边切掉就行。如果这条树边在两个环中，那么就需要切掉两条非树边。如果不在环中，那么不需要切非树边，即随意切一条非树边即可。
那么如何统计一条树边，在几个环中呢？对每一条非树边$(u,v)$，对$u,v$树上路径上的所有边$+1$，最后统计每条边被加了多少次即可。这一操作可以通过树上差分来进行。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>

using namespace std;

const int N = 100010, M = 2 * N;

int n, m;
int h[N], e[M], ne[M], idx;
int depth[N], fa[N][17];
int d[N];
queue<int> que;
int ans;

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

void bfs()
{
    memset(depth, 0x3f, sizeof(depth));
    depth[0] = 0, depth[1] = 1;
    que.push(1);
    while(que.size()) {
        int t = que.front();
        que.pop();
        for(int i = h[t]; ~i; i = ne[i]) {
            int j = e[i];
            if(depth[j] > depth[t] + 1) {
                depth[j] = depth[t] + 1;
                que.push(j);
                fa[j][0] = t;
                for(int k = 1; k <= 16; k ++) {
                    fa[j][k] = fa[fa[j][k-1]][k-1];
                }
            }
        }
    }
}

int lca(int a, int b)
{
    if(depth[a] < depth[b]) swap(a, b);
    for(int k = 16; k >= 0; k --) {
        if(depth[fa[a][k]] >= depth[b]) {
            a = fa[a][k];
        }
    }
    if(a == b) return a;
    for(int k = 16; k >= 0; k --) {
        if(fa[a][k] != fa[b][k]) {
            a = fa[a][k];
            b = fa[b][k];
        }
    }
    return fa[a][0];
}

void dfs_sum(int u, int fa)
{
    for(int i = h[u]; ~i; i = ne[i]) {
        int j = e[i];
        if(j == fa) continue;
        dfs_sum(j, u);
        d[u] += d[j];
    }
}

void dfs_ans(int u, int fa)
{
    for(int i = h[u]; ~i; i = ne[i]) {
        int j = e[i];
        if(j == fa) continue;
        if(d[j] == 0) ans += m;
        else if(d[j] == 1) ans += 1;
        dfs_ans(j, u);
    }
}

int main()
{
    memset(h, -1, sizeof(h));
    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);
    }
    bfs();
    for(int i = 1; i <= m; i ++) {
        int a, b;
        scanf("%d%d", &a, &b);
        int p = lca(a, b);
        d[a] ++, d[b] ++, d[p] -= 2;
    }
    dfs_sum(1, -1);
    dfs_ans(1, -1);
    printf("%d\n", ans);
    return 0;
}