【图论】POJ1679 - The Unique MST

题意:

求一个连通有权无向图的最小生成树是否唯一。若唯一,输出最小生成树的权值;若不唯一,输出"Not Unique!"。

思路:

求出最小生成树和次小生成树,判断权值是否相等,若相等,则不唯一。

最小生成树可用Prim算法求出,设权值为w(MST)。关于次小生成树的求法,可以在求完最小生成树之后,枚举不在MST中的每一条边e,设e的起点为u、终点为v,权值为c,则另一条生成树Ti的权值为w(MST) + c - M(u, v),其中M(u, v)为MST中u到v的最大权值边的权值,可以在Prim算法中同时求出。最后取所有Ti的最小值,则为次小生成树的权值。

代码:

#include 
#include 
#include 
#include 
using namespace std;

int N, M;
int G[105][105];
bool T[105][105];
int LEN[105][105];

struct Edge {
	int u, v, w;
	Edge(): u(-1), v(-1), w(0) {
	}
	Edge(int u, int v, int w): u(u), v(v), w(w) {
	}
};

struct Edge_Greater {
	bool operator () (const Edge lhs, const Edge rhs) const {
		return lhs.w > rhs.w;
	}
};

priority_queue, Edge_Greater> Queue;
int FL[105];

int Kruskal() {
	for (int i=1; i<=N; i++) {
		FL[i] = i;
	}
	memset(LEN, 0, sizeof(LEN));
	memset(T, 0, sizeof(T));
	int ans = 0;
	int cnt = 0;
	while (!Queue.empty() && cnt < N-1) {
		Edge e = Queue.top(); Queue.pop();
		int u = e.u; int v = e.v; int w = e.w; 
		if (FL[u] == FL[v]) {
			continue;
		}
		ans += w;
		T[u][v] = T[v][u] = true;
		int flu = FL[u]; int flv = FL[v];
		for (int i=1; i<=N; i++) {
			if (FL[i] == flu) {
				for (int j=1; j<=N; j++) {
					if (FL[j] == flv) {
						LEN[i][j] = LEN[j][i] = w;
					}
				} 
			}
		}
		for (int i=1; i<=N; i++) {
			if (FL[i] == flv) {
				FL[i] = flu;
			}
		}
	}
	return ans;
}

int main() {
	int K;
	scanf("%d", &K);
	while (K--) {
		scanf("%d%d", &N, &M);
		memset(G, 0x3F, sizeof(G));
		while (!Queue.empty())
			Queue.pop();
		for (int i=0; i

心得:

学会了次小生成树的求法。

你可能感兴趣的:(ACM)