LeetCode
- 题目地址:https://leetcode.com/problems/gas-station/#/solutions
问题描述&解题思路:有n个加油站围成一个圈,其中gas[i]表示第i个加油站的油,cost[i]表示从i到i+1需要耗费的油,然后你的车的容量是无限的,问车从哪个站开始(i从0到size-1)能绕加油站一圈。潜在的意思是,从哪个站开始,我可以绕一圈且每时每刻,车里的汽油大于等于0(这样车才能开下去!)。
- 于是首先计算dif[i] = gas[i]-cost[i],就是车从i到i+1油量会增加多少(负数表示减少),其中总量sum不得小于0,因为加了dif[i]之后sum小于0,说明到达不了i+1个加油站,而到达第i个加油站则由上一次循环中来判断。
- 是否已经绕了一个圈由end和begin的差来决定,因为是绕一圈即绕回原点,所以之差要是size。
- 环的问题就用遍历2次dif来实现。
代码
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
int size = gas.size();
vector<int> dif;
//cal dif
for (int i = 0; i < size; i++) {
dif.push_back(gas[i]-cost[i]);
}
int sum = 0;
int begin = 0, end = 0;
for (int i = 0; i < size*2; i++) {
sum += dif[i%size];
//can't move to i+1
if (sum < 0) {
sum = 0;
begin = i+1;
continue;
} else {
end = i;
}
if (end - begin >= size - 1)
return begin;
}
return -1;
}
};