結果

問題 No.2420 Simple Problem
ユーザー iseeisee
提出日時 2023-08-12 14:12:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 716 ms / 2,000 ms
コード長 1,294 bytes
コンパイル時間 356 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 82,432 KB
最終ジャッジ日時 2024-11-19 17:52:28
合計ジャッジ時間 21,187 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
53,632 KB
testcase_01 AC 343 ms
78,464 KB
testcase_02 AC 67 ms
70,400 KB
testcase_03 AC 176 ms
77,184 KB
testcase_04 AC 558 ms
80,896 KB
testcase_05 AC 341 ms
79,232 KB
testcase_06 AC 675 ms
81,792 KB
testcase_07 AC 574 ms
81,280 KB
testcase_08 AC 625 ms
81,792 KB
testcase_09 AC 629 ms
81,664 KB
testcase_10 AC 685 ms
82,432 KB
testcase_11 AC 694 ms
82,432 KB
testcase_12 AC 593 ms
81,920 KB
testcase_13 AC 687 ms
81,792 KB
testcase_14 AC 632 ms
81,408 KB
testcase_15 AC 679 ms
81,920 KB
testcase_16 AC 586 ms
81,152 KB
testcase_17 AC 624 ms
81,664 KB
testcase_18 AC 657 ms
81,536 KB
testcase_19 AC 625 ms
81,664 KB
testcase_20 AC 597 ms
81,536 KB
testcase_21 AC 587 ms
81,280 KB
testcase_22 AC 581 ms
81,408 KB
testcase_23 AC 626 ms
81,664 KB
testcase_24 AC 716 ms
81,920 KB
testcase_25 AC 595 ms
81,536 KB
testcase_26 AC 53 ms
60,800 KB
testcase_27 AC 510 ms
81,024 KB
testcase_28 AC 510 ms
81,280 KB
testcase_29 AC 501 ms
80,512 KB
testcase_30 AC 512 ms
80,896 KB
testcase_31 AC 504 ms
80,640 KB
testcase_32 AC 41 ms
53,888 KB
testcase_33 AC 300 ms
78,336 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda: sys.stdin.readline().rstrip()

def naive(A, B):
    ng = -1
    ok = (A+B)*2
    while abs(ng-ok) > 1:
        mid = (ok+ng)//2
        if mid*mid-A-B > 0 and 4*A*B < (mid*mid-A-B)**2:  # ここ頑張る
            ok = mid
        else:
            ng = mid
    return ok

def isqrt(n):
    # n <= X^2 なる最小の X を求める
    ng = 0
    ok = n*2
    while abs(ng-ok) > 1:
        mid = (ok+ng)//2
        if n <= mid*mid : # ここ頑張る
            ok = mid
        else:
            ng = mid
    return ok

def solve(A, B):
    # vA + vB < X
    # A + 2vAB + B < X2
    # 2vAB < X2 - A - B =: C
    # 4AB < C2
    iseq = True
    C = isqrt(4*A*B)
    iseq &= 4*A*B == C * C
    X = isqrt(C+A+B)
    iseq &= C+A+B == X * X
    return X+iseq

def main():
    # 入力
    N = int(input())
    for _ in range(N):
        A, B = map(int, input().split())
        # 計算・出力
        ans = solve(A, B)
        print(ans)

if __name__ == "__main__":
    main()
    from random import randint
    # for _ in range(10000):
    #     A = randint(1, 10**9)
    #     B = randint(1, 10**9)
    #     ans1 = naive(A, B)
    #     ans2 = solve(A, B)
    #     if ans1 != ans2:
    #         print(A, B)
    #         print(ans1, ans2)
    #         break
0