文章目录
- 刷题前唠嗑
- 题目：到达首都的最少油耗
- 题目描述
- 代码与解题思路
刷题前唠嗑
LeetCode？启动！！！
题目：到达首都的最少油耗
题目链接：2477. 到达首都的最少油耗
题目描述
代码与解题思路
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) intdfs = func(cur, father int) int {size := 1for _, 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 记录子节点的大小（子节点的高度），如果当前节点不是根节点，那我们就可以开始累加计算题目要求的油耗了（我们就是根据子节点的大小和题目给的座位进行计算）
