最小生成树(例子)

N个点M条边的无向连通图,每条边有一个权值,求该图的最小生成树。
Input
第1行:2个数N,M中间用空格分隔,N为点的数量,M为边的数量。(2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行:每行3个数S E W,分别表示M条边的2个顶点及权值。(1 <= S, E <= N,1 <= W <= 10000)
Output
输出最小生成树的所有边的权值之和。
Input示例
9 14
1 2 4
2 3 8
3 4 7
4 5 9
5 6 10
6 7 2
7 8 1
8 9 7
2 8 11
3 9 2
7 9 6
3 6 4
4 6 14
1 8 8
Output示例
37


prim算法

#include <iostream>

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
using namespace std;
int g[1010][1010];
int b[1010][1010];//邻接矩阵
int vis[1000];//是在哪
int p[1000];//a到b的权值
int maxx=0x3f3f3f3f;
int l(int n,int m)
{
    int i,minn,ans,k,j;
    ans=0;
    memset(vis,0,sizeof(vis));
    memset(p,0,sizeof(p));
    vis[1]=1;
    for(i=2; i<=n; i++)
        p[i]=g[1][i];
    for(i=2; i<=n; i++)
    {
        minn=maxx;
        k=-1;
        for(j=1; j<=n; j++)
            if(vis[j]!=1&&minn>p[j])
            {
                minn=p[j];
                k=j;
            }
        ans+=minn;
        vis[k]=1;
        for(j=1; j<=n; j++)
        {
            if((vis[j]!=1&&p[j]>g[k][j]))
                p[j]=g[k][j];
        }
    }
    return ans;
}
int main()
{
    int n,m,i,j,a,b,c;
    scanf("%d%d",&n,&m);
    memset(g,0,sizeof(g));
    for(i=1; i<=m; i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        g[a][b]=c;
        g[b][a]=c;
    }
    for(i=1; i<=n; i++)
        for(j=1; j<=n; j++)
            if(g[i][j]==0)
            {
                g[i][j]=maxx;
                g[j][i]=maxx;
            }
    printf("%d\n",l(n,m));
    return 0;
}



猜你喜欢

转载自blog.csdn.net/najiuzheyangbaacm/article/details/80546534