首页 > 代码库 > HDU 3367 Pseudoforest 最大生成树

HDU 3367 Pseudoforest 最大生成树

题目来源:HDU 3367 Pseudoforest

题意:每个连通块最多可以有一个环 求最大的森林

思路:考虑最大生成树 如果祖先一样没有环 那就合并 如果祖先不一样 如果2棵树都没有环 合并 如果有1棵树有环 合并 标记该棵树有环

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 100010;
struct edge
{
	int u, v, w;
}e[maxn];
int f[10010];
int vis[maxn];
int ok[10010];
int n, m;
bool cmp(edge a, edge b)
{
	return a.w > b.w;
}
int find(int x)
{
	if(x != f[x])
		return f[x] = find(f[x]);
	return f[x];
}

int MST()
{
	int sum = 0;
	memset(vis, 0, sizeof(vis));
	memset(ok, 0, sizeof(ok));
	for(int i = 0; i < m; i++)
	{
		int x = find(e[i].u);
		int y = find(e[i].v);
		if(x != y)
		{
			if(!ok[x] && !ok[y])
			{
				sum += e[i].w;
				f[x] = y;	
			}
			else if(!ok[x] || !ok[y])
			{
				ok[x] = ok[y] = 1;
				f[x] = y;
				sum += e[i].w;
			}
		}
		else if(!ok[x])
		{
			ok[x] = 1;
			sum += e[i].w;
		}
	}
	
	return sum;
}

int main()
{
	while(scanf("%d %d", &n, &m) && (n+m))
	{
		for(int i = 0; i < n; i++)
			f[i] = i;
		for(int i = 0; i < m; i++)
		{
			scanf("%d %d %d", &e[i].u, &e[i].v, &e[i].w);
		}
		sort(e, e+m, cmp);
		printf("%d\n", MST());
	}
	return 0;
}