给你一个下标从 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
解题思路:使用两个数组分别记录每个位置左边最大的数字和右边最小的数字。
class Solution(object):
def sumOfBeauties(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
n=len(nums)
left_max=[0]*n
right_min=[0]*n
left_max[0]=nums[0]
right_min[n-1]=nums[n-1]
ans=0
i=1
while i<n:
left_max[i]=max(nums[i],left_max[i-1])
i+=1
i=n-2
while i>=0:
right_min[i]=min(nums[i],right_min[i+1])
i-=1
i=1
while i<n-1:
if nums[i]>left_max[i-1] and nums[i]<right_min[i+1]:
ans+=2
elif nums[i-1]<nums[i] and nums[i]<nums[i+1]:
ans+=1
else:
ans+=0
i+=1
return ans