这是个次小生成树的题,首先介绍一下什么事次小生成树,,,就是存在有两个及以上的最小生成树,,,,这一题用krusal TLE。。。因此选用了prim算法,,,,思想就是:如果存在一个点i它的dist【i】可以由两个或两个以上的其他点更新得到,那么最小生成树肯定不唯一了,,,,
法一:
#include<iostream> #include<cstdio> #define N 101 #define M 99999999 using namespace std; int n,m,ans; int map[N][N]; bool visit[N]; int change[N]; int dist[N]; void init_set() { ~scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) for(int j=1;j<=n;++j) if(i!=j) map[i][j]=M; else map[i][j]=0; for(int i=0;i!=m;++i) { int a,b,c; ~scanf("%d%d%d",&a,&b,&c); if(map[a][b]>c) map[a][b]=map[b][a]=c; } for(int i=1;i<=n;i++) dist[i]=M,visit[i]=true,change[i]=0; } bool prim() { ans=0; dist[1]=0; int now=0; for(int i=1;i<=n;++i) { int minx=M; for(int j=1;j<=n;++j) if(minx>dist[j]&&visit[j]) minx=dist[now=j]; visit[now]=false; if(!now) break; ans+=dist[now]; if(change[now]>1) return false; for(int j=1;j<=n;++j) if(visit[j]) if(dist[j]>map[now][j]) dist[j]=map[now][j],change[j]=1; else if(dist[j]==map[now][j]) change[j]++; } return true; } int main() { int Case; ~scanf("%d",&Case); while(Case--) { init_set(); if(prim()) printf("%d\n",ans); else printf("Not Unique!\n"); } }