lua写普里姆最小生成树算法

--普里姆最小生成树算法
local function MiniSpanTree_Prim( MGraph_G )
	local MAXSIZE = #MGraph_G
	if not MGraph_G[MAXSIZE][MAXSIZE] then		--判断邻接矩阵是否存在(是一个二维数组)
		return
	end
	local NodeTable = {}        		--顶点下标表
	local CostTable = {} 				--顶点间的边的权值表
	local MiniCost,NodeSubscript		--分别用来保存最小权值和顶点下标
	NodeTable[1] = 1					--初始化第一个顶点下标为1
	CostTable[1] = 0					--初始化第一个权值为0(为0代表已经走过或执行过)，表示第一个顶点加入生成树
	for i = 2,MAXSIZE do				--循环遍历除下标为1的其余顶点
		NodeTable[i] = 1				--初始化都为第一个顶点的下标
		CostTable[i] = MGraph_G[1][i]	--将第一个顶点与之有边的权值存入数组
	end
	for i = 1,MAXSIZE do
		MiniCost = 65535
		--找出当前行最小的权值，以及对应的下标，保存
		local j = 1,k = 0
		while j < MAXSIZE 
		do
			if CostTable[j] ~= 0 and CostTable[j] < MiniCost then
				MiniCost = CostTable[j]
				k = j
			end
			j = j + 1
		end
		--任务:打印当前顶点的边中，权值最小的一条边(也可以是其他操作)
		print(string.format("(%d,%d)",NodeTable[k],k))
		--打印过后，将当前顶点的权值置为0，表示已经完成任务
		CostTable[k] = 0
		--下标为K的顶点的所有边的权值，如果有小于当前权值表里面的值的，就用其替换。
		for j = 1,MAXSIZE do
			if CostTable[j] ~= 0 and MGraph_G[k][j] < CostTable[j] then
				NodeTable[j] = k				--将k的下标存入顶点下标表
				CostTable[j] = MGraph_G[k][j]	--替换权值
			end
		end
	end
end