Submission #14808048

---

Source Code Expand

import sys
import numpy as np

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

def get_input():
    D = int(readline())
    C = np.array(readline().split(), np.int32)
    S = np.array([readline().split() for _ in range(D)], np.int32)
    T = np.array(read().split(), np.int32)
    T -= 1
    return D, C, S, T

def print_scores(D, C, S, T):
    x = 0
    last = np.zeros(26, np.int32)
    for d in range(D):
        i = T[d]
        x += S[d, i]
        last[i] = d + 1
        for i in range(26):
            x -= np.sum(C[i] * (d + 1 - last[i]))
        print(x)

D, C, S, T = get_input()
print_scores(D, C, S, T)

Submission Info

Judge Result