开启我的LeetCode刷题日记:2045. 到达目的地的第二短时间

90 阅读3分钟

「这是我参与2022首次更文挑战的第7天,活动详情查看:2022首次更文挑战

编程世界总是离不了算法

最近在看框架源码时,会有很多算法的实现逻辑,有时候会感到吃力

于是决定蹭着假期,加强算法和数据结构相关的知识

那怎么提升呢?

其实我知道算法这东西没有捷径,多写多练才能提升,于是我开启我的LeetCode刷题之旅

第一阶段目标是:200道,每天12

为了不乱,本系列文章目录分为三部分:

  1. 今日题目:xxx
  2. 我的思路
  3. 代码实现

今天题目:2045. 到达目的地的第二短时间

难度:困难

城市用一个 双向连通 图表示,图中有 n 个节点,从 1 到 n 编号(包含 1 和 n)。图中的边用一个二维整数数组 edges 表示,其中每个 edges[i] = [ui, vi] 表示一条节点 ui 和节点 vi 之间的双向连通边。每组节点对由 最多一条 边连通,顶点不存在连接到自身的边。穿过任意一条边的时间是 time 分钟。

每个节点都有一个交通信号灯,每 change 分钟改变一次,从绿色变成红色,再由红色变成绿色,循环往复。所有信号灯都 同时 改变。你可以在 任何时候 进入某个节点,但是 只能 在节点 信号灯是绿色时 才能离开。如果信号灯是  绿色 ,你 不能 在节点等待,必须离开。

第二小的值 是 严格大于 最小值的所有值中最小的值。

例如,[2, 3, 4] 中第二小的值是 3 ,而 [2, 2, 4] 中第二小的值是 4 。 给你 n、edges、time 和 change ,返回从节点 1 到节点 n 需要的 第二短时间 。

注意:

你可以 任意次 穿过任意顶点,包括 1 和 n 。 你可以假设在 启程时 ,所有信号灯刚刚变成 绿色 。

我的思路

将路径数组 edges 转为邻接矩阵便于后续搜索

转换过程中注意本题是双向连通图,任意边上的两个点都可以作为出发点

利用BFS求最短路径步数 min1 和次短路径步数 min2,BFS过程中需要注意剪枝:

剪枝1:若 min2 已经找到则退出

剪枝2:若从当前节点到达下一节点的步数 step+1 与到达下一节点的最小步数 minSteps.get(k) 之差不小于2 step + 1 - minSteps.get(k) >= 2,则跳过

剪枝3:同一层级每个节点只出现一次

按最短路径逐步叠加通过时间和等待红灯时间

代码实现

var secondMinimum = function (n, edges, time, change) {
    edges = edges.reduce((a, c) => {
        !a[c[0]] ? (a[c[0]] = [c[1]]) : a[c[0]].push(c[1]);
        !a[c[1]] ? (a[c[1]] = [c[0]]) : a[c[1]].push(c[0]);
        return a;
    }, new Array(n + 1));

    let min1 = Infinity, min2 = Infinity, nodes = new Set([1]), step = 1;
    const minSteps = new Map();

    while (nodes.size && min2 === Infinity) {
        const nextNodes = new Set();
        for (const i of nodes) {
            minSteps.set(i, step);
            if (i === n) min1 === Infinity || min1 === step ? (min1 = step) : (min2 = step);
            for (const k of edges[i]) {
                if (minSteps.has(k) && step + 1 - minSteps.get(k) >= 2) continue;
                nextNodes.add(k);
            }
        }

        step++;
        nodes = nextNodes;
    }

    let cost = 0;
    for (let i = 1; i < Math.min(min1 + 2, min2); i++) {
        if (((cost / change) >> 0) % 2) cost += change - (cost % change);
        cost += time;
    }

    return cost;
};

总结

实现方式其实有很多,这里仅供参考~

由于刚开始刷题,也不知道从哪里刷好,如果前辈们有好的建议,希望不吝赐教,感谢🌹