学习了两个最小生成树的基本思想
以及分别的代码实现
什么是最小生成树,即为权值最小,即连接任意两个顶点的距离最小,基于贪婪算法
1.克鲁斯卡尔算法(Kruskal)
基本思想是抓住最小两个字进行做文章,直接抓住权值最小的边,先根据权值大小进行对各个连接的顶点方案进行排列,根据图中是否成环来判断是否该方案可以成为最小生成树其中的一部分,依次向后进行选择,直到选择到(顶点数-1)个即可结束。
具体实现代码
#include <stdio.h>
#define MAXE 100
#define MAXV 100
typedef struct
{
int vex1;
int vex2;
int weight;
}Edge;
void kruskal(Edge E[],int n,int e)
{
int i,j,m1,m2,sn1,sn2,k,sum=0;
int vset[n+1];
for(i=1;i<=n;i++)
{
vset[i]=i;
}
k=1;
j=0;
while(k<e)
{
m1=E[j].vex1;
m2=E[j].vex2;
sn1=vset[m1];
sn2=vset[m2];
if(sn1!=sn2)
{
printf("V%d-V%d=%d\n",m1,m2,E[j].weight);
sum+=E[j].weight;
k++;
if(k>=n)
break;
for(i=1;i<=n;i++)
{
if (vset[i]==sn2)
{
vset[i]=sn1;
}
}
}
j++;
}
printf("最小权值之和=%d\n",sum);
}
//以下为快排
int fun(Edge arr[],int low,int high)
{
int key;
Edge lowx;
lowx=arr[low];
key=arr[low].weight;
while(low<high)
{
while(low<high && arr[high].weight>=key)
high--;
if(low<high)
arr[low++]=arr[high];
while(low<high && arr[low].weight<=key)
low++;
if(low<high)
arr[high--]=arr[low];
}
arr[low]=lowx;
return low;
}
void quick_sort(Edge arr[],int start,int end)
{
int pos;
if(start<end)
{
pos=fun(arr,start,end);
quick_sort(arr,start,pos-1);
quick_sort(arr,pos+1,end);
}
}
int main()
{
Edge E[MAXE];
int nume,numn;
printf("输入顶数和边数:\n");
scanf("%d%d",&numn,&nume);
for(int i=0;i<nume;i++)
scanf("%d%d%d",&E[i].vex1,&E[i].vex2,&E[i].weight);
quick_sort(E,0,nume-1);
kruskal(E,numn,nume);
}
2. 普里姆算法(prim)
基本思想为给每一个顶点更新一个最小权值即开设3个数组,分别存放是否选择结束的flag,最小权值(一开始都为无穷大),父亲顶点(开始时均为-1),从选定的顶点进行向下判断
直到所有的flag都为1,代表选择结束。
具体代码实现
#include<stdio.h>
#define MAX 100
#define MAXCOST 100000
int graph[MAX][MAX];
void prim(int graph[][MAX], int n)
{
int lowcost[MAX];
int mst[MAX];
int i, j, min, minid, sum = 0;
for (i = 2; i <= n; i++)
{
lowcost[i] = graph[1][i];
mst[i] = 1;
}
mst[1] = 0;
for (i = 2; i <= n; i++)
{
min = MAXCOST;
minid = 0;
for (j = 2; j <= n; j++)
{
if (lowcost[j] < min && lowcost[j] != 0)
{
min = lowcost[j];
minid = j;
}
}
printf("V%d-V%d=%d\n",mst[minid],minid,min);
sum += min;
lowcost[minid] = 0;
for (j = 2; j <= n; j++)
{
if (graph[minid][j] < lowcost[j])
{
lowcost[j] = graph[minid][j];
mst[j] = minid;
}
}
}
printf("最小权值之和=%d\n",sum);
}
int main()
{
int i, j, k, m, n;
int x, y, cost;
scanf("%d%d",&m,&n);
for (i = 1; i <= m; i++)
{
for (j = 1; j <= m; j++)
{
graph[i][j] = MAXCOST;
}
}
for (k = 1; k <= n; k++)
{
scanf("%d%d%d",&i,&j,&cost);
graph[i][j] = cost;
graph[j][i] = cost;
}
prim(graph, m);
return 0;
}