#657498 (C++14) No.1508 Avoid being hit

• 問題: No.1508 Avoid being hit
• ユーザー: どらら
• 提出日時: 2021-05-14 23:32:03
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,990 bytes
• コンパイル時間: 3,916 ms
• コンパイル使用メモリ: 232,460 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-10-02 05:32:11
• 合計ジャッジ時間: 8,734 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	AC	2 ms; 6,816 KB
testcase_02	AC	2 ms; 6,816 KB
testcase_03	AC	2 ms; 6,820 KB
testcase_04	AC	2 ms; 6,816 KB
testcase_05	AC	2 ms; 6,820 KB
testcase_06	AC	2 ms; 6,816 KB
testcase_07	AC	2 ms; 6,816 KB
testcase_08	AC	2 ms; 6,816 KB
testcase_09	AC	2 ms; 6,820 KB
testcase_10	AC	2 ms; 6,820 KB
testcase_11	AC	2 ms; 6,820 KB
testcase_12	AC	2 ms; 6,816 KB
testcase_13	AC	2 ms; 6,820 KB
testcase_14	WA	-
testcase_15	AC	2 ms; 6,820 KB
testcase_16	AC	39 ms; 6,816 KB
testcase_17	AC	23 ms; 6,816 KB
testcase_18	AC	22 ms; 6,816 KB
testcase_19	AC	33 ms; 6,816 KB
testcase_20	AC	55 ms; 6,816 KB
testcase_21	AC	34 ms; 6,820 KB
testcase_22	AC	42 ms; 6,816 KB
testcase_23	AC	47 ms; 6,816 KB
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	AC	131 ms; 6,816 KB
testcase_35	AC	15 ms; 6,820 KB
testcase_36	AC	75 ms; 6,820 KB
testcase_37	AC	35 ms; 6,816 KB
testcase_38	AC	105 ms; 6,816 KB
testcase_39	AC	70 ms; 6,816 KB
testcase_40	AC	136 ms; 6,820 KB
testcase_41	AC	78 ms; 6,816 KB
testcase_42	AC	137 ms; 6,816 KB
testcase_43	AC	158 ms; 6,820 KB
testcase_44	AC	2 ms; 6,820 KB
testcase_45	AC	1 ms; 6,820 KB

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
#define REP(i,a,n) for(int i=(a); i<(int)(n); i++)
#define rep(i,n) REP(i,0,n)
#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it)
#define ALLOF(c) (c).begin(), (c).end()
typedef long long ll;
typedef unsigned long long ull;
//using mint = modint1000000007;
//using mint = modint998244353;

vector<int> solve(int N, const vector<pair<int,int>>& v, int pos){
  int Q = v.size();

  vector<int> ret;
  
  rep(i,Q){
    int a = v[i].first;
    int b = v[i].second;

    if(pos != a && pos != b){
      ;
    }else{
      if(pos+1<=N && pos+1 != a && pos+1 != b){
        pos++;
      }
      else if(pos-1>=1 && pos-1 != a && pos-1 != b){
        pos--;
      }
      else{
        return vector<int>();
      }
    }

    ret.push_back(pos);
  }

  return ret;
}


int main(){
  int N, Q;
  cin >> N >> Q;
  vector<pair<int,int>> v(Q);
  rep(i,Q) cin >> v[i].first;
  rep(i,Q) cin >> v[i].second;

  if(v[0].first-1 >= 1){
    vector<int> ret = solve(N, v, v[0].first-1);
    if(ret.size() > 0){
      cout << "YES" << endl;
      cout << ret[0] << endl;
      rep(i,ret.size()) cout << ret[i] << endl;
      return 0;
    }
  }
  if(v[0].second-1 >= 1){
    vector<int> ret = solve(N, v, v[0].second-1);
    if(ret.size() > 0){
      cout << "YES" << endl;
      cout << ret[0] << endl;
      rep(i,ret.size()) cout << ret[i] << endl;
      return 0;
    }
  }
  if(v[0].first+1 <= N){
    vector<int> ret = solve(N, v, v[0].first+1);
    if(ret.size() > 0){
      cout << "YES" << endl;
      cout << ret[0] << endl;
      rep(i,ret.size()) cout << ret[i] << endl;
      return 0;
    }
  }
  if(v[0].second+1 <= N){
    vector<int> ret = solve(N, v, v[0].second+1);
    if(ret.size() > 0){
      cout << "YES" << endl;
      cout << ret[0] << endl;
      rep(i,ret.size()) cout << ret[i] << endl;
      return 0;
    }
  }

  cout << "NO" << endl;
  
  return 0;
}