結果

問題 No.3121 Prime Dance
ユーザー hiikunZ
提出日時 2025-04-19 00:15:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,883 ms / 2,000 ms
コード長 2,967 bytes
コンパイル時間 2,823 ms
コンパイル使用メモリ 227,664 KB
実行使用メモリ 52,992 KB
最終ジャッジ日時 2025-04-19 00:15:24
合計ジャッジ時間 13,547 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

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 = 1700000000000000000;
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(100000,1);
    is_prime[0] = is_prime[1] = 0;
    for(ll i = 2;i < 100000;i++){
        if(is_prime[i] == 0) continue;
        for(ll j = i * 2;j < 100000;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];
    }
    S[Sx][Sy] = '.';
    S[Gx][Gy] = '.';
    bool seen[40][40][170][170];
    for(ll i = 0; i < H; i++){
        for(ll j = 0; j < W; j++){
            for(ll k = 0; k < 170; k++){
                for(ll l = 0; l < 170; 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 >= 170) continue;
            }
            else if(dy[i] == 1){
                nY++;
                if(nY >= 170) 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 i = 0;i < 170;i++){
        for(ll j = 0;j < 170;j++){
            if(seen[Gx][Gy][i][j] == 0) continue;
            ll i2 = i - (Gx - Sx),j2 = j - (Gy - Sy);
            if((i > 0 || i2 > 0) && (j > 0 || j2 > 0)){
                ll K = MAX,L = MAX;
                for(ll k = 0;k < 500;k++){
                    if(is_prime[i + k] == 1 && is_prime[i2 + k] == 1){
                        K = k;
                        break;
                    }
                }
                for(ll k = 0;k < 500;k++){
                    if(is_prime[j + k] == 1 && is_prime[j2 + k] == 1){
                        L = k;
                        break;
                    }
                }
                ans = min(ans,i + j + i2 + j2 + 2 * K + 2 * L);

                break;
            }
        }
    }
    if(ans == MAX) cout << -1 << endl;
    else cout << ans << endl;
}
0