Leetcode—2477.到达首都的最少油耗【中等】

源代码•宸

于 2023-12-05 10:59:42 发布

阅读量426

点赞数 3

分类专栏： LeetCode刷题文章标签： leetcode 深度优先算法 c++ 经验分享贪心

本文链接：https://blog.csdn.net/qq_44631615/article/details/134801794

版权

LeetCode刷题专栏收录该内容

356 篇文章 5 订阅

订阅专栏

2023每日刷题（五十）

Leetcode—2477.到达首都的最少油耗

在这里插入图片描述

算法思想

参考自灵茶山艾府
在这里插入图片描述

实现代码

class Solution {
public:
    long long minimumFuelCost(vector<vector<int>>& roads, int seats) {
        int n = roads.size() + 1;
        vector<int> g[n];
        for(auto &e: roads) {
            int x = e[0], y = e[1];
            g[x].emplace_back(y);
            g[y].emplace_back(x);
        }
        long long ans = 0;
        function<int(int, int)> dfs = [&](int x, int fa) {
            int sze = 1;
            for(auto y: g[x]) {
                if(y != fa) {
                    sze += dfs(y, x);
                }
            }
            // x 不是根节点
            if(x) {
                // (sze / seats)向上取整
                ans += (sze - 1) / seats + 1;
            }
            return sze;
        };
        dfs(0, -1);
        return ans;
    }
};