还是畅通工程
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 39845 Accepted Submission(s): 18095
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
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<climits>
#include<string>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
using namespace std;
#define rep(i,j,k)for(i=j;i<k;i++)
#define per(i,j,k)for(i=j;i>k;i++)
#define MS(x,y)memset(x,y,sizeof(x))
typedef long long LL;
const int INF=0x7FFFFFFF;
const int low(int x){return x&-x;}
const int M=100+1;
int mp[M][M];
int vis[M],v[M];
int i,j,k,n,m,t;
void init()
{
rep(i,1,n+1)
rep(j,1,n+1){
if(i!=j)
mp[i][j]=INF;
else mp[i][j]=0;
}
MS(vis,0);
}
void Prim()
{
LL sum=0;vis[1]=1;
rep(i,1,n+1)v[i]=mp[1][i];
rep(i,1,n){
int minx=INF;
int mink=1;
rep(j,1,n+1)
if(!vis[j]&&v[j]<minx)
minx=v[j],mink=j;
sum+=minx;
vis[mink]=1;
rep(j,1,n+1)
if(!vis[j]&&mp[mink][j]<v[j])
v[j]=mp[mink][j];
}
printf("%I64d\n",sum);
}
int main()
{
while(~scanf("%d",&n),n)
{
m=n*(n-1)/2;
init();
rep(i,0,m){
int a,b,c;
scanf("%d %d %d",&a,&b,&c);
if(c<mp[a][b])
mp[a][b]=mp[b][a]=c;
}
Prim();
}
return 0;
}