結果

問題 No.28 末尾最適化
ユーザー maine_honzukimaine_honzuki
提出日時 2020-05-05 21:46:33
言語 C++14
(gcc 10.2.0 + boost 1.73.0)
結果
AC  
実行時間 304 ms / 5,000 ms
コード長 1,445 Byte
コンパイル時間 1,590 ms
使用メモリ 3,440 KB
最終ジャッジ日時 2021-04-11 08:57:04
合計ジャッジ時間 2,800 ms
ジャッジサーバーID
(参考情報)
judge9 / judge6
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 2 ms
3,304 KB
testcase_01 AC 304 ms
3,440 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int MOD = 100000009;

vector<pair<int, int>> prime_factorize(int n) {
    vector<pair<int, int>> ret;
    for (int i = 2; i * i <= n; i++) {
        if (n % i == 0) {
            int num = 0;
            while (n % i == 0) {
                num++;
                n /= i;
            }
            ret.push_back({i, num});
        }
    }
    if (n != 1)
        ret.push_back({n, 1});
    return ret;
}

int main() {
    int Q;
    cin >> Q;
    while (Q--) {
        ll seed, N, K, B;
        cin >> seed >> N >> K >> B;
        vector<ll> X(N + 1);
        X[0] = seed;
        for (int i = 1; i <= N; i++) {
            X[i] = 1 + ((X[i - 1] * X[i - 1] + X[i - 1] * 12345) % MOD);
        }
        int ans = 1e9;
        auto pf = prime_factorize(B);
        for (auto& p : pf) {
            vector<int> cnt(N + 1);
            int prime = p.first;
            for (int i = 0; i <= N; i++) {
                int n = 0;
                int tmp = prime;
                while (X[i] && X[i] % tmp == 0) {
                    tmp *= prime;
                    n++;
                }
                cnt[i] = n;
            }
            sort(cnt.begin(), cnt.end());
            int sum = 0;
            for (int i = 0; i < K; i++) {
                sum += cnt[i];
            }
            ans = min(ans, sum / p.second);
        }
        cout << ans << endl;
    }
}
0