题意:

第一行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);
}
05-11 20:14