結果

問題 No.1804 Intersection of LIS
ユーザー gyouzasushi
提出日時 2022-08-27 20:05:50
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,970 bytes
コンパイル時間 2,438 ms
コンパイル使用メモリ 205,056 KB
最終ジャッジ日時 2025-01-31 06:31:03
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 34 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
template <typename T, typename Compare>
vector<int> LongestIncreasingSubsequence(vector<T> &A, Compare comp) {
    int N = (int)A.size();
    vector<T> DP;
    vector<int> nxt(N, N);
    vector<int> mem(N);
    T inf;
    if (comp(0, numeric_limits<T>::max()))
        inf = numeric_limits<T>::min();
    else
        inf = numeric_limits<T>::max();
    auto comp2 = [&](T l, T r) { return !comp(l, r); };
    for (int i = N - 1; i >= 0; i--) {
        auto it = upper_bound(DP.begin(), DP.end(), A[i], comp2);
        int d = it - DP.begin();
        if (it == DP.end())
            DP.emplace_back(A[i]);
        else
            *it = A[i];
        mem[i] = d;
    }
    stack<int> st;
    for (int i = N - 1; i >= 0; i--) {
        while (!st.empty() && !comp(A[i], A[st.top()])) st.pop();
        if (!st.empty()) nxt[i] = st.top();
        st.emplace(i);
    }
    int len = (int)DP.size();
    vector<int> ret;
    for (int i = 0; len && i < N; i++) {
        if (mem[i] == len - 1) {
            ret.emplace_back(i);
            len--;
        }
    }
    return ret;
}

int main() {
    int n;
    cin >> n;
    vector<int> v(n);
    for (int i = 0; i < n; i++) cin >> v[i];
    auto lis_min = LongestIncreasingSubsequence(v, less<int>());
    reverse(v.begin(), v.end());
    auto lis_max = LongestIncreasingSubsequence(v, greater<int>());
    reverse(lis_max.begin(), lis_max.end());
    for (int i = 0; i < lis_max.size(); i++) lis_max[i] = n - 1 - lis_max[i];
    reverse(v.begin(), v.end());

    vector<int> ans;
    int m = lis_min.size();
    assert(int(lis_max.size()) == m);
    for (int i = 0; i < m; i++) {
        cerr << lis_min[i] << " " << lis_max[i] << '\n';
        if (lis_min[i] == lis_max[i]) {
            ans.push_back(v[lis_min[i]]);
        }
    }
    cout << ans.size() << '\n';
    for (int i = 0; i < ans.size(); i++) {
        cout << ans[i] << " \n"[i == int(ans.size()) - 1];
    }
}
0