#include <iostream>
using namespace std;
#define MaxSise 10
#define positive_infinity 1000
int visited[MaxSise]={0};
typedef char datatype;
typedef struct {
datatype vertex[MaxSise];
int edge[MaxSise][MaxSise];
int vertexnum,edgenum;
}MGraph;
void CreatGraph(MGraph *g,datatype a[],int n,int e){
int i,j;
//初始化矩阵
for(i=0;i<n;i++){
for(j=0;j<n;j++){
if(i==j)g->edge[i][j]=0;
else g->edge[i][j]=positive_infinity;
}
}
g->vertexnum=n;
g->edgenum=e;
cout<<"请输入"<<e<<"条边的两邻接点和权值"<<endl;
for(i=0;i<n;i++)g->vertex[i]=a[i];
int la,lb,value;
i=e;
while(i){
cin>>la>>lb>>value;//或设置有边标志(1)
if(value<=0||value>=positive_infinity||la<0||la>=n||lb<0||lb>=n){
cout<<"数据有误,请重新输入"<<endl;
continue;
}
g->edge[la][lb]=value;
g->edge[lb][la]=value;
i--;
}
cout<<"建表完成"<<endl;
}
int MinEdge(int a[],int n){
int i,t,ti;
t=positive_infinity;
for(i=0;i<n;i++){
if(a[i]!=0&&a[i]<t){
t=a[i];
ti=i;
}
}
return ti;
}
void Prim(MGraph* g,int v){
int i,j,k;
int adjvex[MaxSise],lowcost[MaxSise];
for(i=0;i<g->vertexnum;i++){
lowcost[i]=g->edge[v][i];
adjvex[i]=v;
}
lowcost[v]=0;
for(k=1;k<g->vertexnum;k++){
j=MinEdge(lowcost,g->vertexnum);
cout<<j<<' '<<adjvex[j]<<' '<<lowcost[j]<<endl;
lowcost[j]=0;//该点标记已连入
for(i=0;i<g->vertexnum;i++){
if(/*lowcost[i]!=0&&g->edge[i][j]!=0&&*/g->edge[i][j]<lowcost[i]){
lowcost[i]=g->edge[i][j];
adjvex[i]=j;
}
}
}
}
void dfs(MGraph *g,int v){//深度优先遍历
cout<<g->vertex[v]<<' ';
visited[v]=1;
for(int j=0;j<g->vertexnum;j++){
if(g->edge[v][j]!=0&&g->edge[v][j]!=positive_infinity&&visited[j]!=1){
dfs(g,j);
}
}
}
void bfs(MGraph *g,int v){
int i,j,q[MaxSise]={0};
int front=-1,rear=-1;
cout<<g->vertex[v]<<' ';
visited[v]=1;
q[++rear]=v;
while(front!=rear){
i=q[++front];
for(j=0;j<g->vertexnum;j++){
if(g->edge[i][j]!=0&&g->edge[i][j]!=positive_infinity&&visited[j]!=1){
cout<<g->vertex[j]<<' ';
visited[j]=1;
q[++rear]=j;
}
}
}
}
//0 1 12
//0 2 5
//2 3 7
//3 0 8
int main(){
int n,e;
MGraph MG;
cout<<"请输入节点个数和边数"<<endl;
cin>>n>>e;
datatype a[n];
cout<<"请输入"<<n<<"个节点"<<endl;
for(int i=0;i<n;i++)cin>>a[i];
CreatGraph(&MG,a,n,e);
int v;//遍历起始点
// cout<<"请输入深度优先遍历起点"<<endl;
// cin>>v;
// dfs(&MG,v);
// for(int i=0;i<n;i++)visited[i]=0;cout<<endl;
//
// cout<<"请输入广度优先遍历起点"<<endl;
// cin>>v;
// bfs(&MG,v);
cout<<"输入生成树的起点"<<endl;
cin>>v;
Prim(&MG,v);
return 0;
}
运行结果:(无向图)
请输入节点个数和边数
6 9
请输入6个节点
0 1 2 3 4 5
请输入9条边的两邻接点和权值
0 1 34
0 2 46
0 5 19
1 4 12
2 3 17
2 5 25
3 4 38
3 5 25
4 5 26
建表完成
输入生成树的起点
0
结果输出:
5 0 19
2 5 25
3 2 17
4 5 26
1 4 12