蓝桥杯-百亿富翁

题意:n个点,求出每个点的左边,右边,第一个比他高的点。

思路:单调栈。

void solve(){
int n;
cin >> n;

vector a(n);
for (auto&x : a){
cin >> x;
}

stack stk;
vector<pair<int, int>> ans(n, {-1, -1});
stk.push(0);
for (int i = 1; i < n; ++i){
while (!stk.empty() && a[stk.top()] <= a[i]){
stk.pop();
}
if (!stk.empty()){
ans[i].first = stk.top() + 1;
}
stk.push(i);
}
while (!stk.empty()){
stk.pop();
}
stk.push(n - 1);

for (int i = n - 2; i >= 0; --i){
while (!stk.empty() && a[stk.top()] <= a[i]){
stk.pop();
}
if (!stk.empty()){
ans[i].second = stk.top() + 1;
}
stk.push(i);
}

for (int i = 0; i < n; ++i){
cout << ans[i].first << " \n"[i == n - 1];
}
for (int i = 0; i < n; ++i){
cout << ans[i].second << " \n"[i == n - 1];
}
}

总结:为什么stack没有clear呢?

posted @ 2024-03-29 09:18  _Yxc  阅读(39)  评论(0)    收藏  举报