畅通工程
Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 44040 Accepted Submission(s): 19688
Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 );随后的 N
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
Output
对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。
Sample Input
3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100
Sample Output
3
?
首次提交的wrong
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<cstring>
#include<sstream>
#include<cmath>
#define NUM 1001
int undef=-1;
int parent[NUM];
int connect[NUM];
int village[NUM][NUM];
struct Ro{
int min,max;
int len;
}road[NUM];
int findRoot(int par[],int node){
if(par[node]==undef)
return node;
par[node]=findRoot(par, par[node]);
return par[node];
}
bool compare(Ro a,Ro b){
return a.len<b.len;
}
using namespace std;
int main()
{
int a,b,c,N,M,treeCount,minCost;
while(cin>>N>>M && N){//先判断是不是连通
memset(parent,undef,sizeof(parent));
for(int i=0; i<N; i++){
cin >> a >> b >> c;
if(a>b){
road[i].min=b;
road[i].max=a;
parent[a]=b;
findRoot(parent,a);
}else{
road[i].min=a;
road[i].max=b;
parent[b]=a;
findRoot(parent,b);
}
road[i].len=c;
}
treeCount=0;
for(int i=1; i<=M; i++)
if(parent[i]==undef)
treeCount++;
if(treeCount==1){//连通,求出最小连通值
sort(road,road+N,compare);
minCost=0;
memset(connect,undef,sizeof(connect));
for(int i=0; i<N; i++){
if(findRoot(connect,road[i].max)!=findRoot(connect,road[i].min)){
minCost+=road[i].len;
connect[road[i].max]=road[i].min;
findRoot(connect,road[i].max);
}
}
cout << minCost << endl;
} else{
cout << "?" << endl;
}
}
return 0;
}
没有很好地理解两个集合怎么用一条语句就能合并起来,经过调试发现,找到两个集合的根,然后结合;
之后查找根的时候,就会递归地更新两个集合;终于AC啦
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<cstring>
#include<sstream>
#include<cmath>
#define NUM 1001
int undef=-1;
int M;
int parent[NUM];
int connect[NUM];
int village[NUM][NUM];
struct Ro{
int min,max;
int len;
}road[NUM];
int findRoot(int par[],int node){
if(par[node]==undef)
return node;
par[node]=findRoot(par, par[node]);
return par[node];
}
bool compare(Ro a,Ro b){
return a.len<b.len;
}
void merge(int par[],int minRoot,int maxRoot){
for(int i=1; i<=M; i++){
if(i==maxRoot)
par[i]=minRoot;
if(par[i]==maxRoot)
par[i]=minRoot;
}
}
using namespace std;
int main()
{
int a,b,c,N,treeCount,minCost,minRoot,maxRoot;
while(cin>>N>>M && N){//先判断是不是连通
memset(parent,undef,sizeof(parent));
for(int i=0; i<N; i++){
cin >> a >> b >> c;
if(a>b){
road[i].min=b;
road[i].max=a;
parent[a]=b;
findRoot(parent,a);
}else{
road[i].min=a;
road[i].max=b;
parent[b]=a;
findRoot(parent,b);
}
road[i].len=c;
}
treeCount=0;
//if(treeCount==1){//连通,求出最小连通值
sort(road,road+N,compare);
minCost=0;
memset(connect,undef,sizeof(connect));
for(int i=0; i<N; i++){
minRoot=findRoot(connect,road[i].min);
maxRoot=findRoot(connect,road[i].max);
if(minRoot!=maxRoot){
//为不同的集合,合并至小的头
minCost+=road[i].len;
connect[minRoot]=maxRoot;
// connect[road[i].max]=road[i].min;
// findRoot(connect,road[i].max);
// if(minRoot<maxRoot){
// merge(connect,minRoot,maxRoot);
// }else{
// merge(connect,maxRoot,minRoot);
// }
}
}
for(int i=1; i<=M; i++)
if(connect[i]==undef)
treeCount++;
if(treeCount==1){
cout << minCost << endl;
} else{
cout << "?" << endl;
}
}
return 0;
}