ARC074 D - 3N Numbers - 競プロ

ARC074 D - 3N Numbers

https://atcoder.jp/contests/arc074/tasks/arc074_b

提出

code: python

n = int(input())

a = list(map(int, input().split()))

a_sum = [a0]

for i in range(1, len(a)):

a_sum.append(a_sum-1 + ai)

print(a_sum)

解答

code: python

import heapq

n = int(input())

a = list(map(int, input().split()))

# 3, 1, 4, 1, 5, 9

former, latter = a:n, [-1 * i for i in a2*n:]

# print(former, latter)

sum_former, sum_latter = sum(former), sum(latter)

results = 0 * (n+1)

results0 = sum_former

resultsn = sum_latter

heapq.heapify(former)

heapq.heapify(latter)

for i in range(n, 2*n):

v = ai

p = heapq.heappushpop(former, v)

sum_former += v - p

resultsi-n+1 += sum_former

for i in range(2*n-1, n-1, -1):

v = -1 * ai

p = heapq.heappushpop(latter, v)

sum_latter += v - p

resultsi-n += sum_latter

print(max(results))

テーマ

蟻本 2-4 Expedition

メモ

ABC 062 D：3N Numbers