AcWing 352. 闇の連鎖（公共祖先+树上差分）

题目

传说中的暗之连锁被人们称为 Dark。

Dark 是人类内心的黑暗的产物，古今中外的勇者们都试图打倒它。

经过研究，你发现 Dark 呈现无向图的结构，图中有 N 个节点和两类边，一类边被称为主要边，而另一类被称为附加边。

Dark 有 N – 1 条主要边，并且 Dark 的任意两个节点之间都存在一条只由主要边构成的路径。

另外，Dark 还有 M 条附加边。

你的任务是把 Dark 斩为不连通的两部分。

一开始 Dark 的附加边都处于无敌状态，你只能选择一条主要边切断。

一旦你切断了一条主要边，Dark 就会进入防御模式，主要边会变为无敌的而附加边可以被切断。

但是你的能力只能再切断 Dark 的一条附加边。

现在你想要知道，一共有多少种方案可以击败 Dark。

注意，就算你第一步切断主要边之后就已经把 Dark 斩为两截，你也需要切断一条附加边才算击败了 Dark。

输入格式
第一行包含两个整数 N 和 M。

之后 N – 1 行，每行包括两个整数 A 和 B，表示 A 和 B 之间有一条主要边。

之后 M 行以同样的格式给出附加边。

输出格式
输出一个整数表示答案。

数据范围
N≤100000,M≤200000,数据保证答案不超过231−1
输入样例：
4 1
1 2
2 3
1 4
3 4
输出样例：
3

思路

• 这道题目题意比较绕
  • 一颗n−1条主要边的树,然后增加了m条附加边. 我们只能删除一条主要边,一条附加边,一种边叫做主要边,一种边叫做附加边.要求删除两条边后,这棵树不再是连通的.我们需要统计,有多少种方案可以使得不连通,输出方案数.
• 我们发现其实如果砍掉一条主要边，如果它能和某一条附加边组成环的话，那么那条附加边在没有这条主要边后就可以起到主要边的作用，使得树仍然连通，所以附加边的作用就是相当与使能和他组成环的那些主要边多了一条替代边，那么我们可以统计一些每条主要边的替代边个数
• 如果一条主要边没有替代边，那么就可以有m中方案，因为只要看这条边，另一刀可以随便砍，甚至不砍
• 如果一条主要边只有一条替代边，那么就说明砍这个边可以把图分为两半
• 这里计算主要边的替代边的时候就可以用到树上差分的做法了，把附加边的两个点a，b的值+1，由于是树，所以环上肯定有a，b的公共祖先，并且要到达a，b必须经过他们的公共祖先，所以可以使得他们祖先值-2，最后dfs一下，判断到某一条边的时候它下面点的总值是多少即可。

代码

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

using namespace std;

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

int n, m;
int h[N], e[M], ne[M], idx;
int depth[N], fa[N][17];
int d[N];
int q[N];
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;
    int hh = 0, tt = 0;
    q[0] = 1;

    while (hh <= tt)
    {
        int t = q[hh ++ ];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if (depth[j] > depth[t] + 1)
            {
                depth[j] = depth[t] + 1;
                q[ ++ tt] = 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];
}

int dfs(int u, int father)
{
    int res = d[u];
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j != father)
        {
            int s = dfs(j, u);
            if (s == 0) ans += m;//如果有一条边为0，那么可以在砍这条边的同时，去掉任意一条其他边或者不砍
            else if (s == 1) ans ++ ;//如果该边为1，就表明要通过去掉该边达到题目要求，就一定要去一些附加边
            res += s;
        }
    }

    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    memset(h, -1, sizeof h);
    for (int i = 0; i < n - 1; i ++ )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }

    bfs();

    for (int i = 0; i < m; i ++ )
    {
        int a, b;
        scanf("%d%d", &a, &b);
        int p = lca(a, b);
        d[a] ++, d[b] ++, d[p] -= 2;
    }
    dfs(1, -1);
    printf("%d\n", ans);

    return 0;
}