$#LintCode : 加油站
题目
在一条环路上有 N 个加油站,其中第 i 个加油站有汽油gas[i],并且从第i个加油站前往第i+1个加油站需要消耗汽油cost[i]。
你有一辆油箱容量无限大的汽车,现在要从某一个加油站出发绕环路一周,一开始油箱为空。
求可环绕环路一周时出发的加油站的编号,若不存在环绕一周的方案,则返回-1。
样例
现在有4个加油站,汽油量gas[i]=[1, 1, 3, 1],环路旅行时消耗的汽油量cost[i]=[2, 2, 1, 1]。则出发的加油站的编号为2。
思路
最暴力的 O(n2) 的算法就是从起点开始检查是否可以绕一圈,碰到不足以走到下一个加油站的则从起点+1继续检查,直到检查完n个加油站。
当然这题是有O(n)的解法的,不用那么暴力。我们知道,从起点start开始走,当我们走到第i个加油站发现不足以走到i+1个加油站时我们不需要从start+1开始查找,因为既然从start开始到不了i+1那么从start到i中间的任意一节点开始都是到不了i+1的,所以下一个我们就可以从i+1开始检查,直到发现满足条件的。
代码
int canCompleteCircuit(vector<int> &gas, vector<int> &cost) {
int start = 0;
while(start != gas.size()) {
int total = 0;
int firsttime = 1;
int once = 0;
for(int i = start; i < gas.size(); i++) {
if(i == start && firsttime == 0)
return start;
total += gas[i] - cost[i];
if(total < 0) {
start = i + 1;
break;
}
if(i == gas.size() - 1) {
i = -1;
once = 1;
}
firsttime = 0;
}
if(once == 1)
break;
}
return -1;
}
(因为这里是要转圈的,所以我用了一个once的标志来标记i是否从头开始扫描了,要不然start随着i变的话就一直不会到最后一个了。)