【leetcode】分糖果

分糖果

这道题是在写bjtu oj上的期末考试一题时想到的，一个很简单的贪心卡了我好久…

题目描述

分析思路

设a[i]为每个孩子的评分, b[i]为每个孩子需要的糖果数目

分析题目要求可以发现，结果序列需要满足以下三个条件：

1. 每个人至少一个糖果
2. a[i] < a[i + 1],则有b[i + 1] = b[i] + 1
3. a[i] > a[i + 1],则有b[i] = b[i + 1] +1
   

不难发现b[i]的最终数目大小为在满足初始值为1的条件下，2，3两个限制下的最大值。所以对于数组a，可以对他进行两次遍历操作:

首先将b数组赋初值为1
第一次从左到右遍历，对于所有a[i] < a[i + 1]的数字b[i + 1] = b[i] + 1
第二次从右向左遍历，对于所有a[i] < a[i - 1]且b[i - 1] <= b[i]的数字 b[i - 1] = b[i] + 1

注意要特殊处理a[i - 1] == a[i]的情况，以保证总糖果数量最小。

代码实现

int cnt[100010];
int candy(int* ratings, int ratingsSize) 
{
    int n = ratingsSize - 1;
    for(int i = 0; i <= ratingsSize; i ++) cnt[i] = 1;
    for(int i = 1; i <= n; i ++)
    { 
        if(ratings[i] > ratings[i - 1]) 
            cnt[i] = cnt[i - 1] + 1;
    }
    for(int i = n; i > 0; i --)
    {
        if(ratings[i - 1] > ratings[i] && cnt[i - 1] <= cnt[i])
            cnt[i - 1] = cnt[i] + 1;
    }
    long long ans = 0;
    for(int i = 0; i <= n; i ++) ans += cnt[i];
    return ans;

}