LintCode-187. 加油站

题目

描述

在一条环路上有 N 个加油站，其中第 i 个加油站有汽油gas[i]，并且从第i个加油站前往第i+1个加油站需要消耗汽油cost[i]。
你有一辆油箱容量无限大的汽车，现在要从某一个加油站出发绕环路一周，一开始油箱为空。
求可环绕环路一周时出发的加油站的编号，若不存在环绕一周的方案，则返回-1。

注意事项：数据保证答案唯一。

样例

现在有4个加油站，汽油量gas[i]=[1, 1, 3, 1]，环路旅行时消耗的汽油量cost[i]=[2, 2, 1, 1]。则出发的加油站的编号为2。

解答

思路

油箱置空，从每个加油站出发，加油，减去消耗，出现小于零说明不能从该加油站出发。没有出现说明可以从这个加油站出发。
注意“环路”，用取余的方法循环。

代码

public class Solution {
    /**
     * @param gas: An array of integers
     * @param cost: An array of integers
     * @return: An integer
     */
    public int canCompleteCircuit(int[] gas, int[] cost) {
        // write your code here
        for(int i = 0; i < gas.length; i++){
            int tank = 0;
            for(int j = 0; j < gas.length; j++){
                tank += gas[(i + j) % gas.length];
                tank -= cost[(i + j) % gas.length];
                if(tank < 0){
                    break;
                }
            }
            if(tank >= 0) return i;
        }
        return -1;
    }
}