HDU 1879 继续畅通工程

地址

已经有些边连通了，用并查集的话就是先把这些边union了（记录conn，不加opt），再把未确定的边排序然后挨个union；用prim的话把这些边权值赋为0。

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;

const int MAXN = 1e2 + 5;
int N;
int pre[MAXN];
int conn;
int opt;

struct Edge
{
	int n1, n2, w;
	bool operator<(const Edge& e) const
	{
		return w < e.w;
	}
};
vector<Edge> v;

int f(int x)
{
	int f0 = x, f1 = x;
	for (; pre[f0] > 0;)
	{
		f0 = pre[f0];
	}
	for (; pre[f1] > 0;)
	{
		int t = f1;
		f1 = pre[f1];
		pre[t] = f0;
	}
	return f0;
}

bool u(int a, int b)
{
	int f1 = f(a);
	int f2 = f(b);
	if (f1 != f2)
	{
		conn++;
		if (pre[f1] <= pre[f2])
		{
			pre[f1] += pre[f2];
			pre[f2] = f1;
		}
		else
		{
			pre[f2] += pre[f1];
			pre[f1] = f2;
		}
		return true;
	}
	return false;
}

int main()
{
	int a, b, c, d;
	for (; ~scanf("%d", &N);)
	{
		if (N == 0) return 0;
		v.clear();
		memset(pre, -1, sizeof pre);
		conn = 0;
		opt = 0;
		for (int i = 0; i < N*(N - 1) / 2; i++)
		{
			scanf("%d%d%d%d", &a, &b, &c, &d);
			if (d) u(a, b);
			else v.push_back({ a,b,c });
		}
		sort(v.begin(), v.end());
		for (int i = 0; i < v.size(); i++)
		{
			if (conn == N - 1) break;
			if (u(v[i].n1, v[i].n2))
				opt += v[i].w;
		}
		printf("%d\n", opt);
	}
    return 0;
}