文章目录
- 一、题目
- 二、解法
- 三、完整代码
所有的LeetCode题解索引,可以看这篇文章——【算法和数据结构】LeetCode题解。
一、题目
二、解法
思路分析:本题的思路是要相比较一边,然后在比较另外一边,左右两边一起比较的代码非常难写。每个孩子的糖果数量初始化为1。程序当中我们首先从前往后遍历,若第i个孩子评分大于第i-1个孩子,其糖果数量为第i-1个孩子的糖果数量+1。然后从后往前遍历,若第i个孩子比第i-1个孩子和第i+1个孩子评分都高,则要拿到比傍边孩子更多的糖果,只需要在先前的candyVec[i](先前的candyVec[i]数量必然大于第i-1个孩子的糖果数量)和第i+1个孩子糖果数量+1之间取最大值。
程序如下:
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); // 保证左边更大的孩子得到的糖果比左边和右边都多,取最大值
}
}
return accumulate(candyVec.begin(), candyVec.end(), 0);
}
};
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n)。
- 空间复杂度: O ( n ) O(n) O(n)。
三、完整代码
# include <iostream>
# include <vector>
# include <algorithm>
# include <numeric>
using namespace std;
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); // 保证左边更大的孩子得到的糖果比左边和右边都多,取最大值
}
}
return accumulate(candyVec.begin(), candyVec.end(), 0);
}
};
int main() {
vector<int> ratings = { 1,2,2 }; // 加油站的油量
Solution s1;
int result = s1.candy(ratings);
cout << result << endl;
system("pause");
return 0;
}
end