给你一个下标从 0 开始的整数数组 nums
。对于每个下标 i
(1 <= 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