天天看點

克魯斯卡爾算法(貪心政策)最小生成樹(NJAU 網工)

核心思想當然不是我的,

我隻是實作而已;

#include<stdafx.h>

#include<string>

#include<iostream>

using namespace std;

#include<fstream>

#include<stdio.h>

#define MAX_VEX_NUM 50

typedef char VertexType;

typedef enum {

DG, UDG

} GraphType;

typedef struct

{

int num;

int weight;

}Arcs;

typedef struct {

VertexType vexs[MAX_VEX_NUM];

int arcs[MAX_VEX_NUM][MAX_VEX_NUM];

int vexnum, arcnum;

GraphType type;

} MGraph;

int getIndexOfVexs(char vex, MGraph *MG)

{

int i;

for (i = 1; i <= MG->vexnum; i++)

{

if (MG->vexs[i] == vex)

{

return i;

}

}

return 0;

}

void create_MG(MGraph *MG)

{

int i, j, k;

int v1, v2, type;

char c1, c2;

//printf("Please input graph type DG(0) or UDG(1) :");

//scanf("%d", &type);

type = 1;//預設無向圖

if (type == 0)

MG->type = DG;

else if (type == 1)

MG->type = UDG;

else

{

printf("Please input correct graph type DG(0) or UDG(1)!");

return;

}

ifstream fp("data.txt");//

char line[100];

fp.getline(line, 100);

int num=stoi(line,0,10);

MG->vexnum = num;

fp.getline(line, 100);

num = stoi(line, 0, 10);

MG->arcnum = num;

//printf("Please input vexmun : ");

//scanf("%d", &MG->vexnum);//讀入節點數目

//printf("Please input arcnum : ");

//scanf("%d", &MG->arcnum);//讀入邊的數目

//getchar();

for (i = 1; i <= MG->vexnum; i++)

{

//printf("Please input %dth vex(char):", i);

//scanf("%c", &MG->vexs[i]);//讀入節點字元

//getchar();

fp.getline(line,100);

MG->vexs[i] = line[0];

}

//初始化鄰接矩陣

for (i = 1; i <= MG->vexnum; i++)

{

for (j = 1; j <= MG->vexnum; j++)

{

MG->arcs[i][j] = 0;

}

}

//輸入邊的資訊,建立鄰接矩陣

for (k = 1; k <= MG->arcnum; k++) {

//printf("Please input %dth arc v1(char) v2(char) : ", k);

//scanf("%c %c", &c1, &c2);

fp.getline(line,100);

c1 = line[0];

c2 = line[1];

fp.getline(line, 100);

int num = stoi(line,0,10);

v1 = getIndexOfVexs(c1, MG);

v2 = getIndexOfVexs(c2, MG);

if (MG->type == 1)

MG->arcs[v1][v2] = MG->arcs[v2][v1]= num;

else

MG->arcs[v1][v2] = num;

//getchar();

}

fp.close();

}

void print_MG(MGraph MG)

{

int i, j;

if (MG.type == DG)

{

printf("Graph type : Direct graph:\n");

}

else

{

printf("Graph type: Undirect graph:\n");

}

printf("Graph vertex number: %d \n", MG.vexnum);

printf("Graph arc number: %d \n", MG.arcnum);

printf("Vertex set:");

for (i = 1; i <= MG.vexnum; i++)

printf("%c   ", MG.vexs[i]);

printf("\n");

printf("Adjacency Matrix:\n");

for (i = 1; i <= MG.vexnum; i++)

{

j = 1;

for (; j < MG.vexnum; j++)

{

printf("%d   ", MG.arcs[i][j]);

}

printf("%d   ", MG.arcs[i][j]);

printf("\n");

}

}

typedef struct

{

int begin;

int end;

int weight;

}Item;

typedef struct

{

Item items[100];

int arcnum;

int vexnum;

}AGraph;

void createArcsArray(MGraph g,AGraph &ag)

{

ag.arcnum = g.arcnum;

ag.vexnum = g.vexnum;

int k = 0;

for (int i=1;i<=g.vexnum;i++)

{

for (int j=i;j<=g.vexnum;j++)

{

if (g.arcs[i][j]==0)//權值為零,就是沒邊,跳過

{

continue;

}

ag.items[k].begin = i;

ag.items[k].end = j;

ag.items[k].weight = g.arcs[i][j];

k++;

}

}

//then sort them

//寫一個冒泡排序吧

for (int i=0;i<ag.arcnum;i++)

{

for (int j=i;j<ag.arcnum;j++)

{

if (ag.items[i].weight>ag.items[j].weight)

{

Item temp;

temp = ag.items[i];

ag.items[i] = ag.items[j];

ag.items[j] = temp;

}

}

}

}

void printAG(AGraph ag)

{

cout << endl << endl;

cout << ag.arcnum << endl;

for (int i=0;i<ag.arcnum;i++)

{

cout << ag.items[i].begin << " " << ag.items[i].end << " " << ag.items[i].weight << endl;

}

}

int Find(int *parent,int f)

{

while (parent[f]>0)

{

f = parent[f];

}

return f;

}

void miniTree(AGraph G)

{

int i, n, m;

int parent[50];

for (i=0;i<G.vexnum;i++)

{

parent[i] = 0;

}

for (i=0;i<G.arcnum;i++)

{

n = Find(parent,G.items[i].begin);

m = Find(parent, G.items[i].end);

if (n!=m)

{

parent[n] = m;

printf("(%d,%d),%d \n",G.items[i].begin, G.items[i].end, G.items[i].weight);

}

}

}

int main()

{

MGraph MG;

AGraph AG;

create_MG(&MG);

createArcsArray(MG,AG);

print_MG(MG);

printAG(AG);

miniTree(AG);

system("pause");

return 0;

}

下面是資料文本的結構

4

4

a

b

c

d

ab

10

ac

20

ad

22

bc

12

繼續閱讀