Leetcode496. 下一个更大元素 I

Leetcode496. 下一个更大元素 I

题目：
   给定两个没有重复元素的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。
    nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出-1。

示例 1:

输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
    对于num1中的数字4，你无法在第二个数组中找到下一个更大的数字，因此输出 -1。
    对于num1中的数字1，第二个数组中数字1右边的下一个较大数字是 3。
    对于num1中的数字2，第二个数组中没有下一个更大的数字，因此输出 -1。

题解：
单调栈：
1.维护一个栈，遍历nums2中的每一个元素，入栈，如果后面的元素有大于前面元素的，就出栈一个元素，将两个元素存入一个map中，最后map中存入的都是右边比左边元素大的键值对；
2.然后在遍历nums1,去map中按照key,找对应的value,找到的存入数组，找不到的给-1存入数组。
scala代码：

 /**
    * 单调栈
    *
    * @param nums1
    * @param nums2
    * @return
    */
  def nextGreaterElement(nums1: Array[Int], nums2: Array[Int]): Array[Int] = {
    val stack = new mutable.Stack[Int]()
    val map = new mutable.HashMap[Int, Int]()
    val arr = new Array[Int](nums1.length)

    for (i <- 0 until nums2.length) {

      while (!stack.isEmpty && nums2(i) > stack.top) {
        map.put(stack.pop(), nums2(i))
      }
      stack.push(nums2(i))
    }
    for (i <- 0 until nums1.length) {
      val value = map.getOrElse(nums1(i), -1)
      arr(i) = value
    }
    arr
  }