[ An Ac a Day ^_^ ][kuangbin带你飞]专题八 生成树 POJ 1679 The Unique MST

求最小生成树是否唯一

求一遍最小生成树再求一遍次小生成树 看看值是否相等就可以

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<algorithm>
 4 #include<cmath>
 5 #include<cstring>
 6 #include<string>
 7 #define cl(a,b) memset(a,b,sizeof(a))
 8 #define debug(x) cerr<<#x<<"=="<<(x)<<endl
 9 using namespace std;
10 typedef long long ll;
11 
12 const int maxn=1e2+10;
13 const int inf=0x3f3f3f3f;
14 
15 int n,m;
16 int cost[maxn][maxn];
17 int pre[maxn],lowc[maxn];
18 int Max[maxn][maxn];
19 bool vis[maxn],used[maxn][maxn];
20 
21 int mst()
22 {
23     int ans=0;
24     cl(vis,false),cl(Max,0),cl(used,false);
25     vis[1]=true,pre[1]=-1,lowc[1]=0;
26     for(int i=2;i<=n;i++)
27     {
28         lowc[i]=cost[1][i];
29         pre[i]=1;
30     }
31     for(int i=2;i<=n;i++)
32     {
33         int minc=inf;
34         int p=-1;
35         for(int j=1;j<=n;j++)
36         {
37             if(!vis[j]&&minc>lowc[j])
38             {
39                 minc=lowc[j];
40                 p=j;
41             }
42         }
43 //        if(minc==inf) return -1;
44         ans+=minc;
45         vis[p]=used[p][pre[p]]=used[pre[p]][p]=true;
46         for(int j=1;j<=n;j++)
47         {
48             if(vis[j])
49             {
50                 Max[j][p]=max(Max[j][pre[p]],lowc[p]);
51             }
52             if(!vis[j]&&lowc[j]>cost[p][j])
53             {
54                 lowc[j]=cost[p][j];
55                 pre[j]=p;
56             }
57         }
58     }
59     return ans;
60 }
61 
62 void solve()
63 {
64     int ans=mst();
65     int Min=inf;
66     for(int i=1;i<=n;i++)
67     {
68         for(int j=i+1;j<=n;j++)
69         {
70             if(cost[i][j]!=inf&&!used[i][j])
71             {
72                 Min=min(Min,ans+cost[i][j]-Max[i][j]);
73             }
74         }
75     }
76     if(ans==Min) printf("Not Unique!\n");
77     else printf("%d\n",ans);
78 }
79 
80 int main()
81 {
82     int T;
83     scanf("%d",&T);
84     while(T--)
85     {
86         cl(cost,inf);
87         scanf("%d%d",&n,&m);
88         for(int i=0;i<m;i++)
89         {
90             int u,v,w;
91             scanf("%d%d%d",&u,&v,&w);
92             cost[u][v]=cost[v][u]=w;
93         }
94         solve();
95     }
96     return 0;
97 }

 

转载于:https://www.cnblogs.com/general10/p/6277136.html