图论day61:最小生成树|最小生成树理论基础:prim算法、kruskal算法(思维导图版)、53.寻宝(卡码网 第七期模拟笔试)

发布于:2024-10-17 ⋅ 阅读:(32) ⋅ 点赞:(0)

图论day61:最小生成树|最小生成树理论基础:prim算法、kruskal算法(思维导图版)、53.寻宝(卡码网 第七期模拟笔试)

最小生成树理论基础(思维导图版)

在这里插入图片描述

53.寻宝(卡码网 第七期模拟笔试)

题目描述

在世界的某个区域,有一些分散的神秘岛屿,每个岛屿上都有一种珍稀的资源或者宝藏。国王打算在这些岛屿上建公路,方便运输。

不同岛屿之间,路途距离不同,国王希望你可以规划建公路的方案,如何可以以最短的总公路距离将 所有岛屿联通起来(注意:这是一个无向图)。

给定一张地图,其中包括了所有的岛屿,以及它们之间的距离。以最小化公路建设长度,确保可以链接到所有岛屿。

输入描述

第一行包含两个整数V 和 E,V代表顶点数,E代表边数 。顶点编号是从1到V。例如:V=2,一个有两个顶点,分别是1和2。

接下来共有 E 行,每行三个整数 v1,v2 和 val,v1 和 v2 为边的起点和终点,val代表边的权值。

输出描述

输出联通所有岛屿的最小路径总距离

输入示例

7 11
1 2 1
1 3 1
1 5 2
2 6 1
2 4 2
2 3 2
3 4 1
4 5 1
5 6 2
5 7 1
6 7 1

输出示例

6

提示信息

数据范围:
2 <= V <= 10000;
1 <= E <= 100000;
0 <= val <= 10000;

如下图,可见将所有的顶点都访问一遍,总距离最低是6.

1.prim法

#include <iostream>
#include <vector>
#include <climits>
using namespace std;

int main()
{
    int v,e,x,y,k;
    cin>>v>>e;
    vector<vector<int>> grid(v+1,vector<int>(v+1,10001));
    for(int i=0;i<e;i++)
    {
        cin>>x>>y>>k;
        grid[x][y]=k;
        grid[y][x]=k;
    }

vector<int> minDist(v+1,10001);

vector<bool> isInTree(v+1,false);

for(int i=1;i<=v;i++)
{
    int cur=-1;
    int minVal=INT_MAX;
    for(int j=1;j<=v;j++)
        if(!isInTree[j]&&minDist[j]<minVal)
        {
            minVal=minDist[j];
            cur=j;
        }

    isInTree[cur]=true;

    for(int j=1;j<=v;j++)
        if(!isInTree[j]&&grid[cur][j]<minDist[j])
            minDist[j]=grid[cur][j];
}

int result=0;
for(int i=2;i<=v;i++)
    result+=minDist[i];
cout<<result<<endl;
}

2.kruskal法

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

struct Edge
{
    int l,r,val;
};

int n;
vector<int> father(10001,0);

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

int find(int u)
{
    return u==father[u]?u:father[u]=find(father[u]);
}

void join(int u,int v)
{
    u=find(u);
    v=find(v);
    if(u==v)
        return;
    else
        father[v]=u;
}

bool cmp(Edge& edge1,Edge& edge2)
{
    return edge1.val<edge2.val;
}

int main()
{
    int e,v1,v2,val;
    cin>>n>>e;
    vector<Edge> edges;
    for(int i=1;i<=e;i++)
    {
        cin>>v1>>v2>>val;
        edges.push_back({v1,v2,val});
    }

    sort(edges.begin(),edges.end(),cmp);

    init();

    int result=0;
    for(int i=0;i<e;i++)
    {
        int x=find(edges[i].l);
        int y=find(edges[i].r);
        if(x!=y)
        {
            result+=edges[i].val;
            join(x,y);
        }
    }
    cout<<result<<endl;
}

分析思路:

在这里插入图片描述