結果

問題 No.1508 Avoid being hit
ユーザー どらら
提出日時 2021-05-14 23:32:03
言語 C++14
(gcc 13.3.0 + boost 1.87.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
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 31 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
}

0