题目详情
类型:贪心算法中的分配问题
思路:想到了用从左到右再从右到左两次遍历来更新糖果数,但过程中犯了两个错误:
- 最初想当然的把比左/右边rating大的就直接加1,没考虑糖果数量也得比左/右的大。
- 遍历时比较更新糖果数的对象错了,应该是更新后面要继续和别人比的那个人的糖果数,而不是比完了的那个人的糖果数,不然没有起到串联作用。
class Solution {
public:
int candy(vector<int>& ratings) {
int ans = 0, n = ratings.size();
vector<int>s(n,1);
for(int i = 1; i < n; i++){
if(ratings[i] > ratings[i-1] && s[i] <= s[i-1]) s[i] = s[i-1]+1;
}//&&后的比较代码可以删除,第一次遍历不需要比较。
for(int i = n-2; i >= 0; i--){
if(ratings[i] > ratings[i+1] && s[i] <= s[i+1]) s[i] = s[i+1]+1;
ans+=s[i];//结果的计算可以用std:accumulate计算很方便
}
return ans+s[n-1];
}
};
vector v1, v2( 20 );
vector ::iterator Iter1, Iter2;//使用迭代器
accumulate在头文件 #include 里
主要是用来累加容器里面的值,比如int、string之类,可以少写一个for循环
比如直接统计 vector v 里面所有元素的和:(第三个参数的0表示sum的初始值为0)
int sum = accumulate(v.begin(), v.end(), 0);
比如直接将 vector v 里面所有元素一个个累加到string str中:(第三个元素表示str的初始值为空字符串)
string str = accumulate(v.begin(), v.end(), “”);