题目
在学校中,N个小朋友站成一队, 第i个小朋友的身高为height[i],第i个小朋友可以看到的第一个比自己身高更高的小朋友j,那么j是i的好朋友(要求j > i)。 请重新生成一个列表,对应位置的输出是每个小朋友的好朋友位置,如果没有看到好朋友,请在该位置用0代替。小朋友人数范围是 [0, 40000]。
输入:
第一行N,N表示有N个小朋友
第二行输入N个小朋友的身高
示例:
输入
2
100,95
输出:
0,0
思路
数组,栈
题解
import java.util.Arrays;
import java.util.Stack;
public class Solution {
public int[] findFriends(int[] height) {
int n = height.length;
int[] friends = new int[n];
Arrays.fill(friends, 0); // 初始化好朋友位置为0
Stack<Integer> stack = new Stack<>(); // 创建一个单调递减的栈
for (int i = 0; i < n; i++) {
while (!stack.isEmpty() && height[i] > height[stack.peek()]) {
// 如果栈不为空,且当前小朋友的身高大于栈顶小朋友的身高
// 那么栈顶小朋友的好朋友就是当前小朋友
friends[stack.pop()] = i;
}
stack.push(i); // 将当前小朋友的位置入栈
}
return friends;
}
}