結果

問題 No.1508 Avoid being hit
ユーザー SSRSSSRS
提出日時 2021-05-15 00:02:52
言語 C++14
(gcc 11.2.0 + boost 1.78.0)
結果
AC  
実行時間 139 ms / 3,000 ms
コード長 2,666 bytes
コンパイル時間 1,555 ms
使用メモリ 14,572 KB
最終ジャッジ日時 2022-11-26 01:07:45
合計ジャッジ時間 6,787 ms
ジャッジサーバーID
(参考情報)
judge11 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 133 ms
13,148 KB
testcase_01 AC 2 ms
3,524 KB
testcase_02 AC 1 ms
3,452 KB
testcase_03 AC 1 ms
3,532 KB
testcase_04 AC 1 ms
3,536 KB
testcase_05 AC 1 ms
3,424 KB
testcase_06 AC 1 ms
3,376 KB
testcase_07 AC 2 ms
3,448 KB
testcase_08 AC 2 ms
3,552 KB
testcase_09 AC 1 ms
3,376 KB
testcase_10 AC 1 ms
3,452 KB
testcase_11 AC 1 ms
3,380 KB
testcase_12 AC 1 ms
3,496 KB
testcase_13 AC 1 ms
3,532 KB
testcase_14 AC 3 ms
3,528 KB
testcase_15 AC 1 ms
3,504 KB
testcase_16 AC 55 ms
13,284 KB
testcase_17 AC 32 ms
9,060 KB
testcase_18 AC 25 ms
7,268 KB
testcase_19 AC 48 ms
11,904 KB
testcase_20 AC 69 ms
14,572 KB
testcase_21 AC 44 ms
10,812 KB
testcase_22 AC 54 ms
12,840 KB
testcase_23 AC 58 ms
13,400 KB
testcase_24 AC 139 ms
12,860 KB
testcase_25 AC 126 ms
12,024 KB
testcase_26 AC 11 ms
3,716 KB
testcase_27 AC 66 ms
7,664 KB
testcase_28 AC 65 ms
7,624 KB
testcase_29 AC 136 ms
12,276 KB
testcase_30 AC 74 ms
8,148 KB
testcase_31 AC 117 ms
11,004 KB
testcase_32 AC 127 ms
11,804 KB
testcase_33 AC 36 ms
5,572 KB
testcase_34 AC 110 ms
7,528 KB
testcase_35 AC 12 ms
3,664 KB
testcase_36 AC 63 ms
5,820 KB
testcase_37 AC 30 ms
4,148 KB
testcase_38 AC 91 ms
6,740 KB
testcase_39 AC 59 ms
5,560 KB
testcase_40 AC 115 ms
7,908 KB
testcase_41 AC 70 ms
5,760 KB
testcase_42 AC 116 ms
7,800 KB
testcase_43 AC 138 ms
8,852 KB
testcase_44 AC 2 ms
3,524 KB
testcase_45 AC 1 ms
3,484 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: 関数 ‘int main()’ 内:
main.cpp:105:34: 警告: ‘S’ はこの関数内初期化されずに使用されるかもしれません [-Wmaybe-uninitialized]
  105 |       for (int j = S - 1; j <= S + 1; j++){
      |                                ~~^~~

ソースコード

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]--;
      }
    }
    int S;
    for (int i = L[0] - 1; i <= R[0] + 1; i++){
      if (0 <= i && i < N && i != A[0] && i != B[0]){
        S = i;
      }
    }
    bool ok = true;
    vector<int> ans(Q + 1);
    ans[0] = S;
    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;
        }
      }
      if (K == -1){
        ok = false;
        break;
      }
      S = K;
      ans[i + 1] = K;
    }
    if (!ok){
      cout << "NO" << endl;
    } else {
      cout << "YES" << endl;
      for (int i = 0; i <= Q; i++){
        cout << ans[i] + 1 << endl;
      }
    }
  }
}
0