代码随想录Day62

今天继续学习单调栈解决相关问题。

496.下一个更大的元素|

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。

给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。

对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。

示例 1：

输入：nums1 = [4,1,2], nums2 = [1,3,4,2].
输出：[-1,3,-1]
解释：nums1 中每个值的下一个更大元素如下所述：
- 4 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
- 1 ，用加粗斜体标识，nums2 = [1,3,4,2]。下一个更大元素是 3 。
- 2 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
示例 2：

输入：nums1 = [2,4], nums2 = [1,2,3,4].
输出：[3,-1]
解释：nums1 中每个值的下一个更大元素如下所述：
- 2 ，用加粗斜体标识，nums2 = [1,2,3,4]。下一个更大元素是 3 。
- 4 ，用加粗斜体标识，nums2 = [1,2,3,4]。不存在下一个更大元素，所以答案是 -1 。

思路：

1.本题在Day61初次接触的那道每日温度的基础之上稍加变化了一下，原本是只有一个数组，我们只需要按部就班找到每一个元素右边第一个比他大的就行，本题则是采用了两个数组并且其中一个数组是另一个数组的子集，让我们找子集数组里面所有元素在另一个数组里面右边的第一个比他大的元素。

2.因为分成了两个无重复元素的数组，因此我们需要建立起两个数组之间的联系。既然无重复元素，可以想到利用哈希表来建立映射，以元素值为键，以下标为值，正好也对应上我们用单调栈存储的是遍历过的元素下标。

3.单调栈中依旧采取从栈头到栈底递增的顺序，这样才能保证找到的是右边第一个更大的元素。

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> result(nums1.size(), -1);
        stack<int> st;//单调栈存储遍历过的元素下标
        unordered_map<int,int> umap;

        //哈希表建立映射，键为元素值，值为下标
        for(int i = 0; i < nums1.size(); i++){
            umap[nums1[i]] = i;
        }
        st.push(0);

        for(int i = 1; i < nums2.size(); i++){
            if(nums2[i] <= nums2[st.top()]){
                st.push(i);
            }
            else{
                while(!st.empty() && nums2[i] > nums2[st.top()]){
                    //先判断栈顶元素是否存在于nums1中，如果存在才进行结果存储
                    if(umap.count(nums2[st.top()])){
                        int index = umap[nums2[st.top()]];
                        result[index] = nums2[i];
                    }
                    //判断完栈顶元素后要将栈顶元素出栈
                    st.pop();
                }
                st.push(i);
            }
        }

        return result;
    }
};

启发：

1.本题将单个数组拆成了两个数组，因此我们需要建立起两个数组之间的联系进一步来转化为类似单个数组的解题方法。

503.下一个更大元素||

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

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

示例 1:

输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数； 
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。
示例 2:

输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]

思路：

1.本题同样是在Day61的每日温度基础之上进行了另一种变化，只有一个数组但是数组成环了。既然成环了那么就会很容易想到进行取模的操作来达成环的效果。

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> result(nums.size(), -1);
        stack<int> st;

        st.push(0);
        for(int i = 1; i < nums.size() * 2; i++){
            if(nums[i % nums.size()] <= nums[st.top()]){
                st.push(i % nums.size());
            }
            else{
                while(!st.empty() && nums[i % nums.size()] > nums[st.top()]){
                    result[st.top()] = nums[i % nums.size()];
                    st.pop();
                }
                st.push(i % nums.size());
            }
        }

        return result;
    }
};

启发：

1.本题其实细节还是不少的，如在for循环中i的判断条件，尽管成环了理论上来说可以无限次循环整个数组，但实际上只要循环数组的大小*2就足够我们找到所有元素的情况了，后续的循环没有意义。

2.本题中result中的元素是否会被后续成环后的第二次循环覆盖掉呢？其实并不会，实际代入几个结果模拟后就会发现尽管第二轮循环中会将result的值进行“改变”，但实际上对应元素赋的都是同一个值，就好像第一次循环我们找到了每个元素右边第一大的元素，第二轮循环我们只是从头开始又循环了一次，同样会找到每个元素右边第一大的元素（并且这个元素并不会改变）。