結果
問題 | No.77 レンガのピラミッド |
ユーザー |
![]() |
提出日時 | 2020-06-03 15:58:47 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 35 ms / 5,000 ms |
コード長 | 1,326 bytes |
コンパイル時間 | 118 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 11,136 KB |
最終ジャッジ日時 | 2024-11-25 02:52:57 |
合計ジャッジ時間 | 1,859 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 20 |
ソースコード
import sysdef input(): return sys.stdin.readline().strip()def list2d(a, b, c): return [[c] * b for i in range(a)]def list3d(a, b, c, d): return [[[d] * c for j in range(b)] for i in range(a)]def list4d(a, b, c, d, e): return [[[[e] * d for j in range(c)] for j in range(b)] for i in range(a)]def ceil(x, y=1): return int(-(-x // y))def INT(): return int(input())def MAP(): return map(int, input().split())def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)]def Yes(): print('Yes')def No(): print('No')def YES(): print('YES')def NO(): print('NO')sys.setrecursionlimit(10 ** 9)INF = 10 ** 19MOD = 10 ** 9 + 7EPS = 10 ** -10N = INT()A = LIST()sm = sum(A)ans = INFi = 1while 1:B1 = []for j in range(1, i+1):B1.append(j)B2 = []for j in range(i-1, 0, -1):B2.append(j)B = B1 + B2M = len(B)total = sum(B)if total > sm:breakcnt1 = cnt2 = 0for j in range(max(N, M)):if j >= N:cnt1 += B[j]elif j >= M:cnt2 += A[j]else:if A[j] < B[j]:cnt1 += B[j] - A[j]elif A[j] > B[j]:cnt2 += A[j] - B[j]move = min(cnt1, cnt2)add = abs(cnt1 - cnt2)ans = min(ans, move+add)i += 1print(ans)