力扣 496. 下一个更大元素 I

题目来源：https://leetcode.cn/problems/next-greater-element-i/description/

C++题解1：暴力解决

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size(), len2 = nums2.size();
        vector<int> result(len1, -1);
        // 使用两层循环，直接判断nums[i]在nums2中的位置，并且遍历完其右边元素，直至找到或者结束。
        for(int i = 0; i < len1; i++) {
            bool flg = false;
            for(int j = 0; j < len2; j++) {
                if(nums2[j] == nums1[i]) flg = true;
                if(flg && nums2[j] > nums1[i]) {
                    result[i] = nums2[j];
                    break;
                }
            }
        }
        return result;
    }
};

C++题解2：使用栈

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int len1 = nums1.size(), len2 = nums2.size();
        vector<int> result(len1, -1);
        // nums1的元素没有重叠，使用map可以方便查找，umap.count()可以计数，相当于判断有无
        unordered_map<int, int> umap;
        for(int i = 0; i < len1; i++) {
            umap[nums1[i]] = i;
        }
        // stack从底到顶，存放nums2元素，按金字塔状，下面大，上面小。
        // 如果遇到大的，说明栈顶的元素小于当前nums2[j]，为其右边元素
        // 判断栈顶元素是否在nums1中，使用umap可快速查找对应的索引，更新result
        // 栈顶元素弹出，直到栈顶元素大于当前nums2[j]，将其入栈
        stack<int> st;
        st.push(nums2[0]);
        for(int j = 1; j < len2; j++) {
            while(!st.empty() && nums2[j] > st.top()) {
                if(umap.count(st.top()) > 0) result[umap[st.top()]] = nums2[j];
                st.pop();
            }
            st.push(nums2[j]);
        }
        return result;
    }
};