首页 > 代码库 > Ural 1416 Confidential,次小生成树
Ural 1416 Confidential,次小生成树
不严格次小生成树。
注意图可能不连通。
#include <cstdio> #include <cstring> #include <algorithm> #include <vector> using namespace std; const int maxn = 505; const int INF = 1e7; bool vis[maxn]; int d[maxn]; int pre[maxn]; int Max[maxn][maxn]; int g[maxn][maxn]; int n, m; void init() { int u, v, c; scanf("%d%d", &n, &m); for(int i=1; i<=n; ++i) for(int j=1; j<=n; ++j) g[i][j] = INF; for(int i=1; i<=m; ++i) { scanf("%d%d%d", &u, &v, &c); g[u][v] = g[v][u] = c; } } int Prim() { for(int i=2; i<=n; ++i) { pre[i]=1; d[i] = g[1][i]; } memset(vis, 0, sizeof vis ); vis[1] = 1; d[1] = 0; pre[1] = -1; int ans = 0; for(int i=1; i<n; ++i) { int Mn = INF, k = -1; for(int j=1; j<=n; ++j) if(!vis[j] && Mn>d[j]){ Mn = d[k=j];} if(-1==k) return INF; ans += Mn; g[k][pre[k]] = g[pre[k]][k] = INF; for(int j=1; j<=n; ++j) if(vis[j]) Max[j][k] = Max[k][j] = max(Mn, Max[pre[k]][j]); vis[k] = 1; for(int j=1; j<=n; ++j) if(!vis[j]&&d[j]>g[j][k]) { d[j] = g[j][k]; pre[j] = k; } } return ans; } void solve() { int ans1 = Prim(); if(ans1==INF) { printf("Cost: -1\n"); } else { printf("Cost: %d\n", ans1); } int ans2 = INF; for(int i=1; i<=n; ++i) for(int j=i+1; j<=n; ++j) if(g[i][j]!=INF) { ans2 = min(ans2, ans1+g[i][j]-Max[i][j]); } if(ans2==INF){ printf("Cost: -1\n"); } else { printf("Cost: %d\n", ans2); } } int main() { init(); solve(); return 0; }
Ural 1416 Confidential,次小生成树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。