496. 下一个更大元素 I（哈希表方法实现）

 

 

import java.util.*;
class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
     /**
     实现思路:
     首先定义一个哈希表  key存储nums2数组中的每个元素  value 就是nums2中每个元素的下一个最大元素(这个下一个最大元素不是紧邻着的元素 也可以是不紧邻着的)
     要实现上面的逻辑，必须先定义一个栈：
     遍历nums2中元素 将元素放到栈中 如果下一个即将要加入到栈中元素大于栈顶的元素 就将这个元素弹栈 
     然后将栈中其他元素作为key 将这个num元素作为value 存入到哈希表中
     最后需要再次遍历栈  主要是将栈中未找到最大元素的取出来 放到哈希表中 以这些元素作为key -1作为value
     最后遍历nums1 从nums中找到nums1中的最大元素放到一个新的数组中 返回
      */
      // 1，定义返回存储返回最大元素的数组 长度以nums1长度一样
      int [] res  = new int [nums1.length];
      //2,定义一个hashMap 作为哈希表使用
      HashMap<Integer,Integer> map1 = new HashMap();
      // 3 定义一个栈
      Stack<Integer> stack = new Stack();
      // 4,遍历nums2数组
      for (int num : nums2) {
          // 5, 如果栈不为空并且num大于栈顶元素 就将栈中元素弹栈 放到哈希表中 以其值作为key num作为value
          while(!stack.isEmpty() && num > stack.peek()) {
              int temp =stack.pop();
              map1.put(temp,num);
          }
          // 6,其余元素入栈
          stack.push(num);

      }
      // 7,将栈中其余元素弹出来 放到哈希表中 作为哈希表的key value都是赋值为-1 因为没有找到最大值元素
      while(!stack.isEmpty()) {
          map1.put(stack.pop(),-1);
      }
      // 8,让nums1去哈希表中找他的最大元素
      for (int i=0;i<nums1.length;i++) {
          res[i] = map1.get(nums1[i]);
      }
      // 9 最后返回结果
      return res;

    }
}