zoukankan      html  css  js  c++  java
  • Leetcode 503. 下一个更大元素 II(单调栈/优先队列)

    给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。

    示例 1:

    输入: [1,2,1]
    输出: [2,-1,2]
    解释: 第一个 1 的下一个更大的数是 2;
    数字 2 找不到下一个更大的数;
    第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
    注意: 输入数组的长度不会超过 10000。

    官方题解给的是单调栈,但注意到10000的范围可以直接打暴力QAQ

    class Solution {
    public:
        int a[200005], ans[200005], n;
        struct node {
            int val, pos;
            bool operator < (const node& b) const 
    	    {
    		    return val > b.val;
    	    }
        };
        vector<int> nextGreaterElements(vector<int>& nums) {
            n = nums.size();
            for(int i = 0; i < n; i++) {
                a[i + 1] = a[i + n + 1] = nums[i];
                ans[i + 1] = 0x3f3f3f3f;
            }
            //int last = 0, pos = 0;
            priority_queue<node> q;
            for(int i = 1; i <= 2 * n; i++) {
                while(1) {
                    if(!q.size()) break;
                    node tmp = q.top();
                    if(tmp.val >= a[i]) break;
                    q.pop();
                    ans[tmp.pos] = a[i];
                }
                if(a[i] < a[i + 1]) ans[i] = a[i + 1];
                else {
                    node tmp;
                    tmp.val = a[i];
                    tmp.pos = i;
                    q.push(tmp);
                }
            }
            vector<int> out;
            for(int i = 1; i <= n; i++) {
                if(ans[i] != 0x3f3f3f3f) out.push_back(ans[i]);
                else out.push_back(-1);
            }
            return out;
        }
    };
    
    
  • 相关阅读:
    51nod1089(最长回文子串之manacher算法)
    51nod1088(最长回文子串)
    51nod1256(乘法逆元)
    51nod1085(01背包)
    51nod1079(中国剩余定理)
    数据的特征工程
    30种提高mysql处理速度的方法
    机器学习资料
    python3.6安装-windows
    python import sklearn出错 "ImportError: DLL load failed: 找不到指定的模块。
  • 原文地址:https://www.cnblogs.com/lipoicyclic/p/14489375.html
Copyright © 2011-2022 走看看