最小生成树Prim算法

template<class T>
bool Graph<T>::Prim(const T& v,PathData* E,int ne)
{
	int s=FindNode(v);
	if(s==-1)
	{
		return 0;
	}

	PathData item;
	double cost;
	int id=0;
	for(int i=0;i<=ne;i++)//初始化边数组 E
	{
		if(i!=s)
		{
			item.start=s;
			item.dest=i;
			cost=GetCost(s,i);

			item.cost=(cost!=0?cost:MAXCOST);
			E[id++]=item;
		}
	}

	int count=0;
	int ns;
	BuildHeap(E,ne);
	for(int i=0;i<ne;i++)//更新入选子网
	{
		if(E[0]<MAXCOST)
			count++;
		ns=E[0].dest;
		for(int j=1;j<ne-i;j++)//更新边集
		{
			cost=GetCost(ns,E[j].dest);
			cost=(cost!=0?cost:MAXCOST);
			if(E[j]>cost)
			{
				E[j].cost=cost;
				E[j].start=ns;
			}
		}
		item=E[0];
		E[0]=E[ne-1-i];
		E[ne-1-i]=item;
		BuildHeap(E,ne-1-i);
	}
	return (count==ne?1:0);
}
template<class T>
void Graph<T>::PercolateDown(PathData* E,int pos,int size)//辅助BuildHeap
{
	int p=pos,c=2*p+1;
	PathData temp=E[p];
	while(c<size)
	{
		if(c+1<size&&E[c+1]<E[c])
		{
			c++;
		}
		if(E[c]>temp)
			break;
		else
		{
			E[p]=E[c];
			p=c;
			c=2*p+1;
		}
	}

	E[p]=temp;
}
template<class T>
void Graph<T>::BuildHeap(PathData* E,int size)//将E调成小根堆
{
	for(int i=size/2-1;i>=0;i--)
	{
		PercolateDown(E,i,size);
	}
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值