【LeetCode】每日一题 2023_12_5 到达首都的最少油耗(树,搜索)

文章目录

刷题前唠嗑


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
}

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

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

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

    g[0] 这个数组里面存储的就是所有和根节点 0 连通的节点

  2. 然后我们就从根节点开始搜索,if child != father 保证只想子节点搜索,size 记录子节点的大小(子节点的高度),如果当前节点不是根节点,那我们就可以开始累加计算题目要求的油耗了(我们就是根据子节点的大小和题目给的座位进行计算)

相关推荐
程序员黄同学1 小时前
贪心算法,其优缺点是什么?
算法·贪心算法
SsummerC2 小时前
【leetcode100】每日温度
数据结构·python·leetcode
仙人掌_lz2 小时前
机器学习ML极简指南
人工智能·python·算法·机器学习·面试·强化学习
Swift社区2 小时前
Swift LeetCode 246 题解:中心对称数(Strobogrammatic Number)
开发语言·leetcode·swift
巷北夜未央3 小时前
Python每日一题(13)
开发语言·python·算法
独好紫罗兰3 小时前
洛谷题单3-P5720 【深基4.例4】一尺之棰-python-流程图重构
开发语言·python·算法
对方正在长头发丿4 小时前
LETTERS(DFS)
c++·笔记·算法·深度优先·图论
Qian_ShouYi4 小时前
MATLAB 代码学习
学习·算法·matlab
小样vvv4 小时前
【面试篇】JVM
jvm·面试·职场和发展
lovebugs4 小时前
K8s面试第一篇:初识Kubernetes——核心概念与组件详解
后端·算法·面试