1129. 颜色交替的最短路径(leetcode,广搜,可重做)-------------------Java实现

1129. 颜色交替的最短路径(leetcode,广搜,可重做)-------------------Java实现

题目表述

给定一个整数 n,即有向图中的节点数,其中节点标记为 0 到 n - 1。图中的每条边为红色或者蓝色,并且可能存在自环或平行边。

给定两个数组 redEdges 和 blueEdges,其中:

redEdges[i] = [ai, bi] 表示图中存在一条从节点 ai 到节点 bi 的红色有向边,

blueEdges[j] = [uj, vj] 表示图中存在一条从节点 uj 到节点 vj 的蓝色有向边。

返回长度为 n 的数组 answer,其中 answer[X] 是从节点 0 到节点 X 的红色边和蓝色边交替出现的最短路径的长度。如果不存在这样的路径,那么 answer[x] = -1。

样例

n =5

redEdges =[[0,1],[1,2],[2,3],[3,4]]

blueEdges =[[1,2],[2,3],[3,1]]

条件

1 <= n <= 100

0 <= redEdges.length, blueEdges.length <= 400

redEdges[i].length == blueEdges[j].length == 2

0 <= ai, bi, uj, vj < n

思路

通过一个二维[n][2]List来记录当前点可以到达的下个点。

广搜,搜索记录到达点的长度。

注意点

x^1 可实现 x1时返回0,x0时返回1,异或运算。

ac代码

Java:

java 复制代码
package leetcode1129;

import java.util.*;

class Solution {
    public int[] shortestAlternatingPaths(int n, int[][] redEdges, int[][] blueEdges) {
    List<Integer> target[][] = new List[n][2];
    int[][] visit = new int[n][2];
    int max = 10000009;
    for (int i=0;i<n;i++) {
        target[i][0] = new ArrayList<Integer>();
        target[i][1] = new ArrayList<Integer>();
        visit[i][0] = max;
        visit[i][1] = max;
    }
    for (int[] x:redEdges)
        target[x[0]][0].add(x[1]);
        for (int[] x:blueEdges)
            target[x[0]][1].add(x[1]);
        Queue<Integer[]> q = new LinkedList<>();
        q.offer(new Integer[]{0,0});
        q.offer(new Integer[]{0,1});
        int this_step = 0;
        while(!q.isEmpty())
        {
            int this_longth = q.size();
            while(this_longth--!=0)
            {
                Integer[] now_sit = q.poll();
                if (visit[now_sit[0]][now_sit[1]]!=max)
                    continue;
                else
                    visit[now_sit[0]][now_sit[1]]=this_step;
                for (Integer now:target[now_sit[0]][now_sit[1]^1])
                    q.offer(new Integer[]{now,now_sit[1]^1});
            }
            this_step++;
        }
        int[] result = new int[n];
        for (int i=1;i<n;i++) {
            result[i] = Math.min(visit[i][0], visit[i][1]);
            if (result[i]==max)
                result[i] =-1;
        }
        return result;
    }
}

public class leetocde1129 {
    public static void main(String[] args) {
        int n = 5;
        int[][] red_edges = new int[][]{
                {0,1},{1,2},{2,3},{3,4}
        };
        int[][] blue_edges = new int[][]{{1,2},{2,3},{3,1}};
        Solution s = new Solution();
        int[] result = s.shortestAlternatingPaths(n,red_edges,blue_edges);
    for (int x:result)
        System.out.println(x+" ");
    };



        }
相关推荐
孤狼warrior16 分钟前
灰色预测模型
人工智能·python·算法·数学建模
京东云开发者17 分钟前
京东零售基于国产芯片的AI引擎技术
算法
Dcs17 分钟前
VSCode等多款主流 IDE 爆出安全漏洞!插件“伪装认证”可执行恶意命令!
java
保持学习ing23 分钟前
day1--项目搭建and内容管理模块
java·数据库·后端·docker·虚拟机
京东云开发者34 分钟前
Java的SPI机制详解
java
超级小忍1 小时前
服务端向客户端主动推送数据的几种方法(Spring Boot 环境)
java·spring boot·后端
程序无bug1 小时前
Spring IoC注解式开发无敌详细(细节丰富)
java·后端
小莫分享1 小时前
Java Lombok 入门
java
程序无bug1 小时前
Spring 对于事务上的应用的详细说明
java·后端
食亨技术团队1 小时前
被忽略的 SAAS 生命线:操作日志有多重要
java·后端