最小生成树(prim)讲解

prim

吐槽：
prim这个算法一开始我并不想去学习它，因为这个算法和krsual比起来真的没可比性，但是最近遇到一些题，用这个思路去解决更简单，索性我就学一下它
讲解：
首先我们定义一个colsest[ ]代表最邻近点
举个栗子：

第一行代表数组下标
第二行代表下标下的值
上表可以画成下图这样理解：

这个图的含义便是最小生成树
定义lowcost[ ]代表这个图的边权
因为是连通的树形结构所以一定只有n-1条边

如果不好理解 ，请先理解贪心思想再看该例题（裸最小生成树）：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 1005
int closest[maxn];//最邻近点
int Map[maxn][maxn];//存图
int lowcost[maxn];//存最小边权
bool vis[maxn];//标记加入集合的点
int n,m;
void prim(int u)
{
    vis[u]=true;
    for(int i=1; i<=n; i++)
    {
        if(i!=u)
        {
            vis[i]=false;
            closest[i]=u;//一开始所有点的最邻近点都是u
            lowcost[i]=Map[u][i];
        }
        else
        {
            lowcost[i]=0;//相同点为0
        }
    }
    for(int i=1; i<=n; i++)
    {
        int temp=0x3f3f3f3f;//枚举最小边权
        int pos=u;//得到最小权值点
        for(int j=1; j<=n; j++)
        {
            if((!vis[j])&&(lowcost[j]<temp))//从非集合中的点中找到与u最近的点
            {
                temp=lowcost[j];
                pos=j;
            }
        }
        if(pos==u)
            break;//说明n个点全部加入了集合
        vis[pos]=true;
        for(int j=1; j<=n; j++)
        {
            if((!vis[j])&&(lowcost[j]>Map[pos][j]))
            {
                lowcost[j]=Map[pos][j];//更新边权
                closest[j]=pos;//新点加入集合
            }
        }
    }
    int sumval=0;
    for(int i=1; i<=n; i++)
        sumval+=lowcost[i];//最小生成树的权值和

    printf("%d\n",sumval);

}
void input()
{
   
    memset(Map,0x3f3f3f3f,sizeof(Map));
    int u,v,w;
    for(int i=0; i<m; i++)
    {
        scanf("%d %d %d",&u,&v,&w);
        Map[u][v]=Map[v][u]=min(Map[u][v],w);
    }
}
int main()
{

    while(~scanf("%d",&n)&&n)
    {
        scanf("%d",&m);
        input();
        prim(1);

    }

}

例二：


AC代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define inf 0x3f3f3f3f
int Map[1005][1005];
int vis[1005];
int dis[1005];
int n,m,w;
void prim(int u)
{
    for(int i=1; i<n; i++)
        dis[i]=Map[0][i];
    dis[u]=0;
    vis[u]=1;
    int ans=0;
    for(int i=0; i<n-1; i++)
    {
        int minn=inf,temp;
        for(int j=0; j<n; j++)
        {
            if(!vis[j]&&dis[j]<minn)
            {
                minn=dis[j];
                temp=j;
            }
        }
        vis[temp]=1;
        ans+=minn;
        for(int j=0; j<n; j++)
        {
            if(!vis[j]&&dis[j]>Map[temp][j])
            {
                dis[j]=Map[temp][j];
            }
        }
    }
    printf("%d\n",ans+m);
}
int main()
{
    char str[1005][15];
    while(~scanf("%d%d%d",&n,&m,&w))
    {
        memset(Map,0,sizeof(Map));
        for(int i=0; i<n; i++)
            scanf("%s",str[i]);
        for(int i=0; i<n-1; i++)
        {
            for(int j=i+1; j<n; j++)
            {
                int s=0;
                for(int k=0; k<m; k++)
                    if(str[i][k]!=str[j][k])
                        s++;
                Map[i][j]=Map[j][i]=min(m,s*w);
            }
        }
        memset(vis,0,sizeof(vis));
        prim(0);
    }
    return 0;

}

吐槽：
第一眼看到这道题，我没有想到最小生成树，因为只会一个krsual算法联想到最小生成树确实不太好想，但理解了prim后这道题往最小生成树方面想就容易很多了