結果

問題 No.3048 Swing
ユーザー 寝癖
提出日時 2025-03-07 21:35:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 40 ms / 2,000 ms
コード長 850 bytes
コンパイル時間 217 ms
コンパイル使用メモリ 82,148 KB
実行使用メモリ 53,876 KB
最終ジャッジ日時 2025-06-20 02:24:22
合計ジャッジ時間 3,991 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff #

x, n = map(int, input().split())

if x >= 0:
    ok = 1<<31
    ng = -1
    while ok - ng > 1:
        mid = (ok + ng) // 2
        if 2 * x <= mid * (mid + 1):
            ok = mid
        else:
            ng = mid
    k = ok
    if n < k:
        print(x - n * (n + 1) // 2)
    else:
        t = x - k * (k + 1) // 2
        if (n - k) % 2 == 0:
            print(t + (k - n) // 2)
        else:
            print(t + (n + k + 1) // 2)
else:
    ok = 1<<31
    ng = -1
    while ok - ng > 1:
        mid = (ok + ng) // 2
        if -2 * x < mid * (mid + 1):
            ok = mid
        else:
            ng = mid
    k = ok
    if n < k:
        print(x + n * (n + 1) // 2)
    else:
        t = x + k * (k + 1) // 2
        if (n - k) % 2 == 0:
            print(t + (n - k) // 2)
        else:
            print(t - (k + 1) - (n - k - 1) // 2)
0