結果

問題 No.1508 Avoid being hit
ユーザー SSRS
提出日時 2021-05-14 23:56:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,346 bytes
コンパイル時間 1,974 ms
コンパイル使用メモリ 181,116 KB
実行使用メモリ 14,796 KB
最終ジャッジ日時 2024-10-02 06:30:17
合計ジャッジ時間 7,705 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int main(){
  int N, Q;
  cin >> N >> Q;
  vector<int> A(Q);
  for (int i = 0; i < Q; i++){
    cin >> A[i];
    A[i]--;
  }
  vector<int> B(Q);
  for (int i = 0; i < Q; i++){
    cin >> B[i];
    B[i]--;
  }
  for (int i = 0; i < Q; i++){
    if (A[i] > B[i]){
      swap(A[i], B[i]);
    }
  }
  vector<vector<int>> E(Q * 2 + 2);
  for (int i = 0; i < Q; i++){
    if (A[i] == 0){
      E[Q * 2].push_back(i * 2);
    }
    if (B[i] == 0){
      E[Q * 2].push_back(i * 2 + 1);
    }
    if (A[i] == N - 1){
      E[i * 2].push_back(Q * 2 + 1);
    }
    if (B[i] == N - 1){
      E[i * 2 + 1].push_back(Q * 2 + 1);
    }
    if (abs(A[i] - B[i]) == 1){
      E[i * 2].push_back(i * 2 + 1);
      E[i * 2 + 1].push_back(i * 2);
    }
    if (i > 0){
      if (A[i] == A[i - 1]){
        E[i * 2 - 2].push_back(i * 2);
        E[i * 2].push_back(i * 2 - 2);
      }
      if (A[i] == B[i - 1]){
        E[i * 2 - 1].push_back(i * 2);
        E[i * 2].push_back(i * 2 - 1);
      }
      if (B[i] == A[i - 1]){
        E[i * 2 - 2].push_back(i * 2 + 1);
        E[i * 2 + 1].push_back(i * 2 - 2);
      }
      if (B[i] == B[i - 1]){
        E[i * 2 - 1].push_back(i * 2 + 1);
        E[i * 2 + 1].push_back(i * 2 - 1);
      }
    }
  }
  vector<bool> used(Q * 2 + 2, false);
  used[Q * 2] = true;
  queue<int> q;
  q.push(Q * 2);
  while (!q.empty()){
    int v = q.front();
    q.pop();
    for (int w : E[v]){
      if (!used[w]){
        used[w] = true;
        q.push(w);
      }
    }
  }
  if (used[Q * 2 + 1]){
    cout << "NO" << endl;
  } else {
    vector<int> L(Q, 0), R(Q, N - 1);
    for (int i = Q - 1; i >= 0; i--){
      if (i < Q - 1){
        L[i] = max(L[i + 1] - 1, 0);
        R[i] = min(R[i + 1] + 1, N - 1);
      }
      if (A[i] == L[i]){
        L[i]++;
      }
      if (B[i] == L[i]){
        L[i]++;
      }
      if (B[i] == R[i]){
        R[i]--;
      }
      if (A[i] == R[i]){
        R[i]--;
      }
    }
    cout << "YES" << endl;
    int S = L[0];
    cout << S + 1 << endl;
    for (int i = 0; i < Q; i++){
      int K = -1;
      for (int j = S - 1; j <= S + 1; j++){
        if (L[i] <= j && j <= R[i] && j != A[i] && j != B[i]){
          K = j;
          break;
        }
      }
      assert(K != -1);
      S = K;
      cout << K + 1 << endl;
    }
  }
}
0