牛客题霸 [分糖果问题] 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;
    }
};