华为OD机试(C卷,100分)- 找朋友
题目描述
在学校中,N 个小朋友站成一队, 第 i 个小朋友的身高为height[i],第 i 个小朋友可以看到的第一个比自己身高更高的小朋友 j ,那么 j 是 i 的好朋友(要求j > i)。
请输出一个数组,对应位置的输出是每个小朋友的好朋友位置,如果没有看到好朋友,则输出0。
输入描述
第一行输入N,N表示有N个小朋友
第二行输入N个小朋友的身高height[i],都是整数
输出描述
输出N个小朋友的好朋友的位置
示例1:
输入
2
100 95
输出
0 0
示例2:
输入:
8
123 124 125 121 119 122 126 123
输出:
1 2 6 5 5 6 0 0
解题思路
利用单调栈来找到数组中每个元素右侧第一个比它大的元素的索引。首先遍历输入数组,并使用栈来存储元素及其索引。对于每个元素,如果栈顶元素小于当前元素,则找到比当前元素高的小朋友,更新栈顶元素在结果数组中的索引值,并弹出栈顶元素。重复此过程,直到栈顶元素不小于当前元素,然后将当前元素及其索引入栈。
#include <stdio.h>
#include <stdlib.h>
int main()
{
int n;
scanf("%d",&n);
int *height = (int *)malloc(n * sizeof(int));
for(int i=0;i<n;i++){
scanf("%d",<