結果
問題 |
No.77 レンガのピラミッド
|
ユーザー |
|
提出日時 | 2017-03-23 07:01:09 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 482 bytes |
コンパイル時間 | 97 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 10,752 KB |
最終ジャッジ日時 | 2024-07-05 20:22:40 |
合計ジャッジ時間 | 1,519 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 3 WA * 17 |
ソースコード
def calc_nb(h): return h**2 N = int(input()) A = [int(i) for i in input().split()] num_b = sum(A) btm = 0 top = 1000 while top - btm > 1: mid = (top + btm) // 2 if num_b >= calc_nb(mid): btm = mid else: top = mid h = min((N + 1) // 2, btm) trash = num_b - calc_nb(h) ans = trash for i in range(h - 1): ans += max(0, (i + 1) - A[i]) for i in range(h - 1): ans += max(0, (i + 1) - A[2*h - 2 - i]) ans += max(0, h - A[h - 1]) print(ans)