传送门:还是畅通工程
还是畅通工程
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 21895 Accepted Submission(s): 9767
Problem Description
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
当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 5Huge input, scanf is recommended.HintHint
Source
解题报告:
此题是最小生成树问题。此题是无向连通图。代码如下:
方法一:prim 邻接矩阵
#include<iostream>
#include<cstdio>
#include<cstring>
#define INF 0x7fffffff
using namespace std;
int a[111][111],dist[111],n;
bool vis[111];
int main(){
while(scanf("%d",&n)==1&&n){
int aa,b,c;
memset(vis,0,sizeof(vis));
memset(a,0,sizeof(a));
for(int i=1;i<=(n*(n-1))/2;i++){
scanf("%d%d%d",&aa,&b,&c);
a[aa][b]=c;
a[b][aa]=c;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(a[i][j]==0)
a[i][j]=INF;
for(int i=1;i<=n;i++)
dist[i]=a[1][i];
vis[1]=true;
int num=1,sum=0;
while(num<n){
int res=INF,pos;
for(int i=1;i<=n;i++)
if(dist[i]<res&&!vis[i]){
pos=i;res=dist[i];
}
vis[pos]=true;
sum+=res;
for(int i=1;i<=n;i++)
if(!vis[i])
dist[i]=min(dist[i],a[pos][i]);
num++;
}
printf("%d\n",sum);
}
return 0;
}
方法二:prim 邻接表
#include<iostream>
#include<cstdio>
#include<cstring>
#define INF 0x7fffffff
using namespace std;
struct edge{
int v,next,cost;
}e[111*111];
int ecnt,first[111];
int n,v;
void addedge(int u,int v,int cost){
e[++ecnt].next=first[u];
e[ecnt].v=v;
e[ecnt].cost=cost;
first[u]=ecnt;
}
int prim(){
bool vis[111];
int dist[111],num=1,sum=0;
memset(vis,0,sizeof(vis));
vis[1]=true;
for(int i=1;i<=n;i++)
dist[i]=INF;
for(int i=first[1];i!=-1;i=e[i].next)
dist[e[i].v]=min(dist[e[i].v],e[i].cost);
while(num<n){
int res=INF,pos;
for(int i=1;i<=n;i++)
if(dist[i]<res&&!vis[i])
pos=i,res=dist[i];
vis[pos]=true;
sum+=res;
for(int i=first[pos]; i!=-1;i=e[i].next){
int v=e[i].v;
if(vis[v]) continue;
dist[v] = min(dist[v], e[i].cost);
}
num++;
}
return sum;
}
int main(){
while(scanf("%d",&n)==1&&n){
memset(first,-1,sizeof(first));
ecnt=-1;
memset(e,0,sizeof(e));
int a,b,c;
for(int i=1;i<=(n*(n-1))/2;i++){
scanf("%d%d%d",&a,&b,&c);
addedge(a,b,c);
addedge(b,a,c);
}
printf("%d\n",prim());
}
return 0;
}
方法三:kruscal
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct edge{
int v,u,cost;
}e[111*111];
bool cmp(const edge &a,const edge &b){
return a.cost<b.cost;
}
int ecnt,fa[111],n,v;
void addedge(int u,int v,int cost){
e[++ecnt].v = v;
e[ecnt].u = u;
e[ecnt].cost = cost;
}
int getf(int x){
return fa[x]==x?x:fa[x]=getf(fa[x]);
}
int kruscal(){
int sum=0;
for(int i=1;i<=n;i++)
fa[i]=i;
sort(e+1,e+1+ecnt,cmp);
for(int i=1;i<=ecnt;i++){
int u=e[i].u,v=e[i].v;
u=getf(u);
v=getf(v);
if(u==v) continue;
sum+=e[i].cost;
fa[u]=v;
}
return sum;
}
int main(){
while(scanf("%d",&n)==1&&n){
ecnt=0;
int a,b,c;
for(int i=1;i<=(n*(n-1))/2;i++){
scanf("%d%d%d",&a,&b,&c);
addedge(a,b,c);
addedge(b,a,c);
}
printf("%d\n",kruscal());
}
return 0;
}