前言
暴力可以解,但是优化非常需要技巧
正文
问题来源
本问题来自leetcode上的135题。
问题描述
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
- 每个孩子至少分配到 1 个糖果。
- 相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?示例 1:
输入: [1,0,2] 输出: 5 解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
输入: [1,2,2] 输出: 4 解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
分析:
假设每个孩子分到的糖果数组为A[N],初始化为{1},因为每个人至少分到一颗糖。
1、与前面的邻居比较,前向遍历权重数组ratings,如果ratings[i]>ratings[i-1],则A[i]=A[i-1]+1;
2、与后面的邻居比较,后向遍历权重数组ratings,如果ratings[i]>ratings[i+1]且A[i]<A[i+1]+1,则更新A,A[i]=A[i+1]+1;
3、对A求和即为最少需要的糖果。
时间复杂度:O(n)
空间复杂度:O(n)
C代码如下:int candy(int* ratings, int ratingsSize) { int *dp = NULL; int i; int sum=0; dp = (int*)malloc(sizeof(int)*ratingsSize); for(i=0;i<ratingsSize;i++)//初始化 { dp[i] = 1; } for(i=1;i<ratingsSize;i++)//正向 { if(ratings[i-1]<ratings[i]) { dp[i] = dp[i-1] + 1; } } for(i=ratingsSize-2;i>=0;i--)//反向 { if(ratings[i+1]<ratings[i]) { dp[i] = max(dp[i],dp[i+1] + 1); } } for(i=0;i<ratingsSize;i++)//求和 { sum += dp[i]; } free(dp); return sum; }
总结:
凭感觉做的。网上把这题归为贪心。
结语
闲来无事的话,多练习练习,做做题。