結果

問題 No.2078 魔抜けなパープル
ユーザー ntudantuda
提出日時 2022-10-12 21:35:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 128 ms / 2,000 ms
コード長 616 bytes
コンパイル時間 575 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 67,840 KB
最終ジャッジ日時 2024-06-26 11:29:56
合計ジャッジ時間 2,124 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

T = int(input())
for _ in range(T):
    X, A = map(int, input().split())
    ans = A * A + X
    for i in range(1, max(1, A // 2) + 1):
        b = A % i
        a = (A - b * (i + 1)) // i
        if a >= 0 and b >= 0 and a * i + b * (i + 1) == A:
            ans = min(ans, a * (i * i + X) + b * ((i + 1) * (i + 1) + X))
            # print(i,a,b,a*i+b*(i+1),ans)
        a = -A % (i + 1)
        b = (A - a * i) // (i + 1)
        if a >= 0 and b >= 0 and a * i + b * (i + 1) == A:
            ans = min(ans, a * (i * i + X) + b * ((i + 1) * (i + 1) + X))
            # print(i,a,b,a*i+b*(i+1),ans)
    print(ans)
0