力扣496.求解下一个最大元素问题(c++)

大致题意为要求找到nums1中x在nums2中对应位置右侧并且比x大的元素,有该元素时要求输出该元素,没有符合条件的元素或者越界时输出-1,其中nums1是nums2的子集,最后的结果要返回一个和nums1长度相同的数组

我们要明确题目要求我们干什么,无非就是两件事

1.找到元素的位置

2.输出该元素右侧且比下大的元素

所以我们需要遍历nums2找到nums1中的对应元素,通过两个while循环找到对应元素右侧是否有比该元素大的更大元素,同时我们也要考虑到数组的长度,越界时的情况也要考虑到,在编写代码时还要创建一个数组用来返回结果

代码如下

 

​​​​​​​​​​​​class Solution {
	vector nextGreaterElement(vector& nums1, vector& nums2) {
		int m = nums1.size();
		int n = nums2.size();//定义数组长度防止越界
		vectorres(m);//创建输出结果的数组
		for (int i = 0; i < m; ++i) {//遍历nums1
			int j = 0;
			while (j < n && nums2[j] != nums1[i]) {
				++j;//在nums2中找nums1的对应元素

			}
			int k = j + 1;
			while (k < n && nums2[k] < nums2[j]) {
				++k;//向右找更大的元素
			}
			res[i] = k < n ? nums2[k] : -1;//判定越界时输出-1

		}
		return res;
	}
};

 本人编程小白一枚,如有错误,欢迎各位大佬指教~

你可能感兴趣的:(leetcode,c++,数据结构)