結果

問題 No.1508 Avoid being hit
コンテスト
ユーザー SSRS
提出日時 2021-05-15 00:00:50
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,458 bytes
コンパイル時間 2,065 ms
コンパイル使用メモリ 179,548 KB
実行使用メモリ 14,848 KB
最終ジャッジ日時 2024-10-02 06:40:02
合計ジャッジ時間 7,289 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41 RE * 2
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:102:17: warning: 'S' may be used uninitialized [-Wmaybe-uninitialized]
  102 |     cout << S + 1 << endl;
      |                 ^
main.cpp:95:9: note: 'S' was declared here
   95 |     int S;
      |         ^

ソースコード

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;
    for (int i = L[0]; i <= R[0]; i++){
      if (i != A[0] && i != B[0]){
        S = i;
        break;
      }
    }
    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