URAL 1416 Confidentia [次小生成树]

题意：

第一行n m代表n个点m条无向边。

接下来m行每行abc，代表ab之间有一条长度为c的无向边。

求:

最小生成树的边权和  次小生成树的边权和

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#include<math.h>
#define INF 0x3f3f3f3f
using namespace std;
int pho[][],n,total,dis[],from[],dp[][];
bool vis[];
void prim(int pos){
    vis[pos]=;
    for(int i=;i<=n;i++){
        if(dis[i]>pho[pos][i]){
            dis[i]=pho[pos][i];
            from[i]=pos;
        }
    }
    int mmin=INF,next=-;
    for(int i=;i<=n;i++){
        if(vis[i]==&&mmin>dis[i]){
            mmin=dis[i];
            next=i;
        }
    }
    if(next>){
        for(int i=;i<=n;i++){
            if(vis[i]){
                dp[i][next]=max(dp[i][from[next]],mmin);
            }
        }
        pho[from[next]][next]=INF;
        pho[next][from[next]]=INF;
        total+=mmin;
        prim(next);
    }
}
int main()
{
    int m;
    scanf("%d%d",&n,&m);
    memset(pho,0x3f,sizeof(pho));
    memset(dis,0x3f,sizeof(dis));
    for(int i=;i<m;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        pho[a][b]=c;pho[b][a]=c;
    }
    prim();
    printf("Cost: %d\n",total);
    bool ok=;
    int tans=INF;
    for(int i=;i<=n;i++){
        for(int j=;j<=n;j++){
            if(pho[i][j]!=INF){
                ok=;
                tans=min(tans,total-max(dp[i][j],dp[j][i])+pho[i][j]);
            }
        }
    }
    if(ok==)tans=-;
    printf("Cost: %d\n",tans);
}