UVA 10462 次小生成树 有重边

题意：一个有重边的无向连通图，是否存在此小生成树。

题解：次小生成树

1.与上篇博客同样的方法，只不过有重边时标记最小生成树中的边需直接标记边，而不能标记边对应的顶点。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<queue>
#define inf 0x3f3f3f3f
using namespace std ;
int n , m ;
int map1[105][105] ;
int max1[105][105] ;//max1[i][j]表示在最小生成树中i到j的最大边 
int pre[105] ;//pre[i]表示在最小生成树中i的父结点
int ans1 , ans2 ; 
int dis[115] ;
struct node
{
	int u , v , w ;
} edge[5005] ;
void prim()
{
	int i , j ;
	int min1 , temp ;
	ans1 = 0 ;
	bool vis[115] ;
	memset(vis , 0 , sizeof(vis)) ;
	dis[1] = 0 ;
	vis[1] = 1 ;
	pre[1] = -1 ;
  	for(i = 2 ; i <= n ; i ++)
  	    pre[i] = 1 ;
	for(i = 2 ; i <= n ; i ++)
	    dis[i] = map1[1][i] ;
	for(i = 2 ; i <= n ; i ++) 
	{
		min1 = inf ;
		for(j = 2 ; j <= n ; j ++)
		    if(!vis[j] && dis[j] < min1)
		    {
		    	min1 = dis[j] ;
		    	temp = j ;
			}
		if(min1 == inf)
		{
		  ans1 = -1 ;
		  break ;	
		} 
		vis[temp] = 1 ;
		ans1 += min1 ;
		for(j = 1 ; j <= n ; j ++)
		{
			if(vis[j] && j != temp)
			   max1[j][temp] = max1[temp][j] = max(max1[j][pre[temp]] , dis[temp]) ;
			if(!vis[j] && dis[j] > map1[temp][j])
			{
			  dis[j] = map1[temp][j] ;
			  pre[j] = temp ;//存储父结点	
			}
		}
	}
}
void prim_2(int num)//次小生成树 
{
	if(ans1 == -1)
	{
		printf("Case #%d : No way\n" , num) ;
	    return ;
	}
	int i , j , k ;
	bool vis[5005] ;
	memset(vis , 0 , sizeof(vis)) ;
	ans2 = inf ;
	for(i = 2 ; i <= n ; i ++)//在最小生成树中的边  
	    for(j = 0 ; j < m ; j ++)
	    	if(edge[j].u == min(i , pre[i]) && edge[j].v == max(i , pre[i]) && edge[j].w == dis[i])
	    	{
	    		vis[j] = 1 ;
	    		break ;
			}
	for(i = 0 ; i < m ; i ++)
	    if(!vis[i])//不在最小生成树中的边 
           ans2 = min(ans2 , ans1 - max1[edge[i].u][edge[i].v] + edge[i].w) ;
    if(ans2 == inf)
    {
    	printf("Case #%d : No second way\n" , num) ;
    	return ;
	}
	printf("Case #%d : %d\n" , num , ans2) ;
}
int main()
{
  int i , j ; 
  int t ;
  int u , v , w ;
  node a ;
  scanf("%d" , &t) ;
  for(i = 1 ; i <= t ; i ++)
  {
  	memset(map1 , inf , sizeof(map1)) ;
  	memset(max1 , 0 , sizeof(max1)) ;
  	scanf("%d%d" , &n , &m) ;
  	for(j = 0 ; j < m ; j ++)
  	{
  		scanf("%d%d%d" , &u , &v , &w) ;
  		map1[u][v] = map1[v][u] = min(map1[u][v] , w) ;
	    a.u = min(u , v) ;
		a.v = max(u , v) ;
		a.w = w ;
		edge[j] = a ;
	}
	prim() ;
	prim_2(i) ;
  }
}