結果

問題 No.2746 Bicolor Pyramid
コンテスト
ユーザー LyricalMaestro
提出日時 2025-10-29 23:32:23
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 874 bytes
コンパイル時間 306 ms
コンパイル使用メモリ 82,640 KB
実行使用メモリ 76,524 KB
最終ジャッジ日時 2025-10-29 23:33:02
合計ジャッジ時間 36,555 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 TLE * 1
other AC * 23 WA * 9 TLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

## https://yukicoder.me/problems/no/3312

from collections import deque

def solve(R, B, value):
    array = [B, R]
    array.sort(reverse=True)

    for i in reversed(range(1, value + 1)):
        if array[0] >= i ** 2:
            array[0] -= i ** 2
        else:
            return False
        array.sort(reverse=True)
    return True


def main():
    R, B = map(int, input().split())

    def f(k):
        y = k * (k + 1) * (2 * k + 1)
        return y // 6

    x = R + B
    k = 1
    while f(k) <= x:
        k += 1
    max_k = k 

    low = 0
    high = max_k
    while high - low > 1:
        mid = (high + low) // 2
        if solve(R, B, mid):
            low = mid
        else:
            high = mid
    if solve(R, B, high):
        print(high)
    else:
        print(low)  

    

    
    






                


if __name__ == "__main__":
    main()
0