给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1] 输出: [2,-1,2] 解释: 第一个 1 的下一个更大的数是 2; 数字 2 找不到下一个更大的数; 第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
基本思路:
本题的解题思路基本上和 https://mp.csdn.net/postedit/82689212 这个一模一样,在这个题中仅有的不同是,这个数组可以循环查找,那这个好办 ,首先遍历一遍找到最大的数,然后从最大的这个数开始遍历即可。若遍历到数组尾部,那么就将下标置为0即可。
class Solution {
public:
struct T_struct
{
int Index;
int Value;
T_struct():Index(0),Value(0){}
};
vector<int> nextGreaterElements(vector<int>& nums) {
int Len = nums.size(), Position = 1, Max = INT32_MIN,Now_P,Time=0;
vector<int> Res(Len,-1);
if (Len == 0|| Len==1) return Res;
for (int i = 0; i < Len;i++) {
if (nums[i] > Max) {
Position = i;
Max = nums[i];
}
}
stack<T_struct> T_stack;
T_struct T; T.Index = 0; T.Value = nums[0];
T_stack.push(T); Now_P = Position==Len-1?0: Position+1;
while (Time<=Len)
{
while (T_stack.size() != 0 && nums[Now_P] > T_stack.top().Value) {
Res[T_stack.top().Index] = nums[Now_P];
T_stack.pop();
}
T_struct TT; TT.Index = Now_P; TT.Value = nums[Now_P];
T_stack.push(TT);
Now_P = (Now_P == Len - 1 ? 0 : Now_P+1);
Time++;
}
return Res;
}
};