向右看齐Look Up
题目
约翰的N(1≤N≤10^5)头奶牛站成一排,奶牛i的身高是Hi(l≤Hi≤1,000,000).现在,每只奶牛都在向右看齐.对于奶牛i,如果奶牛j满足i<j且Hi<Hj,我们可以说奶牛i可以仰望奶牛j. 求出每只奶牛离她最近的仰望对象.
输入
第 1 行输入 N,之后每行输入一个身高 H_i。
输出
共 N 行,按顺序每行输出一只奶牛的最近仰望对象,如果没有仰望对象,输出 0。
输入样例
6
3
2
6
1
1
2
输出样例
3
3
0
6
6
0
说明
【输入说明】6 头奶牛的身高分别为 3, 2, 6, 1, 1, 2.
【输出说明】奶牛#1,#2 仰望奶牛#3,奶牛#4,#5 仰望奶牛#6,奶牛#3 和#6 没有仰望对象。
数据范围
对于 20%的数据: 1≤N≤10;
对于 50%的数据: 1≤N≤1,000;
对于 100%的数据:1≤N≤100,000;1≤H_i≤1,000,000;
解题思路
其实这道题就是一道单间栈,先把第一个奶牛给扔进去,然后再判断是否有比它大的,如果有大的就出栈,比它小的就把它指向栈里的奶牛然后重复执行即可.
程序如下
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int n,a[100001],ans[100001],sum;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;++i)
scanf("%d",&a[i]);
for(int i=n-1;i>=1;--i)
{
sum=i+1;
while(a[i]>=a[sum]&&a[sum]>0) sum=ans[sum];//判断是否有比它大的
ans[i]=sum;//有就替换
}
for(int i=1;i<=n;++i) printf("%d\n",ans[i]);
return 0;
}
来源:https://blog.csdn.net/CWH2018/article/details/99292019