#coding=utf-8
#输出路径结果
wayList = []

#在该方法中完成寻路算法,并将包含起点和终点在内的路径序号(int类型)按照顺序放入wayList数组中
# nodeCount:抽象图中的节点总数量
# costs:二维权值数组,表示序号0~nodeCount节点之间的权值关系,元素为int类型,比如使用data.costs[0][2]获得第一个节点和第三个节点之间的权值
# startIndex:起始节点序号
# endIndex:终点节点序号
def FindPath(nodeCount, costs, startIndex, endIndex):
    visited = [False] * nodeCount
    path = []
    
    def dfs(node):
        visited[node] = True
        path.append(node)
        
        if node == endIndex:
            wayList.extend(path)
            return True
        
        for i in range(nodeCount):
            if not visited[i] and costs[node][i] != -1:
                if dfs(i):
                    return True
        
        path.pop()
        return False
    
    dfs(startIndex)
    return wayList

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐