右边第二大的数

题意

找到每个数右边第二个比它大的数

constrains:
$1\le n\le 10^5$

思路

我用两个单调栈做的，O(n)。
考虑把问题泛化到找第k个，用pair(-a[i],i)生成新数组b，就是说我们第一关键字是a[i]并从大到小排序，第二关键字是下标i升序，声明一个空的set维护，往里面插下标。遍历b，把当前遍历到的下标j插入set，我们能保证set里的下标表示的数都大于等于当前的b[j].a值，找到当前set中后两位的下标就是答案。
如果是找k，那就是set后第k位的下标，这样复杂度会是$O(nklogn)$。当k很大时考虑优化一下，python的SortedList能秒杀，C++要离散化+树状数组上二分，我们用一颗树状数组，值域是离散化后的a[i]，每次插入下标改为在a[i]节点处加一，那么找树上右边第k个出现的1即可，这个是可以树上二分做的，O(nlognlogk)

代码

class Solution {
public:
    vector<int> secondGreaterElement(vector<int>& nums) {
        int n = nums.size();
        
        // 将所有下标按 nums[i] 降序为第一关键字，i 升序为第二关键字排序
        typedef pair<int, int> pii;
        vector<pii> vec;
        for (int i = 0; i < n; i++) vec.push_back(pii(-nums[i], i));
        sort(vec.begin(), vec.end());

        vector<int> ans(n);
        set<int> st;
        // 按排序后的顺序枚举下标
        for (int i = 0; i < n; i++) {
            // 找到有序列表里比 i 大的第二个数
            auto it = st.upper_bound(vec[i].second);
            if (it != st.end() && next(it) != st.end()) ans[vec[i].second] = nums[*next(it)];
            else ans[vec[i].second] = -1;
            // 将当前下标加入有序列表
            st.insert(vec[i].second);
        }
        return ans;
    }
};