算法竞赛进阶指南 树上差分 闇の連鎖

传说中的暗之连锁被人们称为 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≤20000，数据保证答案不超过2 ^ 31−1

输入样例：

4 1
1 2
2 3
1 4
3 4

输出样例：

3

 难度：困难时/空限制：1s / 64MB总通过数：3306总尝试数：7599来源：《算法竞赛进阶指南》算法标签：

图论    树上差分 LCA

题意大概是，先切一条主要边，然后再切一条此要边求由多少种切法使得树不连通

解题思路：
我们先找性质：

由题意可得主要边组成的树是最小生成树。

如图绿色的边是主要边，红色的边是次要边，边上的数字表示切几刀它会分为两段

 

假设以蓝色的环为例，要切将树分为两段则需要在环上的主要边切一刀，然后在此要边上切一刀此树才可被分为两段。若再加一条次要边如图：

则两条主要的边重合的部分要切两次，且如果切需要切两次的主要边则此要的边也要切两次 

因此我们可以先加入每条树边，然后枚举每条次要边，若遇到他们重合的地方则主要边所需切的次数加一。

因为最后我们求得是方案，将图上所有的主要边的权值全部累加即为最终方案（因为可以在每个绿色的边上都切一刀，所以加起来即最终方案）

如何在他们重合的地方加一这个可以用树上差分来做。

 

如图：假设先插入的是a, b这条边p为a, b的最近公共祖先，假设d为差分数组 ，d[a] ++ 最终会导致以节点1，2， 4， 5，8为根的子树和加一。这里边的权值记录的是以某个节点为根的子树所能切的刀数。d[b] ++ 则会使以1, 3, 6节点为根节点的子树和加一。然后让他们的最近公共祖先即d[1] -= 2即不会影响其他上面的点。

我们再加一条边解释。

 

 如图：在5， 6加一条次要边，会导致如图情况。因为d是差分数组，所以把以p为根节点的所有子树的d数组加起来的值即是这个点p所在的子树能切的刀数

设方案数为ans。

假设他们加起来的值为s

若s等于一说明可以切一刀，即ans++

若s大于1说明要切s刀，直接忽略

若s等于0说明子树的所有次要边都可以被切即ans + m

这道题要解释清楚有点难。菜鸟尽力了。。。。

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

using namespace std;

const int N = 100010, M = 400010;

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

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

void bfs()
{
    memset(depth, 0x3f, sizeof depth);
    
    queue<int> q;
    depth[0] = 0, depth[1] = 1;
    q.push(1);
    
    while (q.size())
    {
        int t = q.front();
        q.pop();
        
        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if (depth[j] > depth[t] + 1)
            {
                depth[j] = depth[t] + 1;
                fa[j][0] = t;
                q.push(j);
                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 fa)
{
    int res = d[u];
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (j == fa) continue;
        int s = dfs(j, u);
        
        if (s == 0) ans += m;//三种情况
        else if (s == 1) ans ++ ;
        
        res += s;
    }
    
    return res;
}

int main()
{
    cin >> 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();//预处理fa数组
    
    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);
    
    cout << ans << endl;
    
    return 0;
}