题三：找好朋友

题目

        在学校中，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;
    }
}