結果

問題 No.2746 Bicolor Pyramid
コンテスト
ユーザー tassei903
提出日時 2024-04-17 03:57:13
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 932 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 956 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 65,536 KB
最終ジャッジ日時 2024-10-09 09:25:35
合計ジャッジ時間 2,853 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 23 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

R, B = map(int, input().split())

if R > B:
    R, B = B, R
def f(R, B):
    dp = [0 for _ in range(R + 1)]
    dp[0] = 1
    k = 1
    z = 0
    while True:
        flag = False
        ndp = [0 for _ in range(R + 1)]
        dp, ndp = ndp, dp
        for i in range(R + 1):
            if ndp[i] == 0:
                continue
            j = z - i
            if i + k**2 <= R and j <= B:
                dp[i + k**2] = 1
                flag = True
            if i <= R and j + k**2 <= B:
                dp[i] = 1
                flag = True
        if flag:
            z += k ** 2
            k += 1
        else:
            break

    return k - 1

def h(x):
    ok = 0
    ng = 10**18
    while ng - ok > 1:
        mid = (ok + ng) // 2
        if mid * (mid + 1) * (2 * mid + 1) // 6 <= x:
            ok = mid
        else:
            ng = mid
    return ok

if B <= 10000:
    print(f(R, B))
else:
    print(h(R + B))
0