結果

問題 No.2746 Bicolor Pyramid
コンテスト
ユーザー tassei903
提出日時 2024-04-17 04:05:07
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 930 bytes
コンパイル時間 354 ms
コンパイル使用メモリ 82,376 KB
実行使用メモリ 75,880 KB
最終ジャッジ日時 2024-10-09 09:25:30
合計ジャッジ時間 18,040 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 34 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

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
R, B = map(int, input().split())

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