LeetCode 503. 下一个更大元素 II 题解 C/C++

类比LeetCode #496. 下一个更大元素 I

#include <iostream>
#include <cstdio>
#include <vector>
#include <stack>
#include <algorithm>
#include <set>
#include <map>
//503. 下一个更大元素 II
//https://leetcode-cn.com/problems/next-greater-element-ii/
using namespace std;


/*
枚举每个元素的左边第一个大于他的值和右边第一个大于他的值，
如果右边存在的话就取右边的，如果右边不存在左边存在则取左边的，
若两边都不存在则为-1
*/
class Solution1 {
public:
	vector<int> nextGreaterElements(vector<int>&nums) {
		int len = nums.size();
		vector<int> vec(len,-1);
		for(int i = 0;i<len;i++) {
			for(int j = 0;j<i;j++) {
				if(nums[j]>nums[i]) {
					vec[i] = nums[j];
					break;
				}
			}
			for(int k = i+1;k<len;k++) {
				if(nums[k]>nums[i]) {
					vec[i] = nums[k];
					break;
				}
			}
		}
		return vec;
	}
};

//方法二：单调栈
/*
因为是循环数组，解决办法：将前n-1个元素在拼接到原序列后面

每次我们移动到数组中的一个新的位置 i，我们就将当前单调栈中
所有对应值小于 nums[i] 的下标弹出单调栈，这些值的下一个更大元
素即为nums[i],随后我们将位置 i 入栈。
*/
class Solution {
public:
	vector<int> nextGreaterElements(vector<int>&nums) {
		int n = nums.size();
		vector<int> vec(n,-1);
		stack<int> stk;//存储下标，对应的元素单调不升
		for(int i = 0;i<2*n-1;i++) {
			while(!stk.empty()&&nums[stk.top()]<nums[i%n]) {
				vec[stk.top()] = nums[i%n];
				stk.pop();
			}
			stk.push(i%n);
		}
		return vec;
	}
};

int main(){

	return 0;
}