题目大意:几个小孩站一排,每个小孩有个等级值,现在给小孩发糖,发的时候要遵守2个规则:(1)每个小孩至少一颗糖(2)两个相邻的小孩中,等级大的小孩一定比等级小的小孩糖多,求发糖的数目的最小值。对于等级序列[1, 2, 2]来说,发糖数目最小值为4,发糖序列为[1, 2, 1]。
分析:初始化所有小孩糖数目为1,从前往后扫描,如果第i个小孩等级比第i-1个高,那么i的糖数目等于i-1的糖数目+1;从后往前扫描,如果第i个的小孩的等级比i+1个小孩高,但是糖的数目却小或者相等,那么i的糖数目等于i+1的糖数目+1。该算法时间复杂度为O(N)。之所以两次扫描,即可达成要求,是因为:第一遍,保证了每一点比他左边candy更多(如果得分更高的话)。第二遍,保证每一点比他右边candy更多(如果得分更高的话),同时也会保证比他左边的candy更多,因为当前位置的candy只增不减。
class Solution {
public:
int candy(vector<int> &ratings)
{
// Note: The Solution object is instantiated only once and is reused by each test case.
int *candyNum = new int[ratings.size()];//每个小孩的糖数目
for(int i = 0; i < ratings.size(); i++)
candyNum[i] = 1;
for(int i = 1; i < ratings.size(); i++)
if(ratings[i] > ratings[i-1])
candyNum[i] = candyNum[i-1] + 1;
for(int i = ratings.size()-2; i>=0; i--)
if(ratings[i] > ratings[i+1] && candyNum[i] <= candyNum[i+1])
candyNum[i] = candyNum[i+1] + 1;
int result = 0;
for(int i = 0; i < ratings.size(); i++)
result += candyNum[i];
delete []candyNum;
return result;
}
};
9.17更新
这道题目的一个陷阱是:只是要求相邻的更高的小孩比低的小孩拿糖多,但是没说两个相等的小孩该如何拿糖。
事实上当序列为1, 2, 2时,拿糖为1, 2, 1。因为题目没说 不允许相等的小孩中a小孩拿的糖不能比b小孩少。
这样的题目,就用经典的 双向扫描就好了。代码如下:
class Solution {
public:
int candy(vector<int>& ratings) {
int n = (int)ratings.size();
int arr[n];
memset(arr, 0, sizeof(arr));
arr[0] = 1;
for (int i=1; i<n; i++) {
if (ratings[i]>ratings[i-1]) {
arr[i] = arr[i-1]+1;
} else {
arr[i] = 1;
}
}
int ret = arr[n-1];
for (int i=n-2; i>=0; i--) {
if (ratings[i]>ratings[i+1]&&arr[i]<=arr[i+1]) {
arr[i] = arr[i+1]+1;
}
ret += arr[i];
}
return ret;
}
};