下一个更大元素

题目描述:给定一个数组,求数组中每个元素在它后面第一个比它大的元素是多少。                   如[2,1,2,4,3],返回数组 [4,2,4,-1,-1]。解释:第一个 2 后面比 2 大的数是 4; 1 后面比 1 大的数是 2;第二个 2 后面比 2 大的数是 4; 4 后面没有比 4 大的数,填 -1;3 后面没有比 3 大的数,填 -1。

知识:单调栈。

vector<int> nextGreaterElement(vector<int>& nums) {
    vector<int> ans; // 存放答案
    stack<int> s;
    for (int i = nums.size() - 1; i >= 0; i--) { // 倒着往栈里放
        while (!s.empty() && s.top() <= nums[i]) { // 判定数的大小
            s.pop(); // 小的出栈
        }
        ans[i] = s.empty() ? -1 : s.top(); // 这个元素身后的第一个比它大的数
        s.push(nums[i]); // 进队,再和前面的数比大小
    }
    return ans;
}

题目2:

 解题思路:第一步:运用单调栈的知识,将nums2数组中各个元素的下一个比它更大的元素是多少求出,并保存到map<int,int >cnt中;第二步:遍历nums1中的元素,找出cnt中对应的数;

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
          int m=nums1.size();
          int n=nums2.size();
          vector<int>ans;
          map<int,int>cnt;
          stack<int>a;
          for(int i=n-1;i>=0;i--)
          {
              while(!a.empty()&&a.top()<=nums2[i])
                   a.pop();
              cnt[nums2[i]]=a.empty()?-1:a.top();//元素当作cnt的下标
              a.push(nums2[i]);
          }
          for(int j=0;j<m;j++)
             {
                 ans.push_back(cnt[nums1[j]]);
             }
             return ans;
    }
};

小结:学会了单调栈的原理知识,

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值