Solution
思路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;
}
}