给你一棵 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;
}
};