1129. 颜色交替的最短路径

题解：
BFS 宽搜

用宽搜求最短路

public int[] shortestAlternatingPaths(int n, int[][] redEdges, int[][] blueEdges) {
        int[] res = new int[n];
        res[0] = 0;
        // 有两种颜色，建立两种颜色的边图
        List<Integer>[][] mat = new ArrayList[2][n];
        for (int i = 0; i < 2; i++) {
            for (int j = 0; j < n; j++) {
                mat[i][j] = new ArrayList<>();
            }
        }
        // 添加该点以红色边能到的点
        for (int[] edge : redEdges) {
            mat[0][edge[0]].add(edge[1]);
        }
        // 添加该点以蓝色边能到的点
        for (int[] edge : blueEdges) {
            mat[1][edge[0]].add(edge[1]);
        }
        // 有两种颜色，对于点x，它可能是通过前一个点以蓝色 or 红色 边到达
        int[][] dis = new int[2][n];
        for (int i = 0; i < 2; i++) {
            Arrays.fill(dis[i], 0x3f3f3f3f);
        }
        // 初始值
        dis[0][0] = 0;
        dis[1][0] = 0;
        // 宽搜开始
        Queue<int[]> q = new ArrayDeque<>();
        q.offer(new int[]{0, 0});
        q.offer(new int[]{1, 0});
        while (!q.isEmpty()) {
            int[] poll = q.poll();
            int color = poll[0];
            int point = poll[1];
            // 1-color ： 红色换成蓝色，蓝色换成红色
            for (Integer next : mat[1 - color][point]) {
                // 如果该点被更新过就不用再更新了，第一次更新时，肯定是最小值了
                // 注意该题有环路，没有这个判断，可能会死循环
                if (dis[1-color][next] != 0x3f3f3f3f) {
                    continue;
                }
                // 更新最短路
                dis[1 - color][next] = Math.min(dis[1 - color][next], dis[color][point] + 1);
                q.offer(new int[]{1 - color, next});
            }
        }
        // 处理答案
        for (int i = 1; i < n; i++) {
            res[i] = Math.min(dis[0][i], dis[1][i]);
            if (res[i] == 0x3f3f3f3f) res[i] = -1;
        }
        return res;
    }