UVA 10600 次小生成树模板 prim + dp

题意：输出一个无向连通图的最小生成树和次小生成树的权值。

题解：prim dp

1.最小生成树直接套prim模板就行。

2.次小生成树是改变最小生成树的一条边，使权值次小。做法如下：

(1)添加一条非最小生成树的边。

(2)第(1)步会产生一个环，去掉这个环上在最小生成树中权值最大的边，会形成一个新的生成树。

假如添加(i,j)这条边，需要去除原先从i到j的最大权值边，prim求最小生成树时用pre记录每个点的父结点，prim结束后通过已经生成的pre数组用connect[i][pre[i]]标记(i,pre[i])已用。max1[i][j]存储从i到j的最大权值边，

dp方程：max1[j][temp] = max(max1[j][temp],max1[j][pre[temp]])。j为已遍历的结点，具体细节见代码。

(3)枚举所有不在最小生成树上的边，重复(1)和(2)即可找到次小生成树。

#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 ; 
struct node
{
	int u , v , w ;
} edge[5005] ;
void prim()
{
	int i , j ;
	int min1 , temp ;
	ans1 = 0 ;
	bool vis[115] ;
	int dis[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 ;
			}
		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 i , j , k ;
	bool connect[105][105] ;
	ans2 = inf ;
	memset(connect , 0 , sizeof(connect)) ;
	for(i = 2 ; i <= n ; i ++)//在最小生成树中的边  
	    connect[i][pre[i]] = connect[pre[i]][i] = 1 ;
	for(i = 0 ; i < m ; i ++)
	    if(!connect[edge[i].u][edge[i].v])//不在最小生成树中的边 
           ans2 = min(ans2 , ans1 - max1[edge[i].u][edge[i].v] + edge[i].w) ;
    printf("%d %d\n" , ans1 , ans2) ;
}
int main()
{
  int i , j ; 
  int t ;
  int u , v , w ;
  node a ;
  scanf("%d" , &t) ;
  while(t --)
  {
  	memset(map1 , inf , sizeof(map1)) ;
  	memset(max1 , 0 , sizeof(max1)) ;
  	scanf("%d%d" , &n , &m) ;
  	for(i = 0 ; i < m ; i ++)
  	{
  		scanf("%d%d%d" , &u , &v , &w) ;
  		map1[u][v] = map1[v][u] = w ;
	    a.u = min(u , v) ;
		a.v = max(u , v) ;
		a.w = w ;
		edge[i] = a ;
	}
	prim() ;
	prim_2() ;
  }
}