結果

問題 No.864 四方演算
ユーザー kawacchu
提出日時 2019-08-16 22:37:24
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 873 bytes
コンパイル時間 325 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 17,568 KB
最終ジャッジ日時 2024-09-22 18:47:15
合計ジャッジ時間 2,729 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 3
other AC * 1 TLE * 1 -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

def trial_division(n):
    a = []  
    while n % 2 == 0:
        a.append(2)
        n //= 2
    f = 3
    while f * f <= n:
        if n % f == 0:
            a.append(f)
            n //= f
        else:
            f += 2   
    if n != 1: a.append(n)
    return a

N = int(input())
K = int(input())

T = trial_division(K)
D = {0:1}
ans = 0

for bit in range(1 << len(T)) :
    A = 1
    B = 1
    for k in range(len(T)) :
        if bit & (1 << k) :
            A *= T[k]
        else :
            B *= T[k]
    if A in D : continue
    D[A] = 1
    cnt1 = 0
    cnt2 = 0

    if A <= N :
        cnt1 = A-1
    elif N < A <= 2*N :
        cnt1 = N-(A-N)+1
    else :
        cnt1 = 0
    
    if B <= N :
        cnt2 = B-1
    elif N < B <= 2*N :
        cnt2 = N-(B-N)+1
    else :
        cnt2 = 0
    #print(A,B,cnt1,cnt2)
    ans += (cnt1*cnt2)
    
print(ans)
0