【LeetCode】每日一题 2023_12_5 到达首都的最少油耗（树，搜索）

戊子仲秋2023-12-05 10:32

文章目录

刷题前唠嗑
题目：到达首都的最少油耗
- 题目描述
- 代码与解题思路

刷题前唠嗑

LeetCode？启动！！！

题目：到达首都的最少油耗

题目链接：2477. 到达首都的最少油耗

题目描述

代码与解题思路

go 复制代码

func minimumFuelCost(roads [][]int, seats int) (ans int64) {
    g := make([][]int, len(roads)+1)
    for _, v := range roads { // g[x] 数组存的是与 x 相连的节点
        x, y := v[0], v[1]
        g[x] = append(g[x], y)
        g[y] = append(g[y], x)
    }
    var dfs func(int, int) int
    dfs = func(cur, father int) int {
        size := 1
        for _, child := range g[cur] {
            // 只从父节点向子节点搜索
            if child != father { 
                // v 从子节点变当前节点, cur 从当前节点变父节点, 统计子树大小
                size += dfs(child, cur) 
            }
        }
        if cur > 0 { // cur 不是根节点了, 可以计算油耗了
            ans += int64((size-1)/seats + 1)
        }
        return size
    }
    dfs(0, -1)
    return ans
}

这道题我自己想不出来解法，学了大佬的题解思路才做的，看到这一坨东西，我是真没什么好的思路，之前也没做过类似的题目

我来讲一下代码的流程和思路：

根据题目的节点编号是从 0 开始作为根，然后依次增加的性质，我们可以用一个二维数组把每一层的树连接到的节点存起来，举个例子：

g[0] 这个数组里面存储的就是所有和根节点 0 连通的节点
然后我们就从根节点开始搜索，if child != father 保证只想子节点搜索，size 记录子节点的大小（子节点的高度），如果当前节点不是根节点，那我们就可以开始累加计算题目要求的油耗了（我们就是根据子节点的大小和题目给的座位进行计算）