LeetCode-134-加油站-贪心思路

题目描述:
在一条环路上有 n 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas 和 cost ,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1 。如果存在解,则 保证 它是 唯一 的。

解题思路:

  1. 需要先计算出 gas[]和 cost[] 对应元素的差diff[],表示走到下一个加油站时还剩的油量;
  2. 需要三个变量,curSum统计当前剩余油量;totalSum统计diff[]总的油量,如果小于0,说明不管从哪开始,都不可能跑完一圈,直接返回-1即可;start记录起始下标,当 curSum <0 就更新到下一个,从下一个继续开始。

题目链接

代码实现:

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int len = gas.length;
        int[] diff = new int[len];// gas[i]-cost[i]的值,表示剩余油量
        int curSum = 0;// 表示当前的总剩余油量
        int totalSum = 0;// diff[]的总的剩余油量,如果 totalSum<0,表示不管从哪开始,都不可能跑完一圈
        int start = 0;// 表示起始位置
        for (int i = 0; i < len; i++) {
            diff[i] = gas[i]-cost[i];
            totalSum += diff[i];
        }
        if (totalSum < 0){
            return -1;
        }
        for (int i = 0; i < len; i++) {
            curSum += diff[i];
            if (curSum < 0){// 表示 i 之前的位置和小于0,不适合作为起始位置
                start = i+1;// 更新起始位置
                curSum = 0;// 清 0
            }
        }
        return start;
    }
}

你可能感兴趣的:(LeetCode,leetcode,贪心算法)