結果

問題 No.3121 Prime Dance
ユーザー hiikunZ
提出日時 2025-04-18 23:57:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,955 bytes
コンパイル時間 2,788 ms
コンパイル使用メモリ 228,176 KB
実行使用メモリ 78,372 KB
最終ジャッジ日時 2025-04-18 23:57:38
合計ジャッジ時間 18,456 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other AC * 7 WA * 13 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize "Ofast,unroll-loops"
#include <bits/stdc++.h>
using namespace std;
//#include<atcoder/all>
//using namespace atcoder;
using ll = long long int;
using ull = unsigned long long int;
using ld = long double;
constexpr ll MAX = 2000000000000000000;
constexpr ld PI = 3.14159265358979;
constexpr ll MOD = 0;//2024948111;
ld dotorad(ld K){ return PI * K / 180.0; }
ld radtodo(ld K){ return K * 180.0 / PI; }
mt19937 mt;
void randinit(){ srand((unsigned)time(NULL));mt = mt19937(rand()); }
const int dx[4] = {1,0,-1,0};
const int dy[4] = {0,1,0,-1};
int main(){
    vector<ll> is_prime(10000,1);
    is_prime[0] = is_prime[1] = 0;
    for(ll i = 2;i < 10000;i++){
        if(is_prime[i] == 0) continue;
        for(ll j = i * 2;j < 10000;j += i){
            is_prime[j] = 0;
        }
    }
    ll H,W;
    cin >> H >> W;
    ll Sx,Sy,Gx,Gy;
    cin >> Sx >> Sy >> Gx >> Gy;
    Sx--;Sy--;Gx--;Gy--;
    vector<string> S(H);
    for(ll i = 0; i < H; i++){
        cin >> S[i];
    }
    if(Sx > Gx){
        Sx = H - Sx - 1;
        Gx = H - Gx - 1;
        reverse(S.begin(),S.end());
    }
    if(Sy > Gy){
        Sy = W - Sy - 1;
        Gy = W - Gy - 1;
        for(ll i = 0; i < H; i++){
            reverse(S[i].begin(),S[i].end());
        }
    }
    assert(S[Sx][Sy] == 'S');
    assert(S[Gx][Gy] == 'G');
    S[Sx][Sy] = '.';
    S[Gx][Gy] = '.';
    bool seen[H][W][200][200];
    for(ll i = 0; i < H; i++){
        for(ll j = 0; j < W; j++){
            for(ll k = 0; k < 200; k++){
                for(ll l = 0; l < 200; l++){
                    seen[i][j][k][l] = 0;
                }
            }
        }
    }
    queue<tuple<ll,ll,ll,ll>> que;
    que.emplace(Sx,Sy,0,0);
    seen[Sx][Sy][0][0] = 1;
    while(!que.empty()){
        auto [x,y,X,Y] = que.front();
        que.pop();
        for(ll i = 0; i < 4; i++){
            ll nx = x + dx[i],ny = y + dy[i];
            if(nx < 0 || H <= nx || ny < 0 || W <= ny) continue;
            if(S[nx][ny] == '#') continue;
            ll nX = X,nY = Y;
            if(dx[i] == 1){
                nX++;
                if(nX >= 200) continue;
            }
            else if(dy[i] == 1){
                nY++;
                if(nY >= 200) continue;
            }
            if(seen[nx][ny][nX][nY] != 0) continue;
            seen[nx][ny][nX][nY] = 1;
            que.emplace(nx,ny,nX,nY);
        }
    }
    ll ans = MAX;
    for(ll k = 0;k < 200;k++){
        for(ll i = 0;i < 200;i++){
            for(ll j = 0;j < 200;j++){
                if(seen[Gx][Gy][i][j] == 0) continue;
                ll i2 = i + (Gx - Sx) + k,j2 = j + (Gy - Sy) + k;
                if(i2 < 0 || j2 < 0) continue;
                if(is_prime[i] == 1 && is_prime[j] == 1 && is_prime[i2] == 1 && is_prime[j2] == 1){
                    ans = min(ans,i + j + i2 + j2);
                }
            }
        }
    }
    if(ans == MAX) cout << -1 << endl;
    else cout << ans << endl;
}
0