下一个更大元素

最新推荐文章于 2024-04-26 21:55:28 发布

YangVijay

最新推荐文章于 2024-04-26 21:55:28 发布

阅读量92

点赞数

分类专栏：算法文章标签：算法栈单调栈

本文链接：https://blog.csdn.net/ywjatjd/article/details/115189373

版权

算法专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题目

给你两个没有重复元素的数组 $n u m s 1$ 和 $n u m s 2$ ，其中 $n u m s 1$ 是 $n u m s 2$ 的子集。请你找出 $n u m s 1$ 中每个元素在 $n u m s 2$ 中的下一个比其大的值。 $n u m s 1$ 中数字 $x$ 的下一个更大元素是指 $x$ 在 $n u m s 2$ 中对应位置的右边的第一个比 $x$ 大的元素。如果不存在，对应位置输出 -1 。

解法一：两层循环，时间复杂度 $O (m n)$

遍历 $n u m s 1$ ;
第一个while，找到在元素 $x$ 在 $n u m s 2$ 中位置;
第二个while，找到离 $x$ 最近的大于 $x$ 的元素;
没找到，设为-1；

参考代码C++：

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) 
{
    int m = nums1.size();
    int n = nums2.size();

    vector<int> ret(m, -1);
    for(int i = 0;i < m; ++i){
        int j = 0;
        while(nums1[i] != nums2[j]){
            ++j;
        }

        while(j < n && nums1[i] >= nums2[j]){
            ++j;
        }

        if(j < n){
            ret[i] = nums2[j];
        }
    }
    return ret;
}

解法二：栈，时间复杂度 $O (m + n)$

遍历 $n u m s 2$ ，
使用单调栈辅助¹，找到每个元素的下一个最大值
使用map存储（ $O (1)$ 时间内查询），「key」: $n u m s 2 [i]$ ，「value」: 下一个最大值
遍历 $n u m s 1$ ，从map中查询其对应的下一个最大值。

参考代码C++

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2)
{
	int m = nums1.size();
	 int n = nums2.size();
	
	 stack<int> monoStk;
	 map<int,int> dict;
	
	 for(int i = 0; i < n; ++i){
	     while(!monoStk.empty() && monoStk.top() < nums2[i]){
	         int tmp = monoStk.top();
	         dict[tmp] = nums2[i];  // 找到tmp的下一个大于它的值为nums2[i]
	         monoStk.pop();
	     }
	     monoStk.push(nums2[i]);
	 }
	
	 vector<int> ret(m, -1);
	
	 for(int i = 0; i < m; ++i){
	     if(dict.count(nums1[i])){
	         ret[i] = dict[nums1[i]];
	     }
	 }
	 return ret;
}