力扣刷题记录132.1-----503. 下一个更大元素 II

最新推荐文章于 2024-08-10 22:52:22 发布

@白圭

最新推荐文章于 2024-08-10 22:52:22 发布

阅读量75

点赞数

分类专栏： # 力扣习题记录册__C++__代码随想录文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/taiyuezyh/article/details/128007228

版权

力扣习题记录册__C++__代码随想录专栏收录该内容

174 篇文章 0 订阅

订阅专栏

一、题目

在这里插入图片描述

二、代码

class Solution 
{
public:
    vector<int> nextGreaterElements(vector<int>& nums) 
    {
       int i,j;
       vector<int> result(2*nums.size(),-1);   //单调栈
       vector<int> return_vector(2*nums.size(),-1);  //最终结果
       vector<int> nums2;
       stack<int> st;
       for(i=0;i<2;i++)  //变成两倍长度
       {
           for(j=0;j<nums.size();j++)
           {
               nums2.push_back(nums[j]);
           }
       }
    //    for(i=0;i<nums2.size();i++) std::cout<<"  "<<nums2[i];
    //    std::cout<<std::endl;
      

      st.push(0);
      for(i=0;i<nums2.size();i++)   //完成单调栈
      {
            if(nums2[i]<nums2[st.top()])
            {
               st.push(i);
            }
            else if(nums2[i]==nums2[st.top()])
            {
               st.push(i);
            }
            else
            {
               while(!st.empty()&&nums2[i]>nums2[st.top()])
               {
                   result[st.top()]=i-st.top();   //为了保持单调栈的完整  这里没有删除
                   return_vector[st.top()]=nums2[i];  //这是这题最后的结果
                   st.pop();
               }
               st.push(i);
            }
      }
       

    //   for(i=0;i<result.size();i++) std::cout<<"  "<<result[i];
    //   std::cout<<std::endl;
       
      result.resize(nums.size());   //裁剪回一倍的长度
      return_vector.resize(nums.size());   //裁剪回一倍的长度

      return return_vector;

    }
};