牛客题霸 [分糖果问题] C++题解/答案

牛客题霸 [分糖果问题] C++题解/答案

题目描述

一群孩子做游戏，现在请你根据游戏得分来发糖果，要求如下：

1. 每个孩子不管得分多少，起码分到一个糖果。
2. 任意两个相邻的孩子之间，得分较多的孩子必须拿多一些糖果。(若相同则无此限制)
   给定一个数组arr代表得分数组，请返回最少需要多少糖果。
   [要求]
   时间复杂度为On, 空间复杂度为O1

题解：

正着来一遍，倒着来一遍
根据大小关系递归赋值

代码：

class Solution {
public:
    /**
     * pick candy
     * @param arr int整型vector the array
     * @return int整型
     */
    int candy(vector<int>& arr) {
        // write code here
        int len=arr.size();
        int sum=0;
        vector<int>a(len,1);
        for(int i=1;i<len;i++)
        {
            if(arr[i]>arr[i-1])a[i]=a[i-1]+1;
        }
        for(int i=len-2;i>=0;i--)
        {
            if(arr[i]>arr[i+1])a[i]=max(a[i],a[i+1]+1);
        }
        for(int i=0;i<len;i++)sum+=a[i];
        return sum;
    }
};