結果

問題 No.2746 Bicolor Pyramid
コンテスト
ユーザー Kude
提出日時 2024-04-21 13:53:21
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 41 ms / 2,000 ms
コード長 812 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 52,352 KB
最終ジャッジ日時 2024-10-13 12:34:19
合計ジャッジ時間 2,728 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

bad = {2,3,6,7,8,11,12,15,18,19,22,23,24,27,28,31,32,33,43,44,47,48,60,67,72,76,92,96,108,112,128}
R, B = map(int, input().split())
if R > B:
    R, B = B, R

def check(x):
    tot = x * (x + 1) * (2 * x + 1) // 6
    if tot > R + B:
        return False
    r = max(0, tot - B)
    b = tot - r
    if x >= 11:
        while r <= min(b, R):
            if r not in bad:
                return True
            r += 1
            b -= 1
    else:
        dp = 1
        for i in range(1, x + 1):
            dp |= dp << i * i
        while r <= min(b, R):
            if dp >> r & 1:
                return True
            r += 1
            b -= 1
    return False

r = 1
while check(r):
    r *= 2
l = r // 2
while r - l > 1:
    c = (l + r) // 2
    if check(c):
        l = c
    else:
        r = c
print(l)
0