結果

問題 No.1508 Avoid being hit
ユーザー chocoruskchocorusk
提出日時 2021-05-15 00:08:31
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 586 ms / 3,000 ms
コード長 2,671 bytes
コンパイル時間 3,985 ms
コンパイル使用メモリ 197,584 KB
実行使用メモリ 155,796 KB
最終ジャッジ日時 2024-10-02 07:13:52
合計ジャッジ時間 12,744 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 570 ms
155,796 KB
testcase_01 AC 4 ms
8,192 KB
testcase_02 AC 5 ms
8,320 KB
testcase_03 AC 5 ms
8,192 KB
testcase_04 AC 5 ms
8,320 KB
testcase_05 AC 4 ms
8,192 KB
testcase_06 AC 5 ms
8,320 KB
testcase_07 AC 4 ms
8,244 KB
testcase_08 AC 4 ms
8,192 KB
testcase_09 AC 4 ms
8,300 KB
testcase_10 AC 5 ms
8,192 KB
testcase_11 AC 5 ms
8,260 KB
testcase_12 AC 5 ms
8,316 KB
testcase_13 AC 5 ms
8,320 KB
testcase_14 AC 5 ms
8,320 KB
testcase_15 AC 5 ms
8,152 KB
testcase_16 AC 353 ms
54,204 KB
testcase_17 AC 186 ms
33,024 KB
testcase_18 AC 238 ms
40,316 KB
testcase_19 AC 272 ms
42,940 KB
testcase_20 AC 586 ms
86,908 KB
testcase_21 AC 359 ms
57,512 KB
testcase_22 AC 454 ms
68,924 KB
testcase_23 AC 535 ms
78,808 KB
testcase_24 AC 262 ms
46,204 KB
testcase_25 AC 240 ms
42,928 KB
testcase_26 AC 19 ms
10,668 KB
testcase_27 AC 126 ms
25,752 KB
testcase_28 AC 124 ms
25,884 KB
testcase_29 AC 240 ms
43,520 KB
testcase_30 AC 132 ms
27,136 KB
testcase_31 AC 213 ms
39,168 KB
testcase_32 AC 232 ms
41,700 KB
testcase_33 AC 66 ms
17,408 KB
testcase_34 AC 142 ms
9,088 KB
testcase_35 AC 18 ms
8,448 KB
testcase_36 AC 82 ms
8,784 KB
testcase_37 AC 39 ms
8,576 KB
testcase_38 AC 114 ms
8,924 KB
testcase_39 AC 74 ms
8,764 KB
testcase_40 AC 149 ms
9,216 KB
testcase_41 AC 86 ms
8,812 KB
testcase_42 AC 150 ms
9,216 KB
testcase_43 AC 173 ms
9,216 KB
testcase_44 AC 5 ms
8,320 KB
testcase_45 AC 5 ms
8,172 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;
map<int, int> pr[100010];
int main()
{
    int n, q; cin>>n>>q;
	int a[100010], b[100010];
	for(int i=0; i<q; i++){
		cin>>a[i]; a[i]--;
	}
	for(int i=0; i<q; i++){
		cin>>b[i]; b[i]--;
	}
    for(int i=-20; i<=20; i++){
        int x=a[0]+i;
        if(x>=0 && x<n) pr[0][x]=0;
    }
    for(int i=0; i<q; i++){
        for(auto p:pr[i]){
            int x=p.first;
            for(int k=-1; k<=1; k++){
                int y=x+k;
                if(y<0 || y>=n) continue;
                int d1=abs(y-a[i]), d2=abs(y-b[i]);
                if(d1>0 && d2>0 && (d1<=20 || d2<=20)){
                    pr[i+1][y]=x;
                }
            }
        }
    }
    if(pr[q].empty()){
        mt19937 mt(334);
        uniform_int_distribution rndn(0, n-1);
        {
            for(int l=0; l<50; l++){
                int s=rndn(mt);
                vector<int> ans;
                bool dame=0;
                ans.push_back(s+1);
                for(int i=0; i<q; i++){
                    vector<int> v;
                    if(s-1>=0 && a[i]!=s-1 && b[i]!=s-1){
                        v.push_back(s-1);
                    }
                    if(a[i]!=s && b[i]!=s){
                        v.push_back(s);
                    }
                    if(s+1<n && a[i]!=s+1 && b[i]!=s+1){
                        v.push_back(s+1);
                    }
                    if(v.empty()){
                        dame=1;break;
                    }
                    s=v[mt()%((int)v.size())];
                    ans.push_back(s+1);
                }
                if(!dame){
                    cout<<"YES"<<endl;
                    for(auto x:ans) cout<<x<<endl;
                    return 0;
                }
            }
            cout<<"NO"<<endl;
            return 0;
        }
        return 0;
    }
    cout<<"YES"<<endl;
    int x=pr[q].begin()->first;
    vector<int> ans;
    for(int i=q; i>=0; i--){
        ans.push_back(x);
        x=pr[i][x];
    }
    reverse(ans.begin(), ans.end());
    for(auto y:ans) cout<<y+1<<endl;
    return 0;
}
0