题解:次小生成树:
步骤是prim算法求最小生成树的同时求出任意两点线路上的权值。
Prim求次小生成树:
先用Prim算法求出最小生成树,并把树上的边做标记。
我们考虑,如果插入一条不在树上的边,则会生成一个环,要构造次小生成树则必须删除一条边。为了使生成树的和最小,肯定要选择边权最大的一条边来删除。
max[][]表示,计算方法用到了动态规划,而且需要记录每个节点的父节点,用father[]表示,所以对于每一个节点c ,转移方程就是max[c][e] = max(max[e][father[c], max[father[c][c]),其中e表示除了c的父亲节点和c本身以外的其他节点,max[father[c]][c] 等于c与其父亲节点的边长,还有一个重要的问题就是prim算法里每次选完一个边,就要删除它,为以后的换边准备
做完上一步,任务就完成了一大半,这时候最小生成树的边都已经删完了,最小生成树的值也求出来了,记为ans,然后剩下的都是不选的边,这时候穷举所有的边,对于剩下的边(a,b) ,比较ans - max[a][b] + dis(a,b) ,dis(a,b)代表ab边的长度,取其中的最小值就是次小生成树。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 1005
#define INF 2000000
#define max(a,b) (a)>(b)? a:b
int n,m,k,t;
int a[maxn][maxn];
int MAX[maxn][maxn];
int dist[maxn],vit[maxn],father[maxn];
int prim()
{
memset(dist,0x3f,sizeof(dist));
memset(father,0,sizeof(father));
memset(vit,0,sizeof(vit));
memset(MAX,0,sizeof(MAX));
//for(int i=1;i<=n;i++)
//dist[i]=a[1][i];
dist[1]=0;
//vit[1]=1;
for(int i=1;i<=n;i++)
{
int MIN=INF;
int flag;
for(int j=1;j<=n;j++)
{
if(!vit[j]&&dist[j]<MIN)
{
MIN=dist[j];
flag=j;
}
}
vit[flag]=1;
k+=MIN;
MAX[flag][father[flag]]=a[flag][father[flag]];
MAX[father[flag]][flag]=a[flag][father[flag]];
a[flag][father[flag]]=0;
a[father[flag]][flag]=0;
for(int j=1;j<=n;j++)
{
if(vit[j]&&j!=flag)
{
MAX[flag][j]=max(MAX[father[flag]][j],MAX[father[flag]][flag]);
MAX[j][flag]=MAX[flag][j];
}
}
for(int j=1;j<=n;j++)
{
if(a[flag][j]&&!vit[j])
{
if(dist[j]>a[flag][j])
{
dist[j]=a[flag][j];
father[j]=flag;
}
}
}
}
int second_tree = 0x7ff;
for(int i = 1; i <= n; i++)
{
for(int j = 1; j <= n; j++)
{
if(a[i][j])
{
if(second_tree > k - MAX[i][j] + a[i][j])
{
second_tree = k - MAX[i][j] + a[i][j];
}
}
}
}
t = second_tree;
}
int main()
{
int T;
cin>>T;
while(T--)
{
cin>>n>>m;
memset(a,0,sizeof(a));
for(int i=0;i<m;i++)
{
int x,y,z;
cin>>x>>y>>z;
a[x][y]=a[y][x]=z;
}
k=0,t=1;
prim();
//cout<<k<<" "<<t<<endl;
if(k==t)
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
return 0;
}