最小生成树——kruskal
今天是我第一次用克鲁斯卡尔a题。纪念一下。
kruskal算法是一种贪心算法,它每次取权重最小的边并判断该边的两个端点是否属于同一棵树,显然当端点属于
同一棵树时加上这条边会使这棵树出现环,此时舍去该边;当端点不属于同一棵树时,将该边加入树中,因为该边的
权重是当前最小的,最后得到的一定是最小生成树。
首先要判断两个点是否在同一棵树上,这就要用到快速查询数据是否属于同一个集合的工具——并查集。
并查集的基本思想:每个集合选一个元素为代表,其他的元素都直接或间接地指向这个代表,形成一棵树。在查找
两个元素时,只要看它们指向的代表的值是否相同就能知道它们是否属于同一个集合。
并查集的实现方法:以一个数组f[]来存储元素的信息,假设为代表的元素为a,则f[a]=a,该元素为树的根节点, 其余
所有该集合元素的数组都是叶子结点,值都为a,即f[b] = a,f[c]=a ...... 当所有元素都直接指向代表时,查询的时间
复杂度为O(1)。但随着元素的加入,该树可能会退化成一条长链,查询的时间复杂度会退化为O(n),此时只要把所有元素
都指向代表即可。
代码如下:
int find(int x)
{
return f[x]==x ? x : f[x]=find(f[x]);
}
每次查询完都会把f[x]的值更新为find(f[x])
有了并查集,就可以正式开始kruskal了。
首先要以边的权重对边进行排序,直接用STL sort即可
然后按顺序遍历每一条边,将边的两端点用并查集判断是否属于同一集合,否则合并这两条边
for(int i=0;i<m;i++)
{
x = find(wall[i].u);
y = find(wall[i].v);
if(x!=y)
{
//cnt++;
cost += wall[i].w;
f[x] = y;
}
}
合并的操作就是让f[x] = y; ,相当于把xy连起来,cost用于记录最小生成树的各边权值之和。
附上刚刚a的题:http://acm.hdu.edu.cn/showproblem.php?pid=6187
ac代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,c,d,cost,cnt,f[200005],sum;
struct Wall
{
int u;
int v;
int w;
}wall[200005];
bool cmp(Wall a,Wall b)
{
return a.w>b.w;
}
int find(int x)
{
return f[x]==x ? x : f[x]=find(f[x]);
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
sum = 0;
cost = 0;
cnt = 0;
for(int i=1;i<=n;i++)
{
scanf("%d%d",&c,&d);
f[i] = i;
}
int u,v,w;
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&u,&v,&w);
wall[i].u = u;
wall[i].v = v;
wall[i].w = w;
sum += w;
}
int x,y;
sort(wall,wall+m,cmp);
//kruskal
for(int i=0;i<m;i++)
{
x = find(wall[i].u);
y = find(wall[i].v);
if(x!=y)
{
cnt++;
cost += wall[i].w;
f[x] = y;
}
}
printf("%d %d\n",m-cnt,sum-cost);
}
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步