专栏:50多种数据结构彻底征服

专栏:50多种经典图论算法全部掌握

以前我一直以为程序员35岁之后就不太好找工作了,昨天我在浏览招聘软件的时候无意间发现,有不少企业在招聘要求上直接写上不能超过30岁,看到这种奇葩要求有时候真的想骂人。。。

有一家上市公司竟然也这样写,不过他要求的是初级,这个勉强能接受。还有招聘高级的也要求30岁以下,真的是无语了,谁给它的勇气。

--------------下面是今天的算法题--------------

来看下今天的算法题,这题是LeetCode的第503题:下一个更大元素 II。

问题描述

来源:LeetCode第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 <= nums.length <= 10^4

  • -10^9 <= nums[i] <= 10^9

问题分析

这题和前几天讲的 类似,只不过这题是个循环数组,我们对数组 遍历两边 即可,超出数组长度之后,要对数组长度取余。

JAVA:

public int[] nextGreaterElements(int[] nums) {
    int length = nums.length;
    int ans[] = new int[length];
    Arrays.fill(ans, -1);// 默认都为-1
    Stack
       
  stk =  new Stack<>();      // 相当于把数组循环两遍      for ( int i =  0; i < length *  2; i++) {          int index = i % length;          // 单调栈,存储的是元素的下标,不是元素的值,从栈顶          // 到栈底所对应的值是递增的。          while (!stk.isEmpty() && nums[stk.peek()] < nums[index])             ans[stk.pop()] = nums[index];          // 当前元素的下标入栈         stk.push(index);     }      return ans; }

C++:

public:
    vector

  nextGreaterElements(vector

  &nums) {         int length = nums.size();         vector

  ans(length, -1); // 默认都为-1         stack

  stk;         // 相当于把数组循环两遍         for (int i = 0; i < length * 2; i++) {             int index = i % length;             // 单调栈,存储的是元素的下标,不是元素的值,从栈顶             // 到栈底所对应的值是递增的。             while (!stk.empty() && nums[stk.top()] < nums[index]) {                 ans[stk.top()] = nums[index];                 stk.pop();             }             // 当前元素的下标入栈             stk.push(index);         }         return ans;     }



Python:

def nextGreaterElements(self, nums: List[int]) -> List[int]:
    length = len(nums)
    ans = [-1] * length  # 默认都为-1
    stk = []
    # 相当于把数组循环两遍
    for i in range(length * 2):
        index = i % length
        while stk and nums[stk[-1]] < nums[index]:
            ans[stk.pop()] = nums[index]
        stk.append(index)
    return ans

笔者简介

博哥,真名:王一博,毕业十多年, 作者,专注于 数据结构和算法 的讲解,在全球30多个算法网站中累计做题2000多道,在公众号中写算法题解800多题,对算法题有自己独特的解题思路和解题技巧,喜欢的可以给个关注,也可以 下载我整理的1000多页的PDF算法文档 。