POJ 1679 The Unique MST 次小生成树

http://poj.org/problem?id=1679

题意：给出N个点和M条边，判断最小生成树是否唯一。

分析：次小生成树基本的思想就是连入一条不在最小生成树上的边，从而形成一个环，去掉在环中并且在最小生成树上最大的边计算权值，遍历所有不在最小生成树上的边并进行同样的操作最小值即为次小生成树。

prime算法：先求出最小生成树，并用数组Max[i][j]记录生成树上任意两点间权值最大的边的权值，最后枚举不再MST中的边（i，j），判断权值map[i][j]是否和Max[i][j]相等，若有边满足，则MST不唯一（可以代替）。

’

#include<iostream>
#include<algorithm>

using namespace std;
const int maxn = 105,inf=1000000;
int map[maxn][maxn], lowcost[maxn];
int pre[maxn];//记录在生成树中的前驱
int Max[maxn][maxn];//记录i,j之间最大边权
int n, m;
bool vis[maxn];

int prim()
{
	int ans = 0;
	memset(vis, false, sizeof(vis));
	memset(Max, 0, sizeof(Max));
	for (int i = 2; i <= n; i++)
	{
		lowcost[i] = map[1][i];
		pre[i] = 1;
	}
	lowcost[1] = 0; vis[1] = true;
	for (int i = 1; i < n; i++)
	{
		int temp = inf, k = 0;
		for (int j = 2; j <= n; j++)
		{
			if (!vis[j] && temp > lowcost[j])
			{
				temp = lowcost[k = j];
			}
		}
		if (temp == inf) return -1;
		vis[k] = true;
		ans += lowcost[k];
		for (int j = 1; j <= n; j++)
		{
			if (vis[j])//若j已加入MST,更新Max
				Max[k][j] = Max[j][k] = max(Max[j][pre[k]], lowcost[k]);
			else if (lowcost[j] > map[k][j])
			{
				lowcost[j] = map[k][j];
				pre[j] = k;
			}
		}
	}
	return ans;
}


int main()
{
	int T;
	cin >> T;
	while (T--)
	{
		int ans;
		cin >> n >> m;
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= n; j++)
				map[i][j] = inf;
		for (int i = 0; i < m; i++)
		{
			int u, v, w;
			scanf("%d%d%d", &u, &v, &w);
			map[u][v] = map[v][u] = w;
		}
		ans=prim();

		if (ans == -1) 
			cout << "Not Unique!" << endl;
		else
		{
			int minn = inf;
			for (int i = 1; i <= n; i++)
				for (int j = 1; j <i; j++)
				{
					if (map[i][j]!=inf&&i != pre[j] && j != pre[i])//（i,j）不再MST中
						minn = min(minn, ans + map[i][j] - Max[i][j]);//map[i][j]==Max[i][j] 不唯一
				}

			if (minn == ans)
				cout << "Not Unique!" << endl;
			else cout << ans << endl;
		}
	}
	return 0;
}