题目: 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;
}