3月11日刷题:LeetCode 2012 数组美丽值求和

给你一个下标从 0 开始的整数数组 nums 。对于每个下标 i1 <= i <= nums.length - 2),nums[i] 的 美丽值 等于:

  • 2,对于所有 0 <= j < i 且 i < k <= nums.length - 1 ,满足 nums[j] < nums[i] < nums[k]
  • 1,如果满足 nums[i - 1] < nums[i] < nums[i + 1] ,且不满足前面的条件
  • 0,如果上述条件全部不满足

返回符合 1 <= i <= nums.length - 2 的所有 **nums[i] **的 美丽值的总和 。

 

示例 1:

输入: nums = [1,2,3]
输出: 2
解释: 对于每个符合范围 1 <= i <= 1 的下标 i :
- nums[1] 的美丽值等于 2

示例 2:

输入: nums = [2,4,6,4]
输出: 1
解释: 对于每个符合范围 1 <= i <= 2 的下标 i :
- nums[1] 的美丽值等于 1
- nums[2] 的美丽值等于 0

示例 3:

输入: nums = [3,2,1]
输出: 0
解释: 对于每个符合范围 1 <= i <= 1 的下标 i :
- nums[1] 的美丽值等于 0
class Solution {
public:
    int sumOfBeauties(vector<int>& nums) {
        int n = nums.size();
        if (n < 3) return 0;
        
        vector<int> left_max(n, 0);
        vector<int> right_min(n, 0);
        
        left_max[1] = nums[0];
        for (int i = 2; i < n; ++i) {
            left_max[i] = max(left_max[i-1], nums[i-1]);
        }
        
        right_min[n-2] = nums[n-1];
        for (int i = n-3; i >= 1; --i) {
            right_min[i] = min(right_min[i+1], nums[i+1]);
        }
        
        int sum = 0;
        for (int i = 1; i <= n-2; ++i) {
            if (nums[i] > left_max[i] && nums[i] < right_min[i]) {
                sum += 2;
            } else if (nums[i] > nums[i-1] && nums[i] < nums[i+1]) {
                sum += 1;
            }
        }
        return sum;
    }
};

用时:11ms

内存:91.02MB