当前位置:   article > 正文

力扣503

力扣503

题目

给定一个循环数组 nums ( nums[nums.length - 1] 的下一个元素是 nums[0] ),返回 nums 中每个元素的 下一个更大元素 。

数字 x 的 下一个更大的元素 是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1 。

示例 1:

输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数; 
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。

示例 2:

输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]

思路

1 可以用暴力方法对数组进行遍历,记录下最大值,找到就break,如果等于最大值就赋为-1,利用取余实现循环

2 用一个单向栈来保存下标,遍历每个数据,判断栈是否为空并且栈顶元素是否小于现在的值,如果是,那现在这个值就是栈里所以小于他的第一个 最大值,这里不需要再做额外的判断,因为从栈顶开始一直到为空或栈底值不小于当前值 。栈里一定是从底到顶递减,否则早出栈了。

循环遍历的话,用取余解决。

代码

  1. class Solution {
  2. public int[] nextGreaterElements(int[] nums) {
  3. int[] ans=new int[nums.length];
  4. int max=nums[0];
  5. for(int i=0;i<nums.length;i++){
  6. max=Math.max(max,nums[i]);
  7. }
  8. for(int i=0;i<nums.length;i++){
  9. int j=(i+1)%nums.length;
  10. while(true){
  11. if(nums[j]>nums[i]){
  12. ans[i]=nums[j];
  13. break;
  14. }
  15. if(nums[i]==max){
  16. ans[i]=-1;
  17. break;
  18. }
  19. j=(j+1)%nums.length;
  20. }
  21. }
  22. return ans;
  23. }
  24. }
  1. class Solution {
  2. public int[] nextGreaterElements(int[] nums) {
  3. int n=nums.length;
  4. int[] ans=new int[n];
  5. Arrays.fill(ans, -1);
  6. Deque<Integer> st=new ArrayDeque<>();
  7. for(int i=0;i<n*2;i++){
  8. while(!st.isEmpty()&&nums[i%n]>nums[st.peek()]){
  9. ans[st.pop()]=nums[i%n];
  10. }
  11. if(i<n){
  12. st.push(i);
  13. }
  14. }
  15. return ans;
  16. }
  17. }

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/775031
推荐阅读
相关标签
  

闽ICP备14008679号