代码随想录算法训练营第 三十三天 |134. 加油站（ cpp）

加油站

题目链接

题目解答

暴力

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        for (int i = 0; i < cost.size(); i++) {
            int rest = gas[i] - cost[i]; // 记录剩余油量
            int index = (i + 1) % cost.size();
            while (rest > 0 && index != i) { // 模拟以i为起点行驶一圈（如果有rest==0，那么答案就不唯一了）
                rest += gas[index] - cost[index];
                index = (index + 1) % cost.size();
            }
            // 如果以i为起点跑一圈，剩余油量>=0，返回该起始位置
            if (rest >= 0 && index == i) return i;
        }
        return -1;
    }
};

说明

虽说这是暴力解法，但是还是有两点值得注意的。

• while用来模拟环形遍历合适一些，这个循环的 index=（index+1）%cost.size()；要注意了。
• 这个&&index==i是必须的，因为如果rest=0跳出循环且index！=i也是很有可能的。

贪心

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int curSum=0, sum=0;
        int index=0;
        for(int i=0; i<gas.size(); i++){
            curSum+=gas[i]-cost[i];
            sum+=gas[i]-cost[i];
            if(curSum<0){
                curSum=0;
                index=i+1;
            }
        }
        if(sum<0) return -1;
        return index;
    }
};

说明

i从0开始累加rest[i]，和记为curSum，一旦curSum小于零，说明[0, i]区间都不能作为起始位置，因为这个区间选择任何一个位置作为起点，到i这里都会断油，那么起始位置从i+1算起，再从0计算curSum。

135. 分发糖果

题目链接

解题代码

class Solution {
public:
    int candy(vector<int>& ratings) {
        vector<int> candyVec(ratings.size(),1);
        for(int i=1; i<ratings.size(); i++){
            if(ratings[i] > ratings[i-1]) candyVec[i]=candyVec[i-1]+1;
        }
        for(int i=ratings.size()-2; i>=0; i--){
            if(ratings[i] > ratings[i+1]) candyVec[i]=max(candyVec[i], candyVec[i+1]+1);
        }
        int sum=0;
        for(int i:candyVec) sum+=i;
        return sum;
    }
};

解题说明

先直接说，这题挺难，想不出，直接就看卡哥的题解

根据题解可知，对于这种两边一起考虑的题目，可以两边分别考虑，先确定一边在确定另一边，这才是解题的关键。

1. 首先右孩子比做孩子得分高，（从前往后遍历）。
2. 然后左孩子比右孩子得分高，从后往前遍历，有人会问，为啥呢？

因为 rating[5]与rating[4]的比较 要利用上 rating[5]与rating[6]的比较结果，所以 要从后向前遍历。

如果从前向后遍历，rating[5]与rating[4]的比较 就不能用上 rating[5]与rating[6]的比较结果了 。