LeetCode 2012. 数组美丽值求和--前缀和

2012. 数组美丽值求和

给你一个下标从 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

提示：

3 <= nums.length <= 105
1 <= nums[i] <= 105

题解

前缀和标记下就行了。

AC代码

class Solution {
public:
    int mx[100010],mi[100010];
    int sumOfBeauties(vector<int>& nums) 
    {
        int mi_data=0;
        for(int i=0;i<nums.size();i++)
        {
            mi_data = max(mi_data,nums[i]);
            mi[i]=mi_data;
        }
        int mx_data = 1e9;
        for(int i=nums.size()-1;i>=0;i--)
        {
            mx_data = min(mx_data,nums[i]);
            mx[i] = mx_data;
        }
        int ans=0;
        for(int i=1;i<nums.size()-1;i++)
        {
            if(nums[i]>mi[i-1]&&nums[i]<mx[i+1])
            ans+=2;
            else if(nums[i-1]<nums[i]&&nums[i]<nums[i+1])
            ans+=1;
        }
        return ans;
    }
};