yukicoder No.308 素数は通れません

問題

No.308 素数は通れません - yukicoder

解法

w = 1, 2, 3, ... と調べていくとわかるのですが, w = 7 までは, 素数に囲まれて 20 程度までしか辿り着けません。
一方で, w = 8 になると, n が 32 以上になれば必ず 8m+2, 8m+4, 8m+6, 8m+8 の列にたどり着けるので, ほとんどすべての数に辿り着けます。例えば 8m+3 に辿り着きたかったら, 8k+2 の列を辿って行って, 最後に 8m+2 から 8m+3 に移るなどすれば勝ちです。

ただ, 8m+1 の場合だけは, w=8 にするとマズイ場合があって, 8m-7 が素数だと8m+1 にたどり着くための道が塞がれてしまいます。よって, 与えられた整数 N が十分大きい時, N-8 が素数であるかどうか確認する必要がありますが, これにはミラー・ラビンテストというのがあるのでこれで調べます。
http://kreisel.fam.cx/webmaster/clog/img/www.ice.nuie.nagoya-u.ac.jp/~h003149b/lang/miller.html

w=8 で行けない場合, 実は w=14 で必ず辿り着けます。これは 14m+1 = 8k+1 となる素数が無いこと, および 14 以下だと途中で塞がれることから示せます。

typedef __int128 ll;

const int MAXN = 100;
bool prime[MAXN];
bool done[MAXN][MAXN];

ll mulMod(ll x, ll k, ll mod) {
    if (k == 0) return 0;
    if (k%2 == 0) return mulMod((x+x)%mod, k/2, mod);
    return (x+mulMod(x, k-1, mod)) % mod;
}

ll pow_mod(ll a, ll p, ll mod) {
    if (a == 0) return 0;
    if (p == 0) return 1;
    if (p == 1) return a;
    if (p%2) return mulMod(a, pow_mod(a, p-1, mod), mod);
    ll tmp = pow_mod(a, p/2, mod);
    return mulMod(tmp, tmp, mod);
}

bool check(ll a, ll n, int k, ll q) {
    ll tmp = pow_mod(a, q, n);
    if (tmp == 1) return true;
    for (int i = 0; i < k; i++) {
        if (tmp == n-1) return true;
        tmp = mulMod(tmp, tmp, n);
    }
    return false;
}

bool isPrime(ll n) {
    ll q = n-1;
    int k = 0;
    while (q%2==0) {
        q /= 2;
        k++;
    }
    vector<ll> p = {2, 3, 5, 7, 11, 13, 17, 19, 23, 27, 29, 31, 37, 41};
    for (ll a : p) {
        if (!check(a, n, k, q)) return false;
    }
    return true;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    for (int i = 2; i < MAXN; i++) prime[i] = true;
    for (int i = 2; i < MAXN; i++) {
        if (prime[i]) for (int j = 2; j*i < MAXN; j++) prime[i*j] = false;
    }
    string s;
    cin >> s;
    ll N = 0;
    for (char el : s) N = N*10+(el-'0');
    if (N < 100) {
        for (int w = 2; w < N; w++) {
            int h = N/w+(N%w!=0);
            vector<vi> board(h, vi(w));
            for (int j = 0; j < N; j++) {
                board[j/w][j%w] = j+1;
            }
            assert(board[(N-1)/w][(N-1)%w] == N);
            queue<pii> que;
            memset(done, false, sizeof(done));
            que.push(pii(0, 0));
            done[0][0] = true;
            while (!que.empty()) {
                pii p = que.front(); que.pop();
                for (int k = 0; k < 4; k++) {
                    int ny = p.first+dy[k];
                    int nx = p.second+dx[k];
                    if (ny < 0 || ny >= h || nx < 0 || nx >= w || board[ny][nx] == 0 || prime[board[ny][nx]]) continue;
                    if (done[ny][nx]) continue;
                    done[ny][nx] = true;
                    que.push(pii(ny, nx));
                }
            }
            if (done[(N-1)/w][(N-1)%w]) {
                cout << w << endl;
                break;
            }
        }
    } else {
        if (N%8 == 1 && isPrime(N-8)) cout << 14 << endl;
        else cout << 8 << endl;
    }
    return 0;
}

問題のコンセプトは好きだけど制約が辛すぎ…