加油站
题目解答
暴力
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;
}
};
解题说明
先直接说,这题挺难,想不出,直接就看卡哥的题解
根据题解可知,对于这种两边一起考虑的题目,可以两边分别考虑,先确定一边在确定另一边,这才是解题的关键。
- 首先右孩子比做孩子得分高,(从前往后遍历)。
- 然后左孩子比右孩子得分高,从后往前遍历,有人会问,为啥呢?
因为 rating[5]与rating[4]的比较 要利用上 rating[5]与rating[6]的比较结果,所以 要从后向前遍历。
如果从前向后遍历,rating[5]与rating[4]的比较 就不能用上 rating[5]与rating[6]的比较结果了 。