百亿富翁 (单调栈)
生活随笔
收集整理的這篇文章主要介紹了
百亿富翁 (单调栈)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
百億富翁
題目描述
這天小明買彩票中了百億獎(jiǎng)金,興奮的他決定買下藍(lán)橋公司旁的一排連續(xù)的樓房。
已知這排樓房一共有 NN 棟,編號(hào)分別為1~N,第 ii 棟的高度為h i 。
好奇的小明想知道對(duì)于每棟樓,左邊第一個(gè)比它高的樓房是哪個(gè),右邊第一個(gè)比它高的樓房是哪個(gè)(若不存在則輸出 -1)。但由于樓房數(shù)量太多,小明無(wú)法用肉眼直接得到答案,于是他花了 1 個(gè)億來(lái)請(qǐng)你幫他解決問(wèn)題,你不會(huì)拒絕的對(duì)吧?
輸入描述
第 1 行輸入一個(gè)整數(shù) N.表示樓房的數(shù)量。
第 2 行輸入 N 個(gè)整數(shù)(相鄰整數(shù)用空格隔開),分別為 h_1,h_2,…,h_N,表示樓房的高度。
1 <= N <= 7 x 10 ^ 5, 1 <= hi <= 10 ^ 9
輸出描述
輸出共兩行。
第一行輸出 N 個(gè)整數(shù),表示每棟樓左邊第一棟比自己高的樓的編號(hào)。
第二行輸出 N 個(gè)整數(shù),表示每棟樓右邊第一棟比自己高的樓的編號(hào)。
輸入輸出樣例
示例 1
輸入
5
3 1 2 5 4
輸出
-1 1 1 -1 4
4 3 4 -1 -1
AC代碼
#include<bits/stdc++.h> using namespace std; const int N = 3e6 + 10; stack<int>st; int n , a[N] , l[N] , r[N]; int main() {cin >> n;for(int i = 1 ; i <= n ; i ++) cin >> a[i];for(int i = 1 ; i <= n ; i ++){while(st.size() && a[st.top()] <= a[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 >= 1 ; i --){while(st.size() && a[st.top()] <= a[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; }總結(jié)
以上是生活随笔為你收集整理的百亿富翁 (单调栈)的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 树莓派驱动1.44寸TFT液晶并实时显示
- 下一篇: amoled led 排列_AMOLED