LeetCode 496 下一个更大元素 I

给定两个无重复元素的数组nums1和nums2,nums1是nums2的子集,目标是找到nums1中每个元素在nums2中对应的下一个更大元素。如果不存在,则标记为-1。解决方案涉及到使用单调栈和哈希映射来加速查找过程。
摘要由CSDN通过智能技术生成

题目: 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 。

思路:

该题目和每日温度有递进关系
首先最终存放结果的数组大小应该是和nums1的大小是一样的
单调栈应该是在nums2上进行了
先对nums1做hash映射,能够通过数组的值找到对应的下标umap[nums1[i]] = i;
题目思路:
在nums2中进行递增单调栈操作,如果当前元素小于等于栈顶元素,直接入栈
如果当前元素大于栈顶元素,那么就要执行相对于的操作了,
如果大于后,首先应该确定栈顶元素是否在nums1中出现了,如果出现了,那么找到栈顶元素对应的下标
在该下标的位置插入当前的nums2中正在访问的元素,如果没出现,那就不用管;后边继续出现判断,
最后小于等于了后入栈,由于result数组初始化为-1,因此栈中元素如果没有比其大的,则默认为-1

class Solution {
public:
	vector<int> incrstack(vector<int>& nums1, vector<int>& nums2) {
		stack<int>st;
		vector<int> result(nums1.size(), -1);
		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()]) {
					if (umap.count(nums2[st.top()]) > 0) {
						int index = umap[nums2[st.top()]];
						result[index] = nums2[i];
					}
					st.pop();
				}
				st.push(i);
			}
		}
		return result;
	}
};

int main() {
	vector<int> nums1 = { 4, 1, 2 };
	vector<int> nums2 = { 1, 3, 4, 2 };
	Solution ss;
	vector<int>result = ss.incrstack(nums1, nums2);
	for (vector<int>::iterator pos = result.begin(); pos != result.end(); pos++) {
		cout << *pos << " ";
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值