贪心算法---加油站

小鱼在乎2024-08-27 17:55

题目：

在一条环路上有 n 个加油站，其中第 i 个加油站有汽油 gas[i]升。

你有一辆油箱容量无限的的汽车，从第i个加油站开往第i+1个加油站需要消耗汽油 cost[i]升。你从其中的一个加油站出发，开始时油箱为空。

给定两个整数数组 gas 和 cost ，如果你可以按顺序绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1 。如果存在解，则保证它是唯一的。

思路：

如果总油量减去总消耗大于等于0，那么一定可以跑完一圈，说明各个站点的加油站剩油量rest[i]之和是大于等于0的。

每个加油站的剩油量rest[i]为gas[i]-cost[i]，i从0开始累加rest[i],和记为curSum，一旦curSum小于0，说明[0,i]区间都不能作为起始位置，因为这个区间选择任何一个位置作为起点，到这里都会断油，所以起点为i+1，curSum归0。

代码：

java 复制代码

    public int canCompleteCircuit(int[] gas, int[] cost) {
        int curSum=0;//记录从起点到现在的点剩余的油量
        int totalSum=0;//记录环路一周剩余的油量
        int index=0;//记录起始点下标
        for(int i=0;i<gas.length;i++){
            curSum+=gas[i]-cost[i];
            totalSum+=gas[i]-cost[i];
            if(curSum<0){
                index=(i+1)%gas.length;//如果不%，遍历到最后一个节点时index会溢出
                curSum=0;
            }
        }
        if(totalSum<0) return -1;
        return index;
    }