【图论】【生成树】最小树形图

default111

已于 2022-03-05 18:42:54 修改

阅读量302

点赞数

分类专栏：算法竞赛文章标签：图论算法 c++

于 2021-11-20 14:37:48 首次发布

本文链接：https://blog.csdn.net/default012/article/details/121439182

版权

算法竞赛专栏收录该内容

31 篇文章 3 订阅

订阅专栏

最小树形图

有向图的最小生成树，从根节点能到所有节点，且每条边权值和最小。
朱刘算法（Edmonds 算法）， $O (n m)$ 。
模板题：给定包含n个结点，m条有向边的一个图。输出以结点r为根的最小树形图每条边的权值之和，如果没有以r为根的最小树形图，输出-1。
步骤
- 统计每个点的最小入边，加入答案
- 如果有一个点找不到入边，说明不能生成最小树形图，返回-1
- 每个点由最小入边找环缩点
- 其余点不变，重构图，边权修改为原边权-到达节点最小入边

int in[N], pre[N]; //in[i]记录每轮i点入边中最小的边权，pre[i]记录最小边权对应的端点编号
int vst[N], scc[N];
int Zhuliu()
{
	int ans = 0;
	while (1)
	{
		for (int i = 1; i <= n; i++)
			in[i] = INF; // 初始化
		for (int i = 1; i <= m; i++)
		{
			int u = edge[i].from, v = edge[i].to;
			if (u != v && edge[i].val < in[v]) // 遍历所有边，对每个点找到最小的入边
				in[v] = edge[i].val, pre[v] = u;
		}
		for (int i = 1; i <= n; i++)	   // 判定无解
			if (i != root && in[i] == INF) //有一个点找不到入边了
				return -1;
		int sc = 0;
		for (int i = 1; i <= n; i++)
			vst[i] = scc[i] = 0;
		for (int i = 1; i <= n; i++)
		{
			if (i == root)
				continue;
			ans += in[i];
			int v = i;
			while (vst[v] != i && !scc[v] && v != root) // 找环，找回自己/找到一个缩过点的/找到根节点了都退出
			{
				vst[v] = i;
				v = pre[v];
			}
			if (!scc[v] && v != root) //一定是找回自己
			{
				scc[v] = ++sc; // 缩点
				for (int u = pre[v]; u != v; u = pre[u])
					scc[u] = sc;
			}
		}
		if (sc == 0)
			break; // 无环，得到解
		for (int i = 1; i <= n; i++)
			if (!scc[i])
				scc[i] = ++sc;
		for (int i = 1; i <= m; i++) //重构图
		{
			int u = edge[i].from, v = edge[i].to;
			edge[i].from = scc[u], edge[i].to = scc[v];
			if (scc[u] != scc[v])
				edge[i].val -= in[v]; // 修改边权
		}
		root = scc[root];
		n = sc;
	}
	return ans;
}