这应该算是最小生成树(Kruskal算法)基础题目吧(虽然现在对自己来说还不是很熟悉)。
题意:n个村庄想要彼此联通,已经有一些之间有了公路,问最短还要建多长的公路。
思路:Kruskal算法,从1~1000枚举道路的长度,根据并查集操作,查询当前边的节点是否已经连通,如果没有联通,则答案加上这条边的长度,并把两个节点并入一个集合。
#include<iostream>
using namespace std;
int vill[105][105];
int fa[105];
int set_find(int p)
{
if(fa[p]==p)
return p;
return fa[p]=set_find(fa[p]);
}
int main()
{
int n,q;
int a,b;
while(cin>>n)
{
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
cin>>vill[i][j];
for(int i=1; i<=n; i++)
fa[i]=i;
cin>>q;
for(int i=1; i<=q; i++)
{
cin>>a>>b;
fa[set_find(a)]=set_find(b);
}
int ans=0;
for(int k=1; k<=1000; k++)
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
if(vill[i][j]==k&&set_find(i)!=set_find(j))
{
ans+=vill[i][j];
fa[set_find(i)]=set_find(j);
}
cout<<ans<<endl;
}
return 0;
}
在这稍微总结一下Kruskal算法求解最小生成树的基本过程:
1.首先根据边的权值对边从小到大排序;
2.建立n棵树组成的森林,每棵树包含一个节点;
3.每次取出最小边,判断两个节点是否分别属于两颗子树(利用并查集操作),如果不属于则合并两棵树。