题目来源:https://leetcode-cn.com/problems/shortest-path-visiting-all-nodes/
大致题意:
给出一个无向图,求出以任意节点为起点,可以遍历所有节点的最短路径长度。
思路
状态压缩 + BFS
因为本题中只考虑路径长度,而不考虑具体路径,可用状态压缩的方法存储当前路径。
在本题中,有
状态压缩:节点个数为n,编号从0至n-1,则使用n位的二进制数存储已访问的路径。从低位到高位,若该位为1则表示当前位在路径中。
对于本题,使用二维布尔数组表示。第一维为当前路径最后的节点,第二维即为状态压缩后的矩阵。
BFS即为图的广度优先搜索。
- 使用队列存储接下来要遍历的节点。节点值由三元组[节点编号,当前的路径,当前路径长度]构成
- 使用大小为 n * 2^n 大小的布尔数组存储状态压缩后的路径
解题过程可概述为:
- 因为最短路径起点可能是任意节点,初始时,将所有节点对应的三元组[节点值,只有当前节点的路径,长度0]放入队列。
- 当队列不空时开始进行循环,每次队首元素出队
- 判断当前路径是否覆盖所有节点,若是则直接返回当前对应长度。因为BFS,第一个找到的路径即为最短路径。
- 若当前路径并未覆盖所有节点,获取当前节点的邻点集,判断每个加上邻点的路径是否查询过,若未查询过则将对应的三元组加入队列。
- 重复 2 - 4,直到队列为空。
代码:
public int shortestPathLength(int[][] graph) {
int n = graph.length;
// 状态压缩,用一个n位的二进制数表示走过的路径
Queue<int[]> queue = new LinkedList<int[]>(); // node path distance
// node path 表示路径最后节点为node的path是否被检查过
boolean[][] checked = new boolean[n][1 << n];
for (int i = 0; i < n; i++) { // 初始化,将每个节点作为起点加入队列
queue.offer(new int[]{i, 1 << i, 0});
checked[i][1 << i] = true; // 节点已加入队列,代表已经走过当前节点为起点,path为当前节点的路径
}
int ans = 0;
while (!queue.isEmpty()) {
int[] tuple = queue.poll();
int node = tuple[0];
int path = tuple[1];
int distance = tuple[2];
if (path == (1 << n) - 1) { // 第一个找到的即为最短路径
ans = distance;
break;
}
for (int target : graph[node]) { // BFS,广播查找当前节点的子节点
// 使用位运算或 | 将当前节点加入路径,因为路径中可能已经有该节点
int newPath = path | (1 << target);
if (!checked[target][newPath]) { // 若该子节点的这条path未查找过
queue.offer(new int[]{target, newPath, distance+1});
checked[target][newPath] = true;
}
}
}
return ans;
}
时间复杂度: O(n^2 · 2^n),广度优先搜索最差的复杂度为O(n ^ 2),这里使用了状态压缩,一共进行了2 ^ n次广搜。