Prim算法实现最小生成树的思想是:在图中取一个顶点为起始点,找出其邻接的所有顶点,将该点和邻接的顶点和边的权值一一压入小根堆中,接着从小根堆中退出小根堆的根,将没访问过的两个顶点及其关联边的权值插入到最小生成树中,以此类推,总共需要循环n-1次。
小根堆模块:
int heapNum=0; //记录堆的结点个数
//堆的结点结构
struct Heap
{
int sta,en;
int weight;
} heap[100];
//下滑操作
void siftDown(int start,int end)
{
//将start号结点向下调整直到end
int i=start,j=2*i;
heap[0]=heap[i]; //用heap[0]来临时保存i结点的值
while(j<=end)
{
//有右孩子并且右孩子比左孩子小时,将j保存右孩子
if(j<end&&heap[j].weight>heap[j+1].weight) ++j;
//比j号结点小时,不需调整
if(heap[0].weight<=heap[j].weight)
break;
else
{
//向下调整
heap[i]=heap[j];
i=j;
j=2*j;
}
}
heap[i]=heap[0];
}
void siftUp(int start)
{
int j=start,i=j/2;
heap[0]=heap[j];
while(j>0)
{
if(heap[i].weight<=heap[0].weight)
break;
else
{
//向上调整工作
heap[j]=heap[i];
j=i;
i=i/2;
}
}
heap[j]=heap[0];
}
//插入操作的实现
bool insert(Heap temp)
{
++heapNum;
heap[heapNum]=temp;
siftUp(heapNum);
return true;
}
//删除操作
bool removeMin(Heap& temp)
{
//保留下根结点
temp=heap[1];
heap[1]=heap[heapNum]; //填补树根
--heapNum;
siftDown(1,heapNum); //将根结点下滑到尾部
return true;
}
图模块:
struct LinkNode
{
int vex; //邻接的结点在数组中的编号
LinkNode* next;
int weig; //结点的权值
};
//定义图结点的最大个数
const int MaxSize=10;
bool visited[10]={false};
struct Node
{
int data;
LinkNode* head;
//将结点邻接的链表头置为空
Node(){ head=0;}
} Adj[MaxSize],miniTree[MaxSize]; //Adj数组表示原来的图
//miniTree表示最小生成树
//建立图的算法
void createLink(int& numNode)
{
int numLink=0;
LinkNode* ptr;
cin>>numNode;
for(int i=1;i<=numNode;++i)
{
cin>>Adj[i].data;
cin>>numLink;
//头插入建表
for(int j=0;j<numLink;++j)
{
ptr=new LinkNode;
cin>>ptr->vex;
cin>>ptr->weig;
ptr->next=Adj[i].head;
Adj[i].head=ptr;
}
}
}
Prim算法模块:
//将图中v所关联的边存入堆中
void inHeap(int v)
{
LinkNode* ptr=0;
ptr=Adj[v].head;
//每个邻接点都有机会访问
while(ptr!=0)
{
if(!visited[ptr->vex])
{
Heap temp;
temp.sta=v;
temp.en=ptr->vex;
temp.weight=ptr->weig;
//将图中的边和所关联的两个结点压入堆中
insert(temp);
}
ptr=ptr->next; //到下个邻接点
}
}
//将两个结点及其对应关系插入最小生成树中
void insertTree(int st,int en,int weig)
{
LinkNode* ptr=new LinkNode;
ptr->vex=en;
ptr->weig=weig;
ptr->next=miniTree[st].head;
miniTree[st].head=ptr;
}
//求从u结点开始遍历生成的最小生成树的算法
void prim(int u)
{
int nodeNum=0;
createLink(nodeNum);
visited[u]=true;
int cntNum=1;
while(cntNum<nodeNum)
{
//将u所关联的边存入堆中
inHeap(u);
while(heapNum!=0)
{
Heap temp;
//取堆中权值最小的结点
removeMin(temp);
if(!visited[temp.en])
{
//将两个顶点及其关联边插入最小生成树中
miniTree[temp.sta].data=Adj[temp.sta].data;
miniTree[temp.en].data=Adj[temp.en].data;
insertTree(temp.sta,temp.en,temp.weight);
insertTree(temp.en,temp.sta,temp.weight);
u=temp.en;
visited[u]=true;
++cntNum;
break;
}
}
}
}