典型的最小生成树-查并集的简单应用
题中的:得到的统计表中列出了任意两村庄间的距离
说明了给的数据一定能形成最小生成树
#include <bits/stdc++.h>
#define ll long long
using namespace std;
//最小生成树问题
int s[110];
struct node {int a, b, d;}c[5000];
bool cmp(node x,node y){return x.d<y.d;}
int findx(int x){//查并集中的查
while(x!=s[x])x=s[x];
return x;
}
int main(){
ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
int n;
while(scanf("%d",&n),n){
int m=n*(n-1)/2;
for(int i=1;i<=m;i++)scanf("%d%d%d",&c[i].a,&c[i].b,&c[i].d);
sort(c+1,c+1+m,cmp);//排序;将边的权值从小到大排序;
for(int i=1;i<=n;i++)s[i]=i;//查并集一定的步骤:“根的初始化”
int ans=0,g=0;
for(int i=1;i<=m;i++){
if(findx(c[i].a) !=findx(c[i].b) ){//只有根节点不同才能进行;否则就不是树就成环了
ans+=c[i].d;
s[ findx(c[i].a) ]=findx(c[i].b);//合并
g++;if(g==n-1)break;//这一步可有可无;这一步剪枝;因为是树所以就是n-1条边;
}
}
printf("%d\n",ans);
}
return 0;
}
此链接:别人的题解,加了个ran数组 记录深度 优化查并集,但是这给题n<100没必要加,可以学习一下别人记录深度的做法
题中的:统计表中列出了有可能建设公路的若干条道路的成本
说明了不一定能形成最小生成树(即不能畅通输出 “?”),且 m<= sum(1+2+```+n-1)=n*n/2=5000;
不能生成最小数的判断有两个方法:
法一:
遍历每个根节点出现不同则不能生成:有两种实现方式
方式①:
#include <bits/stdc++.h>
#define ll long long
using namespace std;
//最小生成树问题
int s[110];
struct node {int a, b, d;}c[5010];
bool cmp(node x,node y){return x.d<y.d;}
int findx(int x){
while(x!=s[x])x=s[x];
return x;
}
int main(){
int n,m;
while(scanf("%d%d",&n,&m)){
if(n==0)break;
//if(n<m-1){printf("?\n");continue;}第一次写加了这句话直接就tle了???
for(int i=1;i<=n;i++)scanf("%d%d%d",&c[i].a,&c[i].b,&c[i].d);
for(int i=1;i<=m;i++)s[i]=i;
sort(c+1,c+1+n,cmp);
int ans=0,g=0,cnt=0;
for(int i=1;i<=n;i++){
if(findx(c[i].a) !=findx(c[i].b) ){
ans+=c[i].d;
s[ findx(c[i].a)]=findx(c[i].b);
g++;if(g==m-1)break;
}
}
bool f=0;
for(int i=2;i<=m;i++){
if(findx(1)!=findx(i)){f=1;break;}//判断根节点是否都相同
}
if(f)printf("?\n");
else printf("%d\n",ans);
}
return 0;
}
方式②:
#include <bits/stdc++.h>
#define ll long long
using namespace std;
//最小生成树问题
int s[110];
struct node {int a, b, d;}c[5010];
bool cmp(node x,node y){return x.d<y.d;}
int findx(int x){
while(x!=s[x])x=s[x];
return x;
}
int main(){
int n,m;
while(scanf("%d%d",&n,&m),n){
for(int i=1;i<=n;i++)scanf("%d%d%d",&c[i].a,&c[i].b,&c[i].d);
for(int i=1;i<=m;i++)s[i]=i;
sort(c+1,c+1+n,cmp);
int ans=0,g=0,cnt=0;
for(int i=1;i<=n;i++){
if(findx(c[i].a) !=findx(c[i].b) ){
ans+=c[i].d;
s[ findx(c[i].a)]=findx(c[i].b);
g++;if(g==m-1)break;
}
}
for(int i=1;i<=m;i++){
if(s[i]==i)cnt++;//cnt计数有多少个根节点
}
if(cnt==1)printf("%d\n",ans);
else printf("?\n");
}
return 0;
}
法二:搜完那些数据以后,能构成的边小于m-1
#include <bits/stdc++.h>
#define ll long long
using namespace std;
//最小生成树问题
int s[110];
struct node {int a, b, d;}c[5010];
bool cmp(node x,node y){return x.d<y.d;}
int findx(int x){
while(x!=s[x])x=s[x];
return x;
}
int main(){
int n,m;
while(scanf("%d%d",&n,&m)){
if(n==0)break;
for(int i=1;i<=n;i++)scanf("%d%d%d",&c[i].a,&c[i].b,&c[i].d);
for(int i=1;i<=m;i++)s[i]=i;
sort(c+1,c+1+n,cmp);
int ans=0,g=0,cnt=0;
for(int i=1;i<=n;i++){
if(findx(c[i].a) !=findx(c[i].b) ){
ans+=c[i].d;
s[ findx(c[i].a)]=findx(c[i].b);
g++;if(g==m-1)break;
}
}
if(g<m-1)printf("?\n");
else printf("%d\n",ans);
}
return 0;
}