leetcode-496. 下一个更大元素 I刷题笔记(c++)

    技术2022-07-11  88

    写在前面

    难度:简单双层循环 或 堆栈 + 数据映射 后者理解、尝试有1定难度

    题目详情

    给定两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。 nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。 示例 1: 输入: nums1 = [4,1,2], nums2 = [1,3,4,2]. 输出: [-1,3,-1] 解释: 对于num1中的数字4,你无法在第二个数组中找到下一个更大的数字,因此输出 -1。 对于num1中的数字1,第二个数组中数字1右边的下一个较大数字是 3。 对于num1中的数字2,第二个数组中没有下一个更大的数字,因此输出 -1。 示例 2: 输入: nums1 = [2,4], nums2 = [1,2,3,4]. 输出: [3,-1] 解释: 对于 num1 中的数字 2 ,第二个数组中的下一个较大数字是 3 。 对于 num1 中的数字 4 ,第二个数组中没有下一个更大的数字,因此输出 -1 。 输入: nums1 = [1,3,5,2,4], nums2 = [6,5,4,3,2,1,7]. 输出: [7,7,7,7,7] 提示: nums1和nums2中所有元素是唯一的。 nums1和nums2 的数组大小都不超过1000。

    ac代码

    2层循环 vector<int>元素查找 vector<int>::iterator it = find(nums2.begin(), nums2.end(), c); class Solution { public: vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) { vector<int> ans; for(auto c : nums1) { vector<int>::iterator it = find(nums2.begin(), nums2.end(), c); for(it=it+1; it != nums2.end(); ++it) { if(*it > c) { ans.push_back(*it); break; } } if(*(it-1) <= c && it == nums2.end()) ans.push_back(-1); } return ans; } }; 堆栈 + 数据映射 简化后的个人逻辑解题思想 遍历nums2数组,逐个元素入栈;当前元素大于栈顶元素,栈顶元素出栈;并将栈顶元素与当前元素封装为key-value键值对存入unordered_map<int, int> umap;针对堆栈元素循环至条件不成立,得到nums2中元素所对应的下1个更大元素的Map表;遍历nums1,umap中存在则输出,不存在输出-1。 class Solution { public: vector<int> nextGreaterElement(vector<int> &nums1, vector<int> &nums) { vector<int> ans; if(nums.size() <=0) return ans; stack<int> stk; unordered_map<int, int> umap; for (int i = 0; i < nums.size(); ++i) { while (stk.size() && stk.top() < nums[i]) { umap[stk.top()] = nums[i]; stk.pop(); } stk.push(nums[i]); } for (int num: nums1) { if(umap.count(num)) ans.push_back(umap[num]); else ans.push_back(-1); } return ans; } }; 参考文章 Leetcode刷题52-496. 下一个更大元素 I(C++详细解法!!!)
    Processed: 0.010, SQL: 12