分发糖果
题目
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
- 每个孩子至少分配到 1 个糖果。
- 相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
动态规划
我采用的是动态规划的方法,设dp[i]表示前i+1个分数时的最优策略
有如下三种情况:
- 如果第i个分数大于第i-1个分数,自然:dp[i] = dp[i-1] + 1
- 如果第i个分数小于第i-1个分数,并且dp[i-1]=1,那么需要将dp[i-1]++,并且i-1前面对dp[i-1]有影响的值,都要进行自增更新,dp[i]=1
- 如果上述条件均不满足:dp[i]=1
最后求和即可
Java代码实现
class Solution {
public int candy(int[] ratings) {
int len = ratings.length;
int[] dp = new int[len];
dp[0] = 1;
for (int i=1;i<len;i++){
if (ratings[i]>ratings[i-1]) dp[i] = dp[i-1] + 1;
else {
if (ratings[i]<ratings[i-1]&&dp[i-1]==1) plus(ratings,dp,i-1);
dp[i] = 1;
}
}
int sum = 0;
for (int c:dp) sum+=c;
return sum;
}
private void plus(int[] ratings,int[] dp, int i) {
dp[i]++;
while (i>0&&ratings[i]<ratings[i-1]&&dp[i-1]-dp[i]==0){
dp[i-1]++;
i--;
}
}
}
贪心
leetcode上比较好的题解:
- 先对candy数据均用1填充
- 遍历ratings数组,如果右边的值大于左边的值,candy[i] = candy[i-1]+1;
- 再遍历ratings数组,如果左边的值大于右边的值,取candy[i] = max(candy[i + 1] + 1, candy[i])
- 再对candy数组求和即可
时间复杂度仅为O(n)
Java代码实现
class Solution {
public int candy(int[] ratings) {
int len = ratings.length;
int[] candy = new int[len];
Arrays.fill(candy, 1);
for (int i = 1; i < len; i++) {
if (ratings[i] > ratings[i - 1]){
candy[i] = candy[i - 1] + 1;
}
}
for (int i = len - 2; i >= 0; i--) {
if (ratings[i] > ratings[i + 1]){
candy[i] = Math.max(candy[i + 1] + 1, candy[i]);
}
}
int result = 0;
for (int a : candy){
result += a;
}
return result;
}
}