Submission #12122679

Source Code Expand

import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
import numpy as np

N = int(readline())
A = np.array(read().split(), np.int64)
I = np.argsort(A)[::-1].tolist()
A = A[I].tolist()

dp = np.zeros(1, np.int64)

for n, (i, x) in enumerate(zip(I, A)):
    newdp = np.zeros(len(dp) + 1, np.int64)
    left_ind = np.arange(len(dp))
    right_ind = N - 1 - n + left_ind
    left_cost = x * np.abs(left_ind - i)
    right_cost = x * np.abs(right_ind - i)
    np.maximum(newdp[:-1], dp + right_cost, out=newdp[:-1])
    np.maximum(newdp[1:], dp + left_cost, out=newdp[1:])
    dp = newdp

print(dp.max())

Submission Info

Submission Time
Task E - Active Infants
User maspy
Language Python (3.8.2)
Score 500
Code Size 686 Byte
Status AC
Exec Time 141 ms
Memory 27992 KiB

Judge Result

Set Name Sample FULL
Score / Max Score 0 / 0 500 / 500
Status
AC × 3
AC × 26
Set Name Test Cases
Sample Sample_01.txt, Sample_02.txt, Sample_03.txt
FULL Sample_01.txt, Sample_02.txt, Sample_03.txt, maxhand_01.txt, maxhand_02.txt, maxhand_03.txt, maxhand_04.txt, maxhand_05.txt, maxrand_01.txt, maxrand_02.txt, maxrand_03.txt, maxrand_04.txt, maxrand_05.txt, minhand_01.txt, minhand_02.txt, minhand_03.txt, minhand_04.txt, minrand_01.txt, minrand_02.txt, minrand_03.txt, minrand_04.txt, ni_01.txt, rand_01.txt, rand_02.txt, rand_03.txt, rand_04.txt
Case Name Status Exec Time Memory
Sample_01.txt AC 100 ms 27048 KiB
Sample_02.txt AC 98 ms 27296 KiB
Sample_03.txt AC 99 ms 27152 KiB
maxhand_01.txt AC 135 ms 27844 KiB
maxhand_02.txt AC 134 ms 27796 KiB
maxhand_03.txt AC 134 ms 27988 KiB
maxhand_04.txt AC 137 ms 27772 KiB
maxhand_05.txt AC 135 ms 27784 KiB
maxrand_01.txt AC 134 ms 27736 KiB
maxrand_02.txt AC 134 ms 27692 KiB
maxrand_03.txt AC 138 ms 27888 KiB
maxrand_04.txt AC 136 ms 27992 KiB
maxrand_05.txt AC 141 ms 27784 KiB
minhand_01.txt AC 101 ms 27072 KiB
minhand_02.txt AC 100 ms 27040 KiB
minhand_03.txt AC 98 ms 27056 KiB
minhand_04.txt AC 101 ms 27016 KiB
minrand_01.txt AC 97 ms 27276 KiB
minrand_02.txt AC 99 ms 27116 KiB
minrand_03.txt AC 102 ms 27232 KiB
minrand_04.txt AC 101 ms 27260 KiB
ni_01.txt AC 99 ms 27004 KiB
rand_01.txt AC 114 ms 27668 KiB
rand_02.txt AC 136 ms 27760 KiB
rand_03.txt AC 125 ms 27704 KiB
rand_04.txt AC 108 ms 27592 KiB