結果

問題 No.2402 Dirty Stairs and Shoes
ユーザー ochiaigawaochiaigawa
提出日時 2023-08-04 22:05:48
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 961 bytes
コンパイル時間 2,611 ms
コンパイル使用メモリ 206,644 KB
実行使用メモリ 13,888 KB
最終ジャッジ日時 2024-12-20 02:40:39
合計ジャッジ時間 3,912 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
int main() {
  int N, K;
  cin >> N >> K;
  vector<int> state(N + 1, 0);
  int M1;
  cin >> M1;
  while(M1--) {
    int A;
    cin >> A;
    state[A] = 1;
  }
  int M2;
  cin >> M2;
  while(M2--) {
    int B;
    cin >> B;
    state[B] = 2;
  }
  vector<vector<int>> dp(N + 1, vector<int>(2, -1));
  dp[0][0] = 2;
  for(int i = 0; i < N; i++) {
    if(state[i + 1] == 0) {
      dp[i + 1][0] = max(dp[i + 1][0], dp[i][0]);
      dp[i + 1][1] = max(dp[i + 1][1], dp[i][1]);
    } else if(state[i + 1] == 1) {
      dp[i + 1][1] = 1;
    } else {
      dp[i + 1][0] = 2;
    }
    if(i + K <= N) {
      if(state[i + K] == 0) {
        dp[i + K][0] = max(dp[i + K][0], dp[i][0]);
        dp[i + K][1] = max(dp[i + K][1], dp[i][1]);
      } else if(state[i + K] == 1) {
        dp[i + K][1] = 1;
      } else {
        dp[i + K][0] = 2;
      }
    }
  }
  cout << (dp.back()[0] == 2 ? "Yes\n" : "No\n");
  return 0;
}
0