做这题,需要懂得最小生成树(MST)和并查集
可以自己先看一下一本通的相关章节
这里简单说明一下
并查集保存每个点的“祖先”,而单纯的保存祖先使得遍历时间较长,故更改保存方式:把这一家的人的祖先,都设为最老的那个
比如1,2,3,4,5
1是最老的,然后依次递减
那么普通方法是1的祖先是1,2的祖先是1,3的祖先是2,4的祖先是3……
而优化方法表示为:1的祖先是1,2的祖先是1,3的祖先是1,4的祖先是1……
这样可以大大提高判断两个人是否是亲属关系的速度
最小生成树指:
n个点,用n-1条边相连(就是树),使得这n-1条边的边权和最小
一般来说,使用克鲁斯卡尔(Kruskal)算法较好(用到并查集)
首先将所有的边按边权排序,然后每次判断这条边连接的两点是否在同一个集合里,如果不是,那么合并这两个点,边数加1,当边数等于n-1时退出循环
在这道题中,只要用总边权减去最小生成树的总边权就行了
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
int n,k;
struct node{
int x,y;//两点
int cd;//边权
}jsj[10007];
int sum,ans,zh;
bool cmp(node x,node y){
return x.cd<y.cd;//按边权排序
}
int fa[107];
int fi(int x){
if(fa[x]!=x)fa[x]=fi(fa[x]);//如果它的祖先不是自己,更改它的祖先(最后为最老的那个人)
return fa[x];
}
void hb(int x,int y){
fa[x]=y;//合并
return;
}
int main(){
cin>>n>>k;
for(int i=1;i<=n;i++)
fa[i]=i;//每个人的祖先先保存为自己
for(int i=1;i<=k;i++){
int a,b,c;
cin>>a>>b>>c;
jsj[i].x=a;
jsj[i].y=b;
jsj[i].cd=c;
zh+=c;//边权总和
}
sort(jsj+1,jsj+k+1,cmp);
for(int i=1;i<=k;i++){
int a=jsj[i].x;
int b=jsj[i].y;//为下面写代码方便
if(fi(a)!=fi(b)){//不在同一个集合里
hb(fi(a),fi(b));//合并
sum++;//边数加1
ans+=jsj[i].cd;//最小生成树的边权加上当前边的边权
}
if(sum==n-1)break;//生成完毕
}
cout<<zh-ans<<endl;//总边权减去最小生成树的边权和
return 0;
}
写得已经很详细了,即使没有学过也应该懂了
不点个赞吗?