题目来源:
leetcode题目,网址:2477. 到达首都的最少油耗 - 力扣(LeetCode)
解题思路:
等到所有通过该城市抵达首都的人全部到达该城市时再出发时,油耗最少。
解题代码:
class Solution {
public:
long long minimumFuelCost(vector<vector<int>>& roads, int seats) {
roads=canGet(roads);
return getMin(roads,seats);
}
long long getMin(vector<vector<int>>& roads,int seats){
unordered_set<int> visited;
visited.insert(0);
return getCostAndPeople(roads,visited,seats,0)[0];
}
vector<long long> getCostAndPeople(vector<vector<int>>& roads,unordered_set<int>& visited,int seats,int pos){
vector<long long>res(2,0);//res[0]为到达的代价,res[1] 是到达的人数
res[1]=1;
for(int i=0;i<roads[pos].size();i++){
if(visited.count(roads[pos][i])!=0){ //已经出现过的节点不判断
continue;
}
visited.insert(roads[pos][i]);
vector<long long> temp=getCostAndPeople(roads,visited,seats,roads[pos][i]);
res[0]+=(temp[0]+temp[1]/seats+(temp[1]%seats==0?0:1));
res[1]+=temp[1];
}
return res;
}
vector<vector<int>> canGet(vector<vector<int>>& roads){
vector<vector<int>> res(roads.size()+1);
for(int i=0;i<roads.size();i++){
res[roads[i][0]].push_back(roads[i][1]);
res[roads[i][1]].push_back(roads[i][0]);
}
return res;
}
};
总结:
官方题解是贪心+深度优先搜索。另外因为只有 n-1 条路,只需要再下一个节点的可抵达节点中跳过该节点即可,不需要保存访问过的节点。