結果
| 問題 | No.837 Noelちゃんと星々2 |
| コンテスト | |
| ユーザー |
tails1434
|
| 提出日時 | 2020-01-07 21:25:52 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 128 ms / 2,000 ms |
| コード長 | 724 bytes |
| コンパイル時間 | 176 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 127,376 KB |
| 最終ジャッジ日時 | 2024-06-11 14:27:27 |
| 合計ジャッジ時間 | 3,425 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 29 |
ソースコード
from itertools import accumulate
def sum_dist(A, l, r):
if (r - l) % 2 == 0:
right = A[r] - A[(l+r)//2]
left = A[(l+r)//2] - A[l]
return - left + right
else:
right = A[r] - A[(l+r)//2 + 1]
left = A[(l+r)//2] - A[l]
return - left + right
def main():
N = int(input())
Y = list(map(int, input().split()))
sort_Y = sorted(Y)
if len(set(Y)) == 1:
print(1)
exit()
elif len(set(Y)) == 2:
print(0)
exit()
A = [0] + list(accumulate(sort_Y))
ans = float('inf')
for i in range(N):
tmp = sum_dist(A,0,i) + sum_dist(A,i,N)
ans = min(ans, tmp)
print(ans)
if __name__ == "__main__":
main()
tails1434