找到最短路径的 K 次跨越

/**
 * LeetCode 2714: 找到最短路径的 K 次跨越
 * 
 * 题目思路:
 * 1. 使用广度优先搜索（BFS）算法，适用于寻找无权图中的最短路径。
 * 2. 利用一个队列来进行层次遍历，每层代表跨越次数。
 * 3. 使用一个三维数组来记录节点是否访问过以及访问时的最小跨越次数。
 * 4. 对于每个节点，遍历其所有邻居节点，进行状态转移并记录最小跨越次数。
 * 5. 当找到目标节点时，返回其最小跨越次数。
 */
import java.util.*;
import java.util.stream.IntStream;

public class SolutionStream {
    public int shortestPathWithKCrossings(int[][] graph, int start, int end, int k) {
        int n = graph.length;
        int[][][] visited = new int[n][n][k + 1];
        Queue<int[]> queue = new LinkedList<>();
        queue.add(new int[]{start, 0, 0}); // {当前节点, 当前跨越次数, 当前路径长度}
        while (!queue.isEmpty()) {
            int[] curr = queue.poll();
            int node = curr[0];
            int crossings = curr[1];
            int length = curr[2];
            if (node == end) return length;
            if (crossings > k) continue;
            IntStream.range(0, n)
                     .filter(next -> graph[node][next] == 1 && visited[node][next][crossings + 1] == 0)
                     .forEach(next -> {
                         visited[node][next][crossings + 1] = 1;
                         queue.add(new int[]{next, crossings + 1, length + 1});
                     });
        }
        return -1; // 如果无法到达目标节点
    }
}