HDU1233(Kruskal&Prim两解)

时间:2023-12-16 11:22:02
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省*“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。

当N为0时,输入结束,该用例不被处理。

Output对每个测试用例,在1行里输出最小的公路总长度。

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output

3
5 Huge input, scanf is recommended.

//Kruskal

 #include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
const int maxn=;
#define ms(x,n) memset(x,n,sizeof x);
const int inf=0x3f3f3f3f;
using namespace std;
int n,m;
int u,v,w,parent[maxn];//ranks[maxn];
struct node
{
int u,v,w;
node(){u=,v=,w=inf;}
//node(int uu,int vv,int ww){u=uu,v=vv,w=ww;}
}edge[maxn*maxn];
bool cmp(node a,node b)
{
return a.w<b.w;
}
int init(int n)
{
for(int i=;i<=n;i++)
parent[i]=-;//,ranks[i]=-1;
}
int finds(int x)
{
if(parent[x]<) return x;
return parent[x]=finds(parent[x]);
}
void unions(int x,int y)
{
x=finds(x),y=finds(y);
if(x==y) return;
if(parent[x]<parent[y])
{
parent[x]+=parent[y];
parent[y]=x;
}
else
{
parent[y]+=parent[x];
parent[x]=y;
}
}
int kruskal()
{
sort(edge+,edge+m+,cmp);
init(n);
int ans=;
for(int i=;i<=m;i++)
{
node a=edge[i];
if(finds(a.u)!=finds(a.v))
{
unions(a.u,a.v);
ans+=a.w;
}
}
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie();
while(~scanf("%d",&n),n)
{
m=n*(n-)/;
for(int i=;i<=m;i++)
edge[i]=node();
for(int i=;i<=m;i++)
{
scanf("%d%d%d",&edge[i].u,&edge[i].v,&w);
edge[i].w=min(edge[i].w,w);
}
printf("%d\n",kruskal());
}
return ;
}

//Prim

 #include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#define ms(x,n) memset(x,n,sizeof x);
const int maxn=;
const int inf=0x3f3f3f3f;
using namespace std;
int d[maxn],g[maxn][maxn];
int n;
bool vis[maxn];
void prim()
{
int i,j,k,tmp,ans=;
ms(d,0x3f);
ms(vis,);
d[]=;
for(i=;i<=n;i++)
{
tmp=inf;
for(j=;j<=n;j++)
{
if(!vis[j]&&tmp>d[j])
{
tmp=d[j];
k=j;
}
}
vis[k]=;
ans+=tmp;
for(j=;j<=n;j++)
{
if(!vis[j]&&d[j]>g[k][j])
d[j]=g[k][j];
}
}
cout<<ans<<endl;
}
int main()
{
while(~scanf("%d",&n),n)
{
int m=n*(n-)/;
for(int i=;i<=n;i++)
for(int j=;j<=n;j++)
if(i!=j)
g[i][j]=inf;
else
g[i][j]=;
for(int i=;i<=m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
g[u][v]=g[v][u]=w;
}
prim();
}
return ;
}