LeetCode-134. 加油站-Java-medium

题目链接

法一（贪心）

    /**
     * 法一（贪心）
     * 时间复杂度：O(n)
     * 空间复杂度：O(1)
     *
     * @param gas
     * @param cost
     * @return
     */
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int curSum = 0, totalSum = 0, start = 0;
        for (int i = 0; i < gas.length; i++) {
            curSum += gas[i] - cost[i];   // 一个或连续多个站两个数组之差的总和
            totalSum += gas[i] - cost[i]; // 所有站两个数组之差的总和
            if (curSum < 0) {  // 一旦curSum小于零，说明[0, i]区间都不能作为起始位
                start = i + 1; // 起始位置从i+1算起
                curSum = 0;    // curSum还原为初始状态0
            }
        }
        return totalSum < 0 ? -1 : start; //  totalSum必须大于等于0，否则不能完成绕行
    }

本地测试

        /**
         * 134. 加油站
         */
        lay.showTitle(134);
        Solution134 sol134 = new Solution134();
        int[] gas134 = new int[]{1, 2, 3, 4, 5};
        int[] cost134 = new int[]{3, 4, 5, 1, 2};
        System.out.println(Arrays.toString(gas134));
        System.out.println(Arrays.toString(cost134));
        System.out.println(sol134.canCompleteCircuit(gas134, cost134));