加油站(LeetCode134)---C++实现

在这里插入图片描述
思路:

1.首先介绍一个定理,如果一个整数序列累加和非负,那么一定可以找到一个起点,以该起点开始的序列累加和都非负。
2.由1知,只要gas累加和>=cost累加和 解一定存在,反之,一定无解
3.进行一次遍历,totalGas记录每个加油站gas[i]-cost[i]的和,curGas记录当前油箱油量,startLoc记录符合要求的起点加油站,所以知道当curGas[i]
< 0时,一定不能以i之前的加油站为起点,此时油量需要重置为0,起点startLoc要走到下一个加油站。依次类推,到末尾看totalGas是否>=0,如果是,则startLoc为答案;反之,则说明gas数组总和小于cost数组总和,不存在解,返回-1.

 int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int totalGas, curGas;
        totalGas = curGas = 0;
        int beginLoc = 0;
        for (int i = 0; i < gas.size(); i++) {
            totalGas += gas[i] - cost[i];
            curGas += gas[i] - cost[i];
            if (curGas < 0) {
                beginLoc = i + 1;
                curGas = 0;
            }
        }
        if (totalGas >= 0)   //等于0不能忘记
            return beginLoc;
        return -1;

猜你喜欢

转载自blog.csdn.net/qq_42673507/article/details/90743779
今日推荐