leetcode做题笔记2477. 到达首都的最少油耗

给你一棵 n 个节点的树（一个无向、连通、无环图），每个节点表示一个城市，编号从 0 到 n - 1 ，且恰好有 n - 1 条路。0 是首都。给你一个二维整数数组 roads ，其中 roads[i] = [ai, bi] ，表示城市 ai 和 bi 之间有一条 双向路 。

每个城市里有一个代表，他们都要去首都参加一个会议。

每座城市里有一辆车。给你一个整数 seats 表示每辆车里面座位的数目。

城市里的代表可以选择乘坐所在城市的车，或者乘坐其他城市的车。相邻城市之间一辆车的油耗是一升汽油。

请你返回到达首都最少需要多少升汽油。

思路一：DFS

c++解法

class Solution {
public:
    long long minimumFuelCost(vector<vector<int>>& roads, int seats) {
        vector<vector<int>> g(roads.size() + 1);
        for (auto& v : roads) {
            g[v[0]].push_back(v[1]);
            g[v[1]].push_back(v[0]);
        }
        auto ans = 0ll;
        auto dfs = [&](auto&& dfs, int node = 0, int fa = -1)->int {
            int sz = accumulate(g[node].begin(), g[node].end(), 
                     , [&](int a, int b) { return b != fa ? a + dfs(dfs, b, node) : a; });
            if (node) ans += (sz - 1) / seats + 1;
            return sz;
        };
        dfs(dfs);
        return ans;
    }
};