結果

問題 No.3532 Non-Fourth-Power Sets
コンテスト
ユーザー GaLLium
提出日時 2026-03-12 14:15:34
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 3,374 ms / 6,000 ms
コード長 877 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 794 ms
コンパイル使用メモリ 85,120 KB
実行使用メモリ 354,428 KB
最終ジャッジ日時 2026-05-04 21:19:11
合計ジャッジ時間 64,400 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
from array import array
def mul(a,b):
  return (a*b)%1000000007

mpf = array('I',[0])*20000005
primes = array('I')
for i in range(2, len(mpf)):
    if mpf[i] == 0:
        mpf[i] = i
        primes.append(i)
    for p in primes:
        if p > mpf[i] or p * i >= len(mpf):
            break
        mpf[p * i] = p


f = array('I',[0])*48000005
f[0] = 1
f[1] = 2
f[2] = 5
f[3] = 15
for n in range(4, len(f)):
    n1 = (n+3)//4
    n2 = (n+2)//4
    n3 = (n+1)//4
    f[n] = f[n-1] + mul(n1+n2+n3,f[n-2]) + mul(n2*n2+n2*n3+n3*n3,f[n-3]) + mul(mul(n3,n3),mul(n3,f[n-4]))
    f[n] %= 1000000007

T = int(sys.stdin.readline())
for _ in range(T):
    A, B = map(int, sys.stdin.readline().split())
    ans = 1
    while A > 1:
        p = mpf[A]
        vp = 0
        while mpf[A] == p:
            A //= p
            vp += 1
        ans = mul(ans,f[B*vp])
    print(ans)
0