F - 数据结构实验之图论六：村村通公路（最小生成树Prim/Kruskal）

最新推荐文章于 2022-05-15 18:24:48 发布

彡倾灬染|

最新推荐文章于 2022-05-15 18:24:48 发布

阅读量736

点赞数 1

分类专栏：最小生成树 Prim Kruskal 文章标签： 1024程序员节 c++ c语言

本文链接：https://blog.csdn.net/qq_45830025/article/details/109267606

版权

最小生成树同时被 3 个专栏收录

3 篇文章 0 订阅

订阅专栏

Prim

2 篇文章 0 订阅

订阅专栏

Kruskal

2 篇文章 0 订阅

订阅专栏

本文探讨了如何利用Prim算法和Kruskal算法解决实际场景中的问题——为农村实现村村通公路，通过计算连接所有村落所需的最低成本。博客详细介绍了两种算法的实现和应用实例，适合理解并应用于网络优化和图论问题解决。

摘要由CSDN通过智能技术生成

Description

当前农村公路建设正如火如荼的展开，某乡镇政府决定实现村村通公路，工程师现有各个村落之间的原始道路统计数据表，表中列出了各村之间可以建设公路的若干条道路的成本，你的任务是根据给出的数据表，求使得每个村都有公路连通所需要的最低成本。

Input

连续多组数据输入，每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000)，随后M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本，村庄从1～N编号。

Output

输出使每个村庄都有公路连通所需要的最低成本，如果输入数据不能使所有村庄畅通，则输出-1，表示有些村庄之间没有路连通。

Sample

Input

Output

答案：

最小生成树Prim算法：

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
const int N = 1e5 + 10;
using namespace std;

int mp[1111][1111];
int vis[1111];

int Prim(int n)
{
    int i;
    int j;
    int dp[1111];
    int sum;
    int minn;
    int flag;
    for(i = 1; i<= n; i++)
    {
        vis[i] = 0;
        dp[i] = mp[1][i];
    }
    vis[1] = 1;
    sum = 0;
    for(i =1; i < n; i++)
    {
        minn = INF;
        flag = -1;
        for(j = 1; j<= n; j++)
        {
            if(!vis[j] && dp[j] < minn)
            {
                minn = dp[j];
                flag = j;
            }
        }
        if(minn == INF)
            return -1;
        vis[flag] = 1;
        sum += minn;
        for(j = 1; j<= n; j++)
        {
            if(!vis[j] && dp[j] > mp[flag][j])
                dp[j] = mp[flag][j];
        }
    }
    return sum;
}

int main()
{
    ios::sync_with_stdio(0);
    int n,m;
    while(cin>>n>>m)
    {
        int i,j;
        for(i=1; i<=n; i++)
        {
            for(j=1; j<=n; j++)
            {
                if(i==j)
                    mp[i][j]=0;
                else
                    mp[i][j]=INF;
            }
        }
        while(m--)
        {
            int x,y,k;
            cin>>x>>y>>k;
            if(k<mp[x][y])
            {
                mp[x][y]=mp[y][x]=k;
            }
        }
        cout<<Prim(n)<<endl;
    }
    return 0;
}

最小生成树Kruskal算法（并查集思想）

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
const int N = 1e5 + 10;
using namespace std;

struct node
{
    int u,v,w;
} dp[1111];

int n;
int vis[1111];

bool cmp(node a,node b)
{
    return a.w<b.w;
}

void init()
{
    for(int i=1; i<=n; i++)
    {
        vis[i]=i;
    }
}

int find_root(int i)
{
    if(vis[i]==i)
    {
        return vis[i];
    }
    else
    {
        vis[i]=find_root(vis[i]);
        return vis[i];
    }
}

bool Merge(int u,int v)
{
    int t1=find_root(u);
    int t2=find_root(v);
    if(t1==t2)
    {
        return true;
    }
    else
    {
        vis[t2]=t1;
        return false;
    }
}

int main()
{
    int m;
    while(cin>>n>>m)
    {
        init();
        for(int i=1; i<=m; i++)
        {
            cin>>dp[i].u>>dp[i].v>>dp[i].w;
        }
        sort(dp+1,dp+1+m,cmp);
        int ans=0,cnt=0;
        for(int i=1; i<=m; i++)
        {
            if(cnt==n-1)
            {
                break;
            }
            if(!Merge(dp[i].u,dp[i].v))
            {
                ans=ans+dp[i].w;
                cnt++;
            }
        }
        if(cnt==n-1)
        {
            cout<<ans<<endl;
        }
        else
        {
            cout<<"-1"<<endl;
        }
    }
    return 0;
}