結果

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

ソースコード

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;
      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;
        }
      }
      S = K;
      cout << K + 1 << endl;
    }
  }
}
0