方法:贪心算法

将每个小孩的糖果数都初始化为1,再遍历数组根据相邻两个孩子的得分情况来更新糖果数目,这样就能得到最少的糖果数

具体做法:

  • step 1:使用一个辅助数组记录每个位置的孩子分到的糖果,全部初始化为1.
  • step 2:从左到右遍历数组,如果右边元素比相邻左边元素大,意味着在递增,糖果数就是前一个加1,否则保持1不变。
  • step 3:从右到左遍历数组,如果左边元素比相邻右边元素大, 意味着在原数组中是递减部分,如果左边在上一轮中分到的糖果数更小,则更新为右边的糖果数+1,否则保持不变。
  • step 4:将辅助数组中的元素累加求和。

时间复杂度:o(n)

空间复杂度:o(n)

class Solution {
  public:
    int candy(vector<int>& arr) {
        // 记录每个小孩的糖果数,初始化为1
        vector<int> num(arr.size(), 1);
	  	// 记录最终的糖果数
        int res = 0;
        // 先从左向右遍历
        for (int i = 1; i < arr.size(); i++) {
            // 如果右边得分大于左边,则右边糖果数是左边糖果数加1
            if (arr[i] > arr[i - 1])
                num[i] = num[i - 1] + 1;
        }

        res = num[arr.size() - 1];
        // 再从右向左遍历,
        for (int i = arr.size() - 2; i >= 0; i--) {
            // 如果左边得分大于右边且左边糖果数不大于右边时,左边糖果数是右边边糖果数加1
            if (arr[i] > arr[i + 1] && num[i] <= num[i + 1])
                num[i] = num[i + 1] + 1;
            res += num[i];
        }

        return res;
    }
};