結果

問題 No.77 レンガのピラミッド
コンテスト
ユーザー cologne
提出日時 2022-01-25 22:09:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 51 ms / 5,000 ms
コード長 640 bytes
コンパイル時間 348 ms
コンパイル使用メモリ 81,968 KB
実行使用メモリ 62,324 KB
最終ジャッジ日時 2024-12-17 18:25:24
合計ジャッジ時間 2,281 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys


def input(): return sys.stdin.readline().rstrip()


def main():
    N = int(input())
    *A, = map(int, input().split())

    def pyramid(M):
        return list(range(1, M)) + list(range(M, 0, -1))

    def dist(a, b):
        u, d = 0, 0
        for i in range(max(len(a), len(b))):
            A = a[i] if i < len(a) else 0
            B = b[i] if i < len(b) else 0
            if A > B:
                u += A-B
            else:
                d += B-A
        return max(u, d) if u <= d else float('inf')

    ans = min(dist(pyramid(i), A) for i in range(1, N+1))

    print(ans)


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