結果
| 問題 | 
                            No.1659 Product of Divisors
                             | 
                    
| コンテスト | |
| ユーザー | 
                             AEn
                         | 
                    
| 提出日時 | 2022-12-03 17:47:00 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 761 bytes | 
| コンパイル時間 | 222 ms | 
| コンパイル使用メモリ | 82,384 KB | 
| 実行使用メモリ | 59,168 KB | 
| 最終ジャッジ日時 | 2024-10-10 20:27:25 | 
| 合計ジャッジ時間 | 2,246 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge2 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 2 | 
| other | AC * 21 WA * 2 | 
ソースコード
def factorization(n):
    arr = []
    temp = n
    for i in range(2, int(-(-n**0.5//1))+1):
        if temp%i==0:
            cnt=0
            while temp%i==0:
                cnt+=1
                temp //= i
            arr.append([i, cnt])
    if temp!=1:
        arr.append([temp, 1])
    if arr==[]:
        arr.append([n, 1])
    return arr
N, K = map(int, input().split())
mod = 10**9+7
mem = {}
arr = factorization(N)
res = 1
for num, cnt in arr:
    if cnt in mem:
        res *= mem[cnt]
        res %= mod
    else:
        cal = 1
        for i in range(cnt):
            cal *= cnt+K-i
            cal %= mod
            cal *= pow(i+1,mod-2,mod)
            cal %= mod
        mem[cnt] = cal
        res *= cal
        res %= mod
print(res)
            
            
            
        
            
AEn