快轉到主要內容

CSES-1645 Nearest Smaller Values

標籤: 單調棧
目錄

題目連結:https://cses.fi/problemset/task/1645

題意
#

對於長度為 n 的陣列的每個元素
找尋前面最近比它小的元素

思路
#

維護 Monotonic stack,其內容元素皆小於當前元素

程式碼
#

#include <bits/stdc++.h>
using namespace std;

#define pb push_back
#define fi first
#define se second
#define INF LONG_LONG_MAX/1000
#define WA() cin.tie(0)->sync_with_stdio(0)
#define all(x) (x).begin(), (x).end()
#define int long long
#define PII pair<int, int>

signed main() { WA();
    int n; cin >> n;
    vector<int> v(n);
    for (auto &i : v) cin >> i;
    stack<int> sk;
    for (auto &i : v) {
        while (sk.size() && i <= v[sk.top()]) sk.pop();
        cout << (sk.size() ? sk.top()+1 : 0) << ' ';
        sk.push(&i-v.data());
    }
}
Piau 的筆記本
作者
Piau 的筆記本
希望我寫下來的東西能夠長久的記在我的腦中