結果

問題 No.816 Beautiful tuples
コンテスト
ユーザー roaris
提出日時 2019-07-25 20:24:06
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 778 bytes
コンパイル時間 95 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-07-02 06:17:24
合計ジャッジ時間 1,295 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 11 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

def factorize(n):
    fct = []
    div = 2
    
    while div ** 2 <= n:
        cnt = 0
        while n % div == 0:
            n //= div
            cnt += 1
        if cnt > 0:
            fct.append((div, cnt))
        div += 1
    
    if n > 1:
        fct.append((n, 1))
    
    return fct

def dfs(fct, depth, cur):
    div, cnt = fct[depth]
    
    if depth == len(fct) - 1:
        return [cur*div**i for i in range(cnt+1)]
        
    rseq = []
    
    for i in range(cnt+1):
        rseq += dfs(fct, depth+1, cur*div**i)
        
    return rseq
    
    
A, B = map(int, input().split())
fct = factorize(A+B)
C_cand = list(sorted(dfs(fct, 0, 1)))

for C in C_cand:
    if (B + C) % A == 0 and (C + A) % B == 0:
        print(C)
        break
else:
    print(-1)
0