题意:判断最小生成树是否唯一
转化为次小生成树做法
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=110;
const int INF=0x3f3f3f3f;
int vis[maxn];
int ans;
int lowc[maxn];
int pre[maxn];
int Max[maxn][maxn];//Max[i][j]表示在最小生成树中从i到j的路径中的最大边权
bool used[maxn][maxn];
int cost[maxn][maxn];
void prim(int n){
ans=0;
memset(vis,0,sizeof(vis));
memset(Max,0,sizeof(Max));
memset(used,false,sizeof(used));
vis[1]=1;
pre[1]=-1;
for(int i=2;i<=n;i++){
lowc[i]=cost[1][i];
pre[i]=1;
}
lowc[1]=0;
for(int i=2;i<=n;i++){
int minc=INF;
int p=-1;
for(int j=1;j<=n;j++)
if(!vis[j]&&minc>lowc[j]){
minc=lowc[j];
p=j;
}
if(minc==INF)
return ;
ans+=minc;
vis[p]=1;
used[p][pre[p]]=used[pre[p]][p]=true;
for(int j=1;j<=n;j++){
if(vis[j])
Max[j][p]=Max[p][j]=max(Max[j][pre[p]],lowc[p]);
if(!vis[j]&&lowc[j]>cost[p][j]){
lowc[j]=cost[p][j];
pre[j]=p;
}
}
}
}
int main(){
int t,n,m;
int a[1101],b[1110],c[1101];
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
int k;
memset(cost,INF,sizeof(cost));
for(int i=1;i<=m;i++){
scanf("%d%d%d",&a[i],&b[i],&k);
c[i]=k;
cost[a[i]][b[i]]=cost[b[i]][a[i]]=min(cost[a[i]][b[i]],k);
}
prim(n);
int flag=0;
for(int i=1;i<=n;i++){
if(used[a[i]][b[i]])
continue;
else{
if(c[i]==Max[a[i]][b[i]]){
flag=1;
break;
}
}
}
if(flag==1){
printf("Not Unique!\n");
}
else
printf("%d\n",ans);
}
return 0;
}