結果

問題 No.28 末尾最適化
ユーザー szkhtsszkhts
提出日時 2023-01-04 07:52:45
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 3,337 ms / 5,000 ms
コード長 1,060 bytes
コンパイル時間 116 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,264 KB
最終ジャッジ日時 2024-05-05 13:08:31
合計ジャッジ時間 3,949 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

def is_prime(a):
    i = 2
    while i * i <= a:
        if a % i == 0:
            return False
        i += 1

    return True

def main():
    Q = int(input())
    for _ in range(Q):
        seed, N, K, B = map(int, input().split())
        N += 1
        X = [0] * N
        X[0] = seed
        for i in range(1, N):
            X[i] = 1 + (X[i - 1] * X[i - 1] + X[i - 1] * 12345) % 100000009

        ans = float('inf')
        for p in range(2, B + 1):
            if not is_prime(p):
                continue
            if B % p != 0:
                continue
            div = 0
            while B % p == 0:
                B = B // p
                div += 1

            ar = [0] * N
            for i in range(N):
                temp = X[i]
                while temp % p == 0:
                    ar[i] += 1
                    temp = temp // p

            ar.sort()

            mul = 0
            for i in range(K):
                mul += ar[i]
            ans = min(ans, mul // div)
        print(ans)

if __name__ == '__main__':
    main()
0