結果

問題 No.308 素数は通れません
ユーザー mayoko_mayoko_
提出日時 2015-12-02 21:21:28
言語 C++11
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 2,835 bytes
コンパイル時間 1,035 ms
コンパイル使用メモリ 98,612 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-14 08:03:06
合計ジャッジ時間 3,638 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 99 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
//#include<cctype>
#include<climits>
#include<iostream>
#include<string>
#include<vector>
#include<map>
//#include<list>
#include<queue>
#include<deque>
#include<algorithm>
//#include<numeric>
#include<utility>
#include<complex>
//#include<memory>
#include<functional>
#include<cassert>
#include<set>
#include<stack>

const int dx[] = {1, 0, -1, 0};
const int dy[] = {0, 1, 0, -1};
using namespace std;
typedef __int128 ll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef pair<int, int> pii;

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

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 a*pow_mod(a, p-1, mod)%mod;
    ll tmp = pow_mod(a, p/2, mod);
    return 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 = (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;
}
0