NYOJ118 修路方案 次小生成树Prim

题解：次小生成树： 步骤是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;
}