leetcode 6227. 下一个更大元素 IV

6227. 下一个更大元素 IV

本题传送门
题意:
给你一个下标从 0 开始的非负整数数组 nums 。对于 nums 中每一个整数，你必须找到对应元素的 第二大 整数。
如果 nums[j] 满足以下条件，那么我们称它为 nums[i] 的 第二大 整数：

• j > i
• nums[j] > nums[i]
• 恰好存在 一个 k 满足 i < k < j 且 nums[k] > nums[i] 。
  如果不存在 nums[j] ，那么第二大整数为 -1 。
  比方说，数组 [1, 2, 4, 3] 中，1 的第二大整数是 4 ，2 的第二大整数是 3 ，3 和 4 的第二大整数是 -1 。
  请你返回一个整数数组 answer ，其中 answer[i]是 nums[i] 的第二大整数。
  样例:
  输入：nums = [2,4,0,9,6]
  输出：[9,6,6,-1,-1]
  关于set和某些函数:
• set里面会自动排序（从小到大），以后可以利用好这个特性;
• set里面用的是平衡二叉搜索树（也就是红黑树）维护;
• set自带的upper_bound是寻找比该元素大的最小值;
  

题解:
我们首先按照nums数组元素的值对下标数组从大到小排序,接着遍历下标数组。因为是按照从大到小排序的每个元素比它大的数在遍历到它的时候已经加入set集合了。对于重复元素特殊处理就行了。然后找答案并判读是否更新即可。最后别忘记更新set集合.
代码:

class Solution {
public:
    vector<int> secondGreaterElement(vector<int>& nums) {
        int n = nums.size();
        vector<int> res;
        vector<int> ans(n, -1);
        for(int i = 0; i < n; i++) res.push_back(i);
        sort(res.begin(), res.end(), [&](int x, int y){
            return nums[x]>nums[y];
        });
        set<int> S;
        S.insert(n+1), S.insert(n+2);//两个哨兵，使得每个元素都能找到比该元素大的下标值
        for(int i = 0; i < n; i++){
            int j = i+1;
            while(j<n&&nums[res[i]]==nums[res[j]]) j++;
            for(int k = i; k < j; k++){
                auto it = S.upper_bound(res[k]);
                ++ it;
                if(*it < n) ans[res[k]] = nums[*it];
            }
            for(int k = i; k < j; k++) S.insert(res[k]);
            i = j - 1;
        }
        return ans;
    }
};