#66. 寻找小朋友的好朋友

寻找小朋友的好朋友

题目

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

输入描述

第一行输入N,N表示有N个小朋友 第二行输入N个小朋友的身高height[i],都是整数

输出描述

输出N个小朋友的好朋友的位置

用例

输入	
2
100 95
输出	
0 0

解题思路

利用单调栈来找到数组中每个元素右侧第一个比它大的元素的索引。首先遍历输入数组,并使用栈来存储元素及其索引。对于每个元素,如果栈顶元素小于当前元素,则更新栈顶元素在结果数组中的索引值,并弹出栈顶元素。重复此过程,直到栈顶元素不小于当前元素,然后将当前元素及其索引入栈。