496.下一个更大元素Ⅰ——记录(C++)

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        for(int i=0;i<nums1.size();i++)
        {
              int t=0;
            for(int j=0;j<nums2.size();j++)
            {
                if(nums2[j]==nums1[i])
                {
                    for(int x=j;x<nums2.size();x++)
                    {
                        if(nums2[x]>nums1[i])
                {
                    nums1[i]=nums2[x];
                    j=nums2.size();
                    x=nums2.size();
                    t=1;
                }
                    }          
                }   
            }
            if (t==0)
            {
                nums1[i]=-1;
            }
        }
        return nums1;
    }
};

又是一次日常暴力解。

看一看答案;

单调栈+哈希表。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> hashmap;
        stack<int> st;
        for (int i = nums2.size() - 1; i >= 0; --i) {
            int num = nums2[i];
            while (!st.empty() && num >= st.top()) {
                st.pop();
            }
            hashmap[num] = st.empty() ? -1 : st.top();
            st.push(num);
        }
        vector<int> res(nums1.size());
        for (int i = 0; i < nums1.size(); ++i) {
            res[i] = hashmap[nums1[i]];
        }
        return res;
    }
};

我开始的思路与此类似,但是觉得自己对栈和哈希表不熟悉,所以没有实施。

所以还是要加强数据结构、算法及语法基础的学习!

加油!

上一篇:LeetCode 496 - 下一个更大元素 I


下一篇:496,字符串中的第一个唯一字符