LeetCode332. 重新安排行程

Posted woodwhale

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode332. 重新安排行程相关的知识,希望对你有一定的参考价值。

【LeetCode】332. 重新安排行程

前言

最近在刷回溯专题的题目,前几种题型(排列、组合、子集、子序列)如果直接套用模板,加一些剪枝,随便过的那种

但是这一题如果带入普通的回溯模板,可能无从下手,主要原因是从数组类型的题目转为图论的题目,需要构建图,相对来说比较麻烦,这题我是用回溯模板进行解答调试了2小时才搞完,写完之后发现很多东西其实没必要写上去,没搞懂就开始写那还是挺憨的

这一题能用回溯做出来的前置条件就是——第一次回溯到的结果一定要是字典序最小的情况,否则如果想把所有的路程跑出来,数据量大了根本不可能。这就涉及到了一些贪心的算法了,到下面的算法部分细讲

当然这一题最好的做法当然不是回溯,而是使用欧拉图的知识来进行解答,在算法部分我会详细讲解两种做法

题目

给你一份航线列表 tickets ,其中 tickets[i] = [fromi, toi] 表示飞机出发和降落的机场地点。请你对该行程进行重新规划排序。

所有这些机票都属于一个从 JFK(肯尼迪国际机场)出发的先生,所以该行程必须从 JFK 开始。如果存在多种有效的行程,请你按字典排序返回最小的行程组合。

  • 例如,行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小,排序更靠前。

假定所有机票至少存在一种合理的行程。且所有的机票 必须都用一次 且 只能用一次。

示例 1:

输入:tickets = [[“MUC”,“LHR”],[“JFK”,“MUC”],[“SFO”,“SJC”],[“LHR”,“SFO”]]
输出:[“JFK”,“MUC”,“LHR”,“SFO”,“SJC”]

示例 2:

输入:tickets = [[“JFK”,“SFO”],[“JFK”,“ATL”],[“SFO”,“ATL”],[“ATL”,“JFK”],[“ATL”,“SFO”]]
输出:[“JFK”,“ATL”,“JFK”,“SFO”,“ATL”,“SFO”]
解释:另一种有效的行程是 [“JFK”,“SFO”,“ATL”,“JFK”,“ATL”,“SFO”] ,但是它字典排序更大更靠后。

提示:

  • 1 <= tickets.length <= 300
  • tickets[i].length == 2
  • fromi.length == 3
  • toi.length == 3
  • fromi 和 toi 由大写英文字母组成
  • fromi != toi

思路

回溯

处理图

先使用一个map来存放fromto,其中from指向to,这样每次取到一个点,都可以查看它的指向。

需要注意的是,这个map的类型是Map<string, string[]>,也就是一个from点,可以指向多个to的点

处理坐标

在我们回溯的过程中,要对每个fromto列表进行所有的路径进行回溯,如何让路径从上一条路走向下一条路,同时又可以回来呢?

我这里定义了一个idx_map,专门用来处理from 这个点当前回溯到了哪一个to的路径,所以这个类型是Map<string[], number>,初始每个from点的对应的to列表的初始下标都是0,等到开始回溯一条路之后,这个值就加1,会u完成后进行减一,如果回溯过程中这个值超过了to的下标长度,那么直接结束回溯

处理次数

这里的次数指的是,同一点到另一点有几次,例如[["EZE","AXA"],["TIA","ANU"],["ANU","JFK"],["JFK","ANU"],["ANU","EZE"],["TIA","ANU"],["AXA","TIA"],["TIA","JFK"],["ANU","TIA"],["JFK","TIA"]]

其中["TIA","ANU"]出现了两次,也就是说需要经过两次才可以

所以定义一个cnt_map用来存放每个from点要去的to点的次数,定义的格式是Map<string, number>,其中,这里的key,也就是这个string存放的是出发点+到达点的形式,例如上述的例子中TIAANU --> 2,也就是表示TIA到ANU需要两次

判断是否经过了某条路线

这里处理的就是回溯中典型的used,其他题目中的used非常简单,一般使用一个数组来存储boolean值就可以了

但是这里需要处理图中经过的路线,所以我的处理方式是,将这个used写成一个Map<string, number>的形式,和上面的处理次数的cnt_map一样,每次回溯进过的fromto都会变成from+to --> cnt的键值对形式,如果我们的cnt超过了cnt_map中记录的最大经过数量,那么就直接结束回溯(道理很简单,如果不结束,那么就可以在两个点之间互相走来走去,成为死循环)

所以在回溯的过程中,判断是否走过该条路的used使用这种方式来判断就可以

判断回溯终止

回溯必定会有一个终止条件,如果到达这个条件就结束回溯

这题中的回溯终止条件就是经过的点的数量 == 飞机票的数量+1

可以自己先想想为啥这样就可以结束回溯了?

把飞机票当作边用来连接两个点,点肯定是比边多一个的(无论多复杂的路径都可以被简化成一条边

处理图的代码

为了让代码更清楚,我这里将处理图和处理回溯分开来将

首先是去处理图的code

function findItinerary(tickets: string[][]): string[] 
    const map: Map<string, string[]> = new Map()
    const idx_map: Map<string[], number> = new Map()
    const used: Map<string, number> = new Map()
    const cnt_map: Map<string, number> = new Map()

    tickets.forEach(([from, to]) => 
        const from_to: string = from+to
        if (cnt_map.has(from_to)) cnt_map.set(from_to, cnt_map.get(from_to)! + 1)
        else cnt_map.set(from_to, 1)
        if (!map.has(from)) 
            const arr: string[] = [to]
            map.set(from, arr)
            idx_map.set(arr, 0)    // 设置初始下标为0
        
        else map.get(from)!.push(to)
    )
    
    ...

以上代码就是处理图的部分,如何理解?

首先遍历所有的tickets,进去后对出现的from+to进行次数累加,同时判断map中是否存在这样的路线,不存在就新建一个数组存放,存在就在原先的数组中添加当前的to

同时,在第一次建立map的key的时候,把当前的idx_map对应的下标初始化成0

回溯的代码

我写的flashback函数就是回溯函数,判断结束的条件就是arr.length == tickets.length + 1

同时,如果map中存在当前开始走的点,那么就可以往下走,取出这个开始的点称为start,这个start点肯定对应一个to的数组,将这个数组称为nexts数组,也就是下一步可以去的地点的数组

这里有一个非常重要的点,nexts数组需要进行排序

排序的方式就是按照字典序升序排序,为啥要这样排序?

因为对每个nexts数组而言,如果下一次去的点的字典序最小,那么首次找到的完整路线一定是整个字典序最小的

所以排序是这一题的回溯算法的最重要的一环(如果没有排序这个步骤,那么会进行找很多次的过程,直到找到所有的路径,但实际很多案例有很多很多的路径,根本找不完,直接超时!)

理解了排序这个点之后,剩下的就直接套回溯的模板就行了

遍历所有的nexts节点,判断每次走是否走了回头路,是否走的路超过了对应路程机票的数量,同时判断处理下一步走的位置,回溯完成后将位置还原,将used还原…需要注意的是,每一层的时候,需要让idx_map的开始值重置为0,否则会找不到很多路径

这里还有一个细节,如果让回溯函数的返回值定为boolean类型,那么一旦找到第一个符合的路径,直接返回true就可以将后续的所有路径全部省略不走了,找不到就返回false

具体细节直接看代码:

const lens: number = tickets.length + 1
let res: string[] = []
const flashback = (arr: string[], start: string) => 
    if (arr.length == lens) 
        res = [...arr]
        return true
    
    if (map.has(start)) 
        // 如果可以往下走
        const nexts: string[] = map.get(start)!
              nexts.sort((a,b) => a < b ? -1 : 1)
        // 重置每次的map,否则idx会累加导致越界异常
        for (let key of idx_map.keys()) idx_map.set(key, 0)

        const idx: number = idx_map.get(nexts)!
              for (let i = idx; i < nexts.length; i++) 
                  // 不能走回头路
                  if (used.has(start+nexts[i]) && used.get(start+nexts[i])! >= cnt_map.get(start+nexts[i])!) continue

                  idx_map.set(nexts, i+1)                // to列表的下标+1
                  arr.push(nexts[i])
                  const cnt: number = used.has(start+nexts[i]) ? used.get(start+nexts[i])! : 0
                  used.set(start+nexts[i], cnt+1)

                  if (flashback(arr, nexts[i])) return true

                  idx_map.set(nexts, i)
                  arr.pop()
                  used.set(start+nexts[i], cnt)

              
    
    return false

完整代码

需要注意的是,这里回溯的调用是:flashback(["JFK"],"JFK"),表示每次开始从JFK开始走

function findItinerary(tickets: string[][]): string[] 
    const map: Map<string, string[]> = new Map()
    const idx_map: Map<string[], number> = new Map()
    const used: Map<string, number> = new Map()
    const cnt_map: Map<string, number> = new Map()

    tickets.forEach(([from, to]) => 
        const from_to: string = from+to
        if (cnt_map.has(from_to)) cnt_map.set(from_to, cnt_map.get(from_to)! + 1)
        else cnt_map.set(from_to, 1)
        if (!map.has(from)) 
            const arr: string[] = [to]
            map.set(from, arr)
            idx_map.set(arr, 0)    // 设置初始下标为0
        
        else map.get(from)!.push(to)
    )
    const lens: number = tickets.length + 1
    let res: string[] = []
    const flashback = (arr: string[], start: string) => 
        if (arr.length == lens) 
            res = [...arr]
            return true
        
        if (map.has(start)) 
            // 如果可以往下走
            const nexts: string[] = map.get(start)!
            nexts.sort((a,b) => a < b ? -1 : 1)
            // 重置每次的map,否则idx会累加导致越界异常
            for (let key of idx_map.keys()) idx_map.set(key, 0)

            const idx: number = idx_map.get(nexts)!
            for (let i = idx; i < nexts.length; i++) 
                // 不能走回头路
                if (used.has(start+nexts[i]) && used.get(start+nexts[i])! >= cnt_map.get(start+nexts[i])!) continue

                idx_map.set(nexts, i+1)                // to列表的下标+1
                arr.push(nexts[i])
                const cnt: number = used.has(start+nexts[i]) ? used.get(start+nexts[i])! : 0
                used.set(start+nexts[i], cnt+1)

                if (flashback(arr, nexts[i])) return true

                idx_map.set(nexts, i)
                arr.pop()
                used.set(start+nexts[i], cnt)

            
        
        return false
    
    flashback(["JFK"],"JFK")    // 从JFK开始走
    return res

回溯的效率感人,但是能过!(而且我这里处理图用了太多的map,导致内存消耗也很大)

欧拉通路

为什么和欧拉图扯上关系?

回想一下欧拉图的定义?

简答来说就是一笔画问题,一笔可以经过所有的边且不重复

具有欧拉回路的称为欧拉图

具有欧拉通路但没欧拉回路称为半欧拉图

回想完了再思考思考本题,为啥可以用欧拉图来解?

所有的机票看成 边, 我们需要将所有的 边 走完, 并且必须走一次且只能走一次

这不就是欧拉图的定义吗?

题目要求我们给出走过每个顶点的路径,不就是欧拉回路中走过的点的次序吗,只要找到一条字典序最短的欧拉回路就是我们想要的答案

同时由于规定了一定从JFK开始,只要从JFK开始找一条欧拉通路就可以了

如何解欧拉图中的欧拉回路?

著名的Hierholzer算法就可以用来解有向欧拉图中的欧拉回路(通路)

过程:

  1. 选择任一顶点为起点,遍历所有相邻边
  2. 深度搜索,访问相邻顶点。将经过的边都删除(重点
  3. 如果当前顶点没有相邻边,则将顶点入栈
  4. 栈中的顶点倒序输出,就是从起点出发的欧拉回路(通路)

当然如果将本题带入,起点一定是JFK

Hierholzer伪代码

可以当作一个模板记一下,使用递归非常的简单

const map: Map<string, string[]> = ...
const res: string[] = []
const dfs = (node: string) => 
    while (map.get(node).length) 
        dfs(map.get(node).pop())
    
    res.push(node)

dfs(node)
res.reverse() // 逆序后就是欧拉回路

上述伪代码的重点就是map.get(node).pop(),每次经过的边都去pop删除,这样下次dfs就不会有该边的记录

如何建图?

我们已经直到如何在欧拉图中使用Hierholzer来获取欧拉回路

现在的问题就是如何构建这个图,其实和回溯中的建图是一样的,只不过少了很多其他的内容,只需要记录fromtos就行了

建图的代码

const map: Map<string, string[]> = new Map()    // 图
tickets.forEach(([from, to]) => 
    // 设置from和to的映射关系
    if (map.has(from)) map.get(from).push(to)
    else map.set(from, [to])
)
for (const val of map.values()) 
    // 按照字典序降序,tos中的最后一个字典序最小
    val.sort((a,b) => a < b ? 1 : -1)

建完图,这一题的一半就做完了,剩下的一半就是确定最小的字典序

如何保证字典序?

在回溯的算法中,我们是每次对即将遍历的所有tos,在回溯之前进行了一次sort

我们在这里也需要进行sort,但是是按照什么方式排序呢?

其实我们sort完毕后的状态是一个降序状态,为什么是降序

因为在Hierholzer算法中,每次都是pop出栈顶元素,也就是list中最后的一个元素,我们想要保证每次走的路线的字典序最小,就要保证list中的最后一个元素的字典序最小,也就是降序

于是我们可以对刚刚建完图中的map进行一个列表元素降序排序

for (const val of map.values()) 
    // 按照字典序降序,tos中的最后一个字典序最小
    val.sort((a,b) => a < b ? 1 : -1)

写道这一部分,将map带入到我们的Hierholzer算法中,就可以求出字典序最小的欧拉回路啦!

带入Hierholzer

因为起点一定是JFK,我们直接从这个点开始进行dfs

let start: string = "JFK"
const res: string[] = []
const dfs = (node: string) => 
    while (map.has(node) && map.get(node).length) dfs(map.get(node).pop())
    res.push(node)

dfs(start)

这样就得到了最终的结果,只需要将res逆序输出就是最终的路径了!

完整代码

function findItinerary(tickets: string[][]): string[] 
    const map: Map<string, string[]> = new Map()    // 图
    tickets.forEach(([from, to]) => 
        // 设置from和to的映射关系
        if (map.has(from)) map.get(from).push(to)
        else map.set(from, [to])
    )
    for (const val of map.values()) 
        // 按照字典序降序,tos中的最后一个字典序最小
        val.sort((a,b) => a < b ? 1 : -1)
    
    // console.log(map)
    let start: string = "JFK"
    const res: string[] = []
    const dfs = (node: string) => 
        while (map.has(node) && map.get(node).length) dfs(map.get(node).pop())
        res.push(node)
    
    dfs(start)
    return (res.reverse())

比回溯的时间快了将近30ms,内存消耗也大大提高,只用一个map建图就好了

其余思考

这一题对于起点已经定死了是JFK,如果存在一个其他的点需要去寻找呢?

或者说如何在一个欧拉图中确定一个起点,可以让从这个起点开始顺利完成一笔画?

熟悉离散数学的同学一定会记得,可以通过入度出度这两个概念来确定起点,出度 = 入度 + 1的点一定可以当作一个起点(在欧拉图中)

如果给了某个提,需要确定起点同时确定一个欧拉通路,就可以使用的知识点来进行解答

没错,我说的这一题就是753. 破解保险箱

如果使用欧拉通路来解决这一题的同学,在753这道题也可以思考一下,如何解?当然,这题使用回溯DSF也是可以完成的,只不过一旦你领悟到了欧拉图的魅力,肯定是更倾向于使用数学知识而非暴力破解!

后话

由于大二离散数学学的本来就不行,所以从回溯的思想跳转到欧拉通路的思想上还是有很多需要改变的,如果思维好一点,融汇贯通,那么其实欧拉的数学方法还是更胜一筹的(在我看来

以上是关于LeetCode332. 重新安排行程的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode332. 重新安排行程

LeetCode332. 重新安排行程

LeetCode——重新安排行程

Leetcode 332 重新安排行程(欧拉路径)

重新安排行程

力扣算法JS LC [491. 递增子序列] LC [332. 重新安排行程]