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