找朋友（单调栈）

问题描述

在学校中，N个小朋友站成一队， 第i个小朋友的身高为height[i]， 第i个小朋友可以看到的第一个比自己身高更高的小朋友j，那么j是i的好朋友【j>i】。
请重新生成一个列表，对应位置的输出是每个小朋友的好朋友位置，如果没有看到好朋友，请在该位置用0代替。
小朋友人数范围是 [0, 40000]

解题思路

代码编写

import java.util.Deque;
import java.util.LinkedList;
import java.util.Scanner;


public class GoodFriend {
    public static void main(String[] args) {
        System.out.println("请输入n个小朋友身高");
        Scanner sc=new Scanner(System.in);
        int n=sc.nextInt();

        System.out.println("请输入小朋友的身高");
        int[] height=new int[n];
        for(int i=0;i<n;i++){
            height[i]=sc.nextInt();
        }

        int[] result=new int[n];

        Deque<Integer> stack=new LinkedList<Integer>();
        for (int i=0;i<n;i++){
            //当前即将入栈的小朋友
            int h=height[i];
            while(!stack.isEmpty() && h>height[stack.peek()]){
                //头元素
                result[stack.peek()]=i;
                stack.pop();

            }
            //栈中存小朋友的位置
            stack.push(i);

        }