数组中元素与下一个比它大的元素之间的距离

/*用辅助数组存储差值;用栈存储数组元素,如果遇到后面的元素大于前一个元素,就弹出栈中的元素*/

也就是说,利用栈;先判断相邻的两个数组的大小;如果后一个元素大于前一个元素preIndex,就让辅助数组中Preindex的值为1;如果当前元素不大于前一个元素;就移动数组;判断当前元素与他的后一个元素大小;满足条件,就判断next和pre的大小,依次进行
peek()表示栈顶元素
pop()表示弹出栈顶元素

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int n=temperatures.length;
        int[] dist=new int[n];
        Stack<Integer> indexs=new Stack<>();
        for(int curIndex=0;curIndex<n;curIndex++){
            //当栈不为空并且数组当前元素大于以栈定元素为索引的元素,就弹出栈顶元素;否则,将当前元素索引进栈
            while(!indexs.isEmpty()&&temperatures[curIndex]>temperatures[indexs.peek()]){
                int preIndex=indexs.pop();
                dist[preIndex]=curIndex-preIndex;
            }
            indexs.push(curIndex);
        }
        return dist;
    }
}

 循环数组中比当前元素大的下一个元素

与 739. Daily Temperatures (Medium) 不同的是,数组是循环数组,并且最后要求的不是距离而是下一个元素。

循环数组循环两遍,主要是找到数组中比最后一个元素大的第一个元素。首先进行元素填充-1;找到比她大的就进行互换

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        /*循环数组的表示法比较难,存储一次栈中的元素;循环比较两次数组中的元素*/
        int n=nums.length;
        Stack<Integer> pre=new Stack<>();
        int[] next=new int[n];
        Arrays.fill(next,-1);
        for(int i=0;i<n*2;i++){
            int num=nums[i%n];
            while(!pre.isEmpty()&&num>nums[pre.peek()]){
                next[pre.pop()]=num;
            }
            if(i<n){
                pre.push(i);
            }
        }
        return next;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_31957639/article/details/83714005