#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 111;
const int inf = 0x3f3f3f3f;
int Map[maxn][maxn];//邻接矩阵存图
int Max[maxn][maxn];//表示最小生成树中i到j的最大边权
bool used[maxn][maxn];//判断该边是否加入最小生成树
int pre[maxn];
int dis[maxn];
void init(int n)
{
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
if (i == j) Map[i][j] = 0;
else Map[i][j] = inf;
}
void read(int m)
{
int u, v, w;
for (int i = 0; i < m; i++)
{
scanf("%d %d %d", &u, &v, &w);
Map[u][v] = Map[v][u] = w;
}
}
int prim(int n)
{
int ans = 0;
bool vis[maxn];
memset(vis, false, sizeof(vis));
memset(used, false, sizeof(used));
memset(Max, 0, sizeof(Max));
for (int i = 2; i <= n; i++)
{
dis[i] = Map[1][i];
pre[i] = 1;
}
pre[1] = 0;
dis[1] = 0;
vis[1] = true; //不要忘记初始化1
for (int i = 2; i <= n; i++)
{
int min_dis = inf, k; //因为一开始有的边是inf所以这里的最小值要弄成inf
for (int j = 1; j <= n; j++)
{
if (!vis[j] && min_dis > dis[j])
{
min_dis = dis[j];
k = j;
}
}
if (min_dis == inf) return -1;//如果不存在最小生成树
ans += min_dis;
vis[k] = true;
used[k][pre[k]] = used[pre[k]][k] = true;
for (int j = 1; j <= n; j++)
{
if (vis[j]&&used[j][k]==0) Max[j][k] = Max[k][j] = max(Max[j][pre[k]], dis[k]);//计算两点之间若无连接,若连接上构成回路这条回路中最大的一条边
if (!vis[j] && dis[j] > Map[k][j])
{
dis[j] = Map[k][j];
pre[j] = k;
}
}
}
return ans;//最小生成树的权值之和
}
int smst(int n, int min_ans)//min_ans 是最小生成树的权值和
{
int ans = inf;
for (int i = 1; i <= n; i++)//枚举最小生成树之外的边
for (int j = i + 1; j <= n; j++)
if (Map[i][j] != inf && !used[i][j])
ans = min(ans, min_ans + Map[i][j] - Max[i][j]);//假设构成回路,再打破回路
if (ans == inf) return -1;
return ans;
}
void solve(int n)
{
int ans = prim(n);
if (ans == -1)
{
puts("Not Unique!");
return;
}
if (smst(n, ans) == ans)
printf("Not Unique!\n");
else
printf("%d\n", ans);
}
int main()
{
int T, n, m;
scanf("%d", &T);
while (T--)
{
scanf("%d %d", &n, &m);
init(n);
read(m);
solve(n);
}
return 0;
}
POJ1679 The Unique MST
题意:给定图,让求它的最小生成树是否唯一。如果唯一的话输出最小生成树的权值和,否则输出Not Unique!
算法:
计算最小生成树,计算的同时,若新加入的点与已加入的点无连接,计算一下若连接上构成回路中的边的最大权值。
最后遍历所有没有被加入的边,将刚才计算的最大权值的边去掉,换成那个没有加入的边,这样就又可以打破回路,构成一个树。