結果

問題 No.1804 Intersection of LIS
ユーザー SSRSSSRS
提出日時 2022-01-07 22:00:37
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 157 ms / 2,000 ms
コード長 1,601 bytes
コンパイル時間 1,804 ms
コンパイル使用メモリ 172,220 KB
実行使用メモリ 16,256 KB
最終ジャッジ日時 2024-11-14 08:47:10
合計ジャッジ時間 5,893 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const int INF = 1000000;
int main(){
  int N;
  cin >> N;
  vector<int> P(N);
  for (int i = 0; i < N; i++){
    cin >> P[i];
    P[i]--;
  }
  vector<int> dpL(N, INF);
  vector<int> L(N, 0);
  for (int i = 0; i < N; i++){
    L[i] = lower_bound(dpL.begin(), dpL.end(), P[i]) - dpL.begin();
    dpL[L[i]] = P[i];
  }
  vector<int> Q(N);
  for (int i = 0; i < N; i++){
    Q[i] = N - 1 - P[i];
  }
  vector<int> dpR(N, INF);
  vector<int> R(N, 0);
  vector<int> pr(N);
  for (int i = N - 1; i >= 0; i--){
    R[i] = lower_bound(dpR.begin(), dpR.end(), Q[i]) - dpR.begin();
    pr[i] = N - 1 - dpR[R[i]];
    if (pr[i] < 0){
      pr[i] = -INF;
    }
    dpR[R[i]] = Q[i];
  }
  int lis = lower_bound(dpL.begin(), dpL.end(), INF) - dpL.begin();
  vector<int> A(lis + 1, INF), B(lis + 1, -INF);
  A[0] = -INF;
  B[lis] = INF;
  for (int i = N - 1; i >= 0; i--){
    B[lis - 1 - R[i]] = P[i];
  }
  int cnt = 0;
  for (int i = 0; i <= lis; i++){
    if (A[i] < B[i]){
      cnt++;
    }
  }
  vector<int> ans;
  for (int i = 0; i < N; i++){
    if (A[lis - 1 - R[i]] < B[lis - 1 - R[i]]){
      cnt--;
    }
    B[lis - 1 - R[i]] = pr[i];
    if (A[lis - 1 - R[i]] < B[lis - 1 - R[i]]){
      cnt++;
    }
    if (cnt == 0){
      ans.push_back(P[i]);
    }
    if (A[L[i] + 1] < B[L[i] + 1]){
      cnt--;
    }
    A[L[i] + 1] = P[i];
    if (A[L[i] + 1] < B[L[i] + 1]){
      cnt++;
    }
  }
  int x = ans.size();
  cout << x << endl;
  for (int i = 0; i < x; i++){
    cout << ans[i] + 1;
    if (i <x - 1){
      cout << ' ';
    }
  }
  cout << endl;
}
0