首先这是一个循环数组,某个数的右边开始循环整个数组找到第一个大于当前这个数的元素,所以我的想法是从原数组[1, 2, 1]中拷贝一份
let nums2 = nums.slice(0, nums.length);
nums2先去循环nums再来和nums2中每一个元素进行比较,当和nums2中元素比较完一次之后,将nums2中下标为0的元素放到末尾,来做到可以进行循环整个数组
nums2.push(nums2.shift());
下面就是我的整个代码
var nextGreaterElements = function(nums) {
let res = [];
let nums2 = nums.slice(0, nums.length);
for (let i = 0; i < nums.length; i++) {
let max = nums[i];
let flag = false;
for (let j = 0; j < nums2.length; j++) {
if(max < nums2[j]){
flag = true;
res.push(nums2[j]);
break;
}
}
if(!flag){
res.push(-1);
}
nums2.push(nums2.shift());
}
return res;
};
但是这样的解法效率还是很低
后面看官方的java版本,通过栈来对比,只需要进行一次循环
时间复杂度:O(N)。
空间复杂度:O(N)。
我这里写出了javascript的版本
var nextGreaterElements = function(nums) {
let len = nums.length;
let res = new Array(len).fill(-1);
let stack = [];
for (let i = len * 2 - 1; i >= 0; i--) {
while(stack.length > 0 && stack[stack.length - 1] <= nums[i % len]){
stack.pop();
}
res[i % len] = stack.length == 0 ? -1 : stack[stack.length - 1];
stack.push(nums[i % len]);
}
return res;
};
在循环的时候i的下标放大了一倍,来做到循环数组的一次遍历,将栈中的数从大到小的排列,并将当前需要比较的数和栈顶元素比较,如果比栈顶元素大的话就弹栈,当栈内没有元素或者栈顶元素大于当前元素时,结束循环,这个时候就找到第一个比当前元素大的数,此时当前这个元素第一大的元素就是栈顶的元素,再将当前元素压入栈中,进行下一次循环