题意:
给吃n个村庄,接下来是n行n列数据,例如第一行第二个数据代表第一个村庄和第二个村庄之间的距离。
输入q;
接下来q行代表哪两个村庄之间已经修好路,让你求,还需要最少修多长的路。
题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=1102
分析:
考查最小生成树,最小生成树有两种算法(Prim算法和Kruskal算法),我用的是Kruskal算法,基于并查集算法。起初交上错了好几遍,原因竟然是输入数据用多组输入,就是用while来输入,可是题目明明要求是一组输入。。。。我也搞不懂啊。。。
代码如下:
#include <iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int n,num;//n代表村庄数目,num是用来计算结构体树
int dis[105][105];
//构建村庄
class village
{
public:
int u,v,w;//u为起始点,v终点,w权值
};
village vil[10000];
int node[105];
//比较函数
int cmp(const village &p,const village &q)
{
return p.w<q.w;
}
//初始化
void init(int n)
{
for(int i=0;i<n;i++)
{
node[i]=i;
}
}
//找根节点
int finds(int x)
{
if(x==node[x])
return x;
return finds(node[x]);
}
//合并根节点
void unite(int x,int y)
{
x=finds(x);
y=finds(y);
if(x==y)
return;
node[x]=y;
}
//Kruskal算法
int kru()
{
sort(vil,vil+num,cmp);
int an=0;
for(int i=0;i<num;i++)
{
if(finds(vil[i].u)!=finds(vil[i].v))//判断不成环
{
an+=vil[i].w;
unite(vil[i].u,vil[i].v);
}
}
return an;
}
int main()
{
int i,j;
while(scanf("%d",&n)!=EOF)
{
num=0;
//处理输入数据
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
cin>>dis[i][j];
}
for(i=1;i<=n;i++)
for(j=1;j<i;j++)
{
vil[num].u=i;
vil[num].v=j;
vil[num++].w=dis[i][j];
}
init(n);
int q;
scanf("%d",&q);
while(q--)
{
int a,b;
scanf("%d%d",&a,&b);
unite(a,b);
}
cout<<kru()<<endl;
}
}