結果

問題 No.12 限定された素数
ユーザー PachicobuePachicobue
提出日時 2017-05-29 04:23:48
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 106 ms / 5,000 ms
コード長 3,309 bytes
コンパイル時間 1,937 ms
コンパイル使用メモリ 175,672 KB
実行使用メモリ 54,896 KB
最終ジャッジ日時 2024-05-03 10:46:48
合計ジャッジ時間 5,196 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 95 ms
53,452 KB
testcase_01 AC 98 ms
53,128 KB
testcase_02 AC 96 ms
53,128 KB
testcase_03 AC 102 ms
52,836 KB
testcase_04 AC 98 ms
53,812 KB
testcase_05 AC 98 ms
52,272 KB
testcase_06 AC 96 ms
53,172 KB
testcase_07 AC 97 ms
52,088 KB
testcase_08 AC 98 ms
51,976 KB
testcase_09 AC 98 ms
52,376 KB
testcase_10 AC 104 ms
53,212 KB
testcase_11 AC 105 ms
54,896 KB
testcase_12 AC 106 ms
54,320 KB
testcase_13 AC 100 ms
52,512 KB
testcase_14 AC 98 ms
52,248 KB
testcase_15 AC 98 ms
52,568 KB
testcase_16 AC 101 ms
54,812 KB
testcase_17 AC 98 ms
52,660 KB
testcase_18 AC 101 ms
52,220 KB
testcase_19 AC 97 ms
52,752 KB
testcase_20 AC 97 ms
52,080 KB
testcase_21 AC 98 ms
52,164 KB
testcase_22 AC 100 ms
53,484 KB
testcase_23 AC 97 ms
53,452 KB
testcase_24 AC 94 ms
54,644 KB
testcase_25 AC 97 ms
52,072 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#define FOR(i, a, b) for (ll i = (a); i < (b); i++)
#define RFOR(i, a, b) for (ll i = (b)-1; i >= (a); i--)
#define rep(i, n) for (ll i = 0; i < (n); i++)
#define rep1(i, n) for (ll i = 1; i <= (n); i++)
#define rrep(i, n) for (ll i = (n)-1; i >= 0; i--)

#define pb push_back
#define mp make_pair
#define fst first
#define snd second
#define show(x) cout << #x << " = " << x << endl
#define chmin(x, y) x = min(x, y)
#define chmax(x, y) x = max(x, y)
#define pii pair<int, int>
#define vi vector<int>

using namespace std;
template <class S, class T>
ostream& operator<<(ostream& o, const pair<S, T>& p)
{
    return o << "(" << p.first << "," << p.second << ")";
}
template <class T>
ostream& operator<<(ostream& o, const vector<T>& vc)
{
    o << "sz = " << vc.size() << endl
      << "[";
    for (const T& v : vc)
        o << v << ",";
    o << "]";
    return o;
}
using ll = long long;
constexpr ll MOD = 1000000007;

constexpr ll MAX = 5000000;
bool isprime[MAX + 1];
vector<vector<ll>> digits;
vector<ll> prime;
bool allowed[10];

bool toomuch(ll head, ll tail)
{
    rep(i, 10)
    {
        if (not allowed[i]) {
            if (digits[tail][i] - digits[head - 1][i] > 0) {
                return true;
            }
        }
    }
    return false;
}

bool shortage(ll head, ll tail)
{
    rep(i, 10)
    {
        if (allowed[i]) {
            if (digits[tail][i] - digits[head - 1][i] <= 0) {
                return true;
            }
        }
    }
    return false;
}

int main()
{
    for (ll i = 2; i <= MAX; i++) {
        isprime[i] = true;
    }

    for (ll i = 2; i <= MAX; i++) {
        if (isprime[i]) {
            for (ll j = 2; i * j <= MAX; j++) {
                isprime[i * j] = false;
            }
        }
    }

    int n;
    cin >> n;
    rep(i, n)
    {
        int a;
        cin >> a;
        allowed[a] = true;
    }


    digits.resize(1);
    digits[0].resize(10);
    rep(i, 10)
    {
        digits[0][i] = 0;
    }
    prime.pb(0);
    for (ll i = 2; i <= MAX; i++) {
        if (isprime[i]) {
            prime.pb(i);
        }
    }
    const ll pnum = prime.size();
    for (ll i = 1; i < pnum; i++) {
        digits.pb(vector<ll>(10));
        bool digit[10];
        rep(j, 10)
        {
            digit[j] = 0;
        }

        ll p = prime[i];
        while (p > 0) {
            digit[p % 10] = true;
            p /= 10;
        }
        rep(j, 10)
        {
            if (digit[j]) {
                digits[i][j] = digits[i - 1][j] + 1;
            } else {
                digits[i][j] = digits[i - 1][j];
            }
        }
    }

    ll maxint = numeric_limits<ll>::min();
    ll head = 1;
    ll tail = head;
    while (tail < pnum) {
        if (tail < head) {
            tail++;
            continue;
        }
        if (toomuch(head, tail)) {
            head++;
            continue;
        }
        if (not shortage(head, tail)) {
            maxint = max(maxint, prime[tail + 1] - prime[head - 1] - 2);
            if (tail == pnum - 1) {
                maxint = max(maxint, MAX - prime[head - 1] - 1);
            }
        }
        tail++;
    }

    if (maxint == numeric_limits<ll>::min()) {
        cout << -1 << endl;
    } else {
        cout << maxint << endl;
    }

    return 0;
}
0