HDU - 5452 - Minimum Cut（树形DP）

HDU - 5452 - Minimum Cut

题意：

有一颗生成树和一些边，将边加入生成树中得到新的图，求最少删多少条边，使得图分为两块，其中有且仅有一条边是属于最初的生成树的，其余的为后来加的边中的。

 

因为必须要在最初的生成树切一条边，而且只能切一条边，所以分离的那一部分一定是原生成树的其中一个子树。

我们就可以计算出所有子树分离出来要切去的边数，然后从中取出最小的即为答案。

然后就是如何计算每颗子树分离的代价。

子树分离的代价就是该子树中所有点连到不是该子树中的点的边数。

可以想到一个子树可以看作是一个父亲节点+它的所有子树构成的子树。

假设已经知道节点 i 所有子树的代价为sum，可以知道

新的子树的代价 = （节点 i 的出度 - 节点 i 连到其子孙节点的边数） + （sum - 节点 i 连到其子孙节点的边数）

 因为之前子树的代价会包含该节点到其子孙节点的边，但是当其构成了一个子树后，这些边不能算，所以要减去这些边

所以每个节点要存的有 出度、连到子孙节点的边数 和 以当前节点作为子树根节点的代价

首先就是要用前面给的 n-1 条边构建一颗生成树出来，然后才能进行接下来的操作。

如何计算每个点到其子树的边数可以利用建树时的 dfs 序号，找到每一个点的子树的dfs序号的区间（必然是连续的一段区间）

然后就能一条边一条边判断下去，然后更新边数。

 

最后再从下往上进行 DFS 求出每个子树的代价，要注意必须要删一条边，所以根节点不做考虑。

 

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <vector>
using namespace std;

typedef long long int LL;
const int maxn = 2e5 + 10;
int T, n, m, ans, x, y, cas;
int st[maxn], ed[maxn], sz; // st 存dfs序开始位置， ed 存结束位置
int e[maxn][2], sum[maxn]; // e[i][0] 存出度 e[i][1] 存到子孙节点的边数 sum 存代价
vector<int>to[maxn]; // 生成树的边
void dfs(int x,int pre){ // 建树 + 子树区间
    st[x] = ++sz;
    for(int i=0;i<to[x].size();i++){
        if(to[x][i] == pre) continue;
        dfs(to[x][i], x);
    }
    ed[x] = sz; // 回溯时的 sz 就是最后一个子孙节点
    e[x][1] += to[x].size(); // 该点连出去的边都是到儿子节点的（有一条是父亲节点）
    if(pre != 0) e[x][1] --; // 如果不是根节点就是有父亲节点，所以要 -1
}

void dfs2(int x,int pre){ // 求出所有结果
    sum[x] = e[x][0] - e[x][1] - e[x][1];
    for(int i=0;i<to[x].size();i++){
        int nt = to[x][i];
        if(nt == pre) continue;
        dfs2(nt, x); // 先求出下面的再求当前的
        sum[x] = sum[x] + sum[nt]; // 子树所有代价求和
    }
    if(pre!=0) ans = min(ans, sum[x]); // 不能是根节点，求最小值
}

int main()
{
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        memset(e, 0, sizeof e);
        for(int i=1;i<=n;i++) to[i].clear();
        for(int i=1;i<n;i++){
            scanf("%d%d",&x,&y);
            to[x].push_back(y);
            to[y].push_back(x);
            e[x][0] ++; e[y][0] ++; // 出度+1 ，无向边
        }
        sz = 0;
        dfs(1,0);
        for(int i=n;i<=m;i++){
            scanf("%d%d",&x,&y);
            e[x][0] ++; e[y][0] ++;
            if(st[x] <= st[y] && st[y] <= ed[x]) e[x][1] ++; // x 是 y 的祖先
            if(st[y] <= st[x] && st[x] <= ed[y]) e[y][1] ++; // y 是 x 的祖先
        }
        ans = m;
        dfs2(1,0);
        printf("Case #%d: %d\n", ++cas, ans);
    }
    return 0;
}