蓝桥 百亿富翁 stack＜Type＞ 栈

题目描述

这天小明买彩票中了百亿奖金，兴奋的他决定买下蓝桥公司旁的一排连续的楼房。

已知这排楼房一共有 N栋，编号分别为 1∼N，第 i栋的高度为hi​。

好奇的小明想知道对于每栋楼，左边第一个比它高的楼房是哪个，右边第一个比它高的楼房是哪个（若不存在则输出 -1）。但由于楼房数量太多，小明无法用肉眼直接得到答案，于是他花了 1 个亿来请你帮他解决问题，你不会拒绝的对吧？

输入描述

第 1 行输入一个整数 NN，表示楼房的数量。

第 2 行输入 N 个整数（相邻整数用空格隔开），分别为h1​,h2​,...,hN​，表示楼房的高度。

1 \leq N \leq 7\times 10^51≤N≤7×105，1\leq h_i \leq 10^91≤hi​≤109。

输出描述

输出共两行。

第一行输出 N个整数，表示每栋楼左边第一栋比自己高的楼的编号。

第二行输出 N 个整数，表示每栋楼右边第一栋比自己高的楼的编号。

输入输出样例

示例 1

输入

5
3 1 2 5 4 

输出

-1 1 1 -1 4
4 3 4 -1 -1

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

题解代码

单调栈，就是一个栈，不过栈内元素保证单调性。即，栈内元素要么从小到大，要么从大到小。 如何求右边第一个比它高的楼房？ 通过观察，我们会发现，在当前点后面的，比它矮的，一定会被它遮住。那么，这个点就没用了。而根 据现实生活和刚才的推断，我们看到的楼房肯定是一个比一个高的，而没看到的，留着也没用，直接扔 了QwQ。那么，这就是符合单调性的。再看，那些没用的楼房什么时候扔掉？当然是遇到比它高的楼房 了。那么就可以一个一个地弹出，而且肯定是在已经判断过的人的前面（中间和后面的在之前就弹出 了），所以就直接从前面弹出。咦？这不就像一个栈吗？没错，这就是单调栈的实现方法。 再归纳一下： 从后往前扫 对于每个点： 弹出栈顶比它小的元素 此时栈顶就是答案 加入这个元素

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=3e6+10;
int h[N],l[N],r[N];
int n;
stack<int> st; //栈 先进后出

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	   cin>>h[i];
	   
	for(int i=1;i<=n;i++){
		while(st.size()&&h[st.top()]<=h[i]) st.pop(); //移除比自身矮的（左边） 
		if(st.size()) l[i]=st.top();
		else l[i]=-1;
		st.push(i);
	}
	
	while(!st.empty()) st.pop();  //清空栈 
	
	for(int i=n;i>0;i--){
		while(st.size()&&h[st.top()]<=h[i]) st.pop();  //移除比自身矮的（右边）
		if(st.size()) r[i]=st.top();
		else r[i]=-1;
		st.push(i);    
	}
	
	for(int i=1;i<=n;i++) cout<<l[i]<<" \n"[i==n];
	for(int i=1;i<=n;i++) cout<<r[i]<<" \n"[i==n];
	
	return 0;
 }