結果
問題 | No.2746 Bicolor Pyramid |
ユーザー |
![]() |
提出日時 | 2024-04-21 06:57:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 124 ms / 2,000 ms |
コード長 | 1,358 bytes |
コンパイル時間 | 194 ms |
コンパイル使用メモリ | 82,224 KB |
実行使用メモリ | 63,312 KB |
最終ジャッジ日時 | 2024-10-13 05:41:59 |
合計ジャッジ時間 | 4,113 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
def var_calc(n): var = 0 for i in range(1, n + 1): var += i * i return var def shirabe(n): var = 0 for i in range(1, n + 1): var += i * i a = [0] * (var + 1) a[0] = 1 for i in range(1, n + 1): b = a[::] for j in range(var + 1): if i * i + j <= var and a[j]: b[i * i + j] = 1 a = b tar = [] for i in range(var + 1): if a[i] == 0: tar.append(i) return tar # 1 + 4 + 9 + 16 + 25 + 36 umekomi = [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()) ub = 2 * 10 ** 6 lb = 1 while ub - lb > 1: t = (ub + lb) // 2 av = var_calc(t) if r + b < av: ub = t continue if t <= 16: g = [i * i for i in range(1, t + 1)] mode = 0 for i in range(1 << t): at = 0 bt = 0 for j in range(t): if i >> j & 1: at += g[j] else: bt += g[j] if at <= r and bt <= b: mode = 1 break if mode: lb = t else: ub = t else: norep = set(umekomi) for i in umekomi: norep.add(av - i) if r >= av or b >= av: lb = t continue good = 0 for i in range(0, min(r+1, 10000)): if r - i in norep: continue if av - (r - i) > b: continue if av - (r - i) < 0: continue if av - (r - i) in norep: continue good = 1 break if good: lb = t else: ub = t print(lb)