uva10462-(带重边次小生成树)

题解:次小生成树,不过这次带重边,那么可以用邻接表存数据,然后判断一下是不是prim算法里面的那条接上去边即可


#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int inf = 0x3f3f3f3f;
const int mx = 1e3+5;
typedef pair<int,int> P;
#define x first
#define y second
int dis[mx];
int vis[mx];
int g[mx][mx];
int Max[mx][mx];
int pre[mx];
vector<P>map[mx];
int mark[mx][mx];
int n,m;
int prim(){
    memset(dis,inf,sizeof(dis));
    memset(vis,0,sizeof(vis));
    dis[1] = 0;
    int ans = 0;
    for(int i = 1; i <= n; i++){
        int minn = inf,x = 0;
        for(int j = 1; j <= n; j++)
            if(!vis[j]&&minn>dis[j])
                minn = dis[x = j];
        if(x==0) return -1;
        vis[x] = 1;
        ans += dis[x];
        for(int j = 1; j <= n; j++)
            if(vis[j]) Max[x][j] = Max[j][x] = x==j?0:max(Max[pre[x]][j],dis[x]);
            else if(dis[j]>g[x][j])
                dis[j] = g[x][j],pre[j] = x;
    }
    return ans;
}
int check(int sum){
    int ans = inf;
    for(int u = 1; u <= n; u++)
        for(auto d: map[u]){
            int v = d.x;
            int w = d.y;
            if((pre[v]!=u&&pre[u]!=v)||g[u][v]!=w||mark[u][v])
                ans = min(ans,sum-Max[u][v]+w);
            if((pre[v]==u||pre[u]==v)&&g[u][v]==w)
                mark[u][v] = mark[v][u] = 1;
        }
    return ans;
}
int main(){
    int t;
    scanf("%d",&t);
    for(int casei = 1; casei <= t ; casei++){
        memset(g,inf,sizeof(g));
        memset(mark,0,sizeof(mark));
        scanf("%d%d",&n,&m);
        for(int i = 1; i <= n; i++)
            map[i].clear();
        for(int i = 1; i <= m; i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            g[u][v] = g[v][u] = min(g[u][v],w);
            if(u>v)
                swap(u,v);
            map[u].push_back({v,w});
        }
        printf("Case #%d : ",casei);
        int ans = prim();
        if(ans==-1)
            puts("No way");
        else{
            ans = check(ans);
            ans==inf?puts("No second way"):printf("%d\n",ans);
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值