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 result = 0;
for (int i = 0; i < candyVec.size(); i++) result += candyVec[i];
return result;
}
};
135. 分发糖果
最新推荐文章于 2024-03-28 10:03:45 发布
该代码实现了一个动态规划算法,用于解决糖果分配问题。根据ratings数组中每个孩子的评分,保证评分高的孩子至少获得比评分低的孩子多一个糖果。算法从前向后和从后向前遍历数组,更新candyVec,最后计算总糖果数。
摘要由CSDN通过智能技术生成