给定一个包含n个顶点的正权无向图,编号为0至n-1。请编写程序求出其最小支撑树,并计算其代价。
输入格式:
输入包含多组数据。每组数据第一行为2个整数n和e,均不超过200,分别表示图的顶点数和边数。接下来e行表示每条边的信息,每行为3个非负整数a、b、c,其中a和b表示该边的端点编号,c表示权值。各边并非按端点编号顺序排列。
输出格式:
对于每组数据,若存在最小支撑树则输出一个整数,为最小支撑树各边权值之和;若不存在最小支撑树,则输出“There is no minimum spanning tree.”。
输入样例:
4 5
0 1 1
0 3 1
1 3 5
1 2 1
2 3 8
4 2
0 1 1
2 3 8
输出样例:
3
There is no minimum spanning tree.
思路
为什么不能用
#include<string.h>
memset(xx,MAX,sizeof(xx));
初始化数组xx
1.memset函数按字节对内存块进行初始化,所以不能用它将int数组初始化为0和-1之外的其他值(除非该值高字节和低字节相同)。
2.第二:memset(void *s, int ch,size_t n);中ch实际范围应该在0~~255,因为该函数只能取ch的后八位赋值给你所输入的范围的每个字节
代码
#include<iostream>
#include<string.h>
#define MAX 100000
using namespace std;
int visit[201],xx[201][201];
int prim(int n){
visit[n]={0};//是否在树中
int cost[n];
visit[0]=1;
memset(cost,MAX,sizeof(cost));
for(int i=1;i<n;i++){
cost[i]=xx[0][i];
}
for(int i=1;i<n;i++){
int mi=MAX,k;
for(int j=1;j<n;j++){
if(cost[j]<mi&&visit[j]==0){
mi=cost[j];
k=j;
}
}
visit[k]=1;
for(int j=1;j<n;j++){
if(visit[j]==0&&xx[k][j]<cost[j]){
cost[j]=xx[k][j];
}
}
}
for(int i=0;i<n;i++){
if(visit[i]==0) return 0;
}
int sum=0;
for(int i=1;i<n;i++){
sum+=cost[i];
}
return sum;
}
int main(){
int n,e,a,b,c;
while(cin>>n>>e){
memset(visit,0,sizeof(visit));
for(int i=0;i<n;i++){
for(int j=0;j<n;j++)
xx[i][j]=xx[j][i]=MAX;
}
for(int i=0;i<e;i++){
cin>>a>>b>>c;
xx[a][b]=xx[b][a]=c;
}
int flag=prim(n);
if(flag) cout<<flag<<endl;
else cout<<"There is no minimum spanning tree."<<endl;
}
return 0;
}