思路1:
暴力
class Solution { public int[] nextGreaterElement(int[] nums1, int[] nums2) { int[] ans = new int[nums1.length]; int cur = 0; for (int x: nums1) { boolean flag = false; boolean isOk = false; for (int y: nums2) { if (flag && y > x) { ans[cur++] = y; isOk = true; break; } if (y == x) flag = true; } if (isOk == false) ans[cur++] = -1; } return ans; } }
思路2:
首先考虑因为要找右边的第一个比它大的数,所有从右边往左维护,就容易想到单调栈,遇到比自己小的就出栈,找到之前比自己大的第一个数,然后用Map映射一下。
class Solution { public int[] nextGreaterElement(int[] nums1, int[] nums2) { int n = nums1.length, m = nums2.length; Deque<Integer> stack = new ArrayDeque<Integer>(); Map<Integer, Integer> mp = new HashMap<Integer, Integer>(); int[] ans = new int[n]; for (int i = m - 1; i >= 0; i--) { int x = nums2[i]; while (!stack.isEmpty() && stack.peek() <= x) stack.pop(); mp.put(x, stack.isEmpty() ? -1 : stack.peek()); stack.push(x); } for (int i = 0; i < n; i++) { nums1[i] = mp.get(nums1[i]); } return nums1; } }