算法——加油站

在一条环路上有 N 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。
如果你可以绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1。
leetcode

解题思路:依次遍历出发点,然后出发,如果当前油量小于0,则从这次出发点到当前点都不可能是答案,则选下一个点为出发点。

如果之前走过的会产生油量小于零的部分都会有解,那么后面的就一定也有解,由于答案是唯一的,所以,之前走过的无法产生答案的地方是没有解的。

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int n = gas.length;

        for(int i = 0, j = 0; i < n; i += j + 1) {
            int gases = 0;
            for(j = 0; j < n; j++) {
                int k = (i + j) % n;
                gases += gas[k] - cost[k];
                if(gases < 0) break;
            }

            if(j >= n) return i;
        }

        return -1;
    }
}
posted @ 2020-11-18 19:30  lippon  阅读(104)  评论(0编辑  收藏  举报