Problem:


(Please read problem A first. The maximum score you can get by solving this problem B is 1, which will have almost no effect on your ranking.)

Beginner's Guide

Let's first write a program to calculate the score from a pair of input and output. You can know the total score by submitting your solution, or an official program to calculate a score is often provided for local evaluation as in this contest. Nevertheless, writing a score calculator by yourself is still useful to check your understanding of the problem specification. Moreover, the source code of the score calculator can often be reused for solving the problem or debugging your solution. So it is worthwhile to write a score calculator unless it is very complicated.

Problem Statement

You will be given a contest schedule for D days.
For each d=1,2,\ldots,D, calculate the satisfaction at the end of day d.

Input

Input is given from Standard Input in the form of the input of Problem A followed by the output of Problem A.

D
c_1 c_2 \cdots c_{26}
s_{1,1} s_{1,2} \cdots s_{1,26}
\vdots
s_{D,1} s_{D,2} \cdots s_{D,26}
t_1
t_2
\vdots
t_D

The constraints and generation methods for the input part are the same as those for Problem A.

For each d, t_d is an integer satisfying 1\leq t_d \leq 26, and your program is expected to work correctly for any value that meets the constraints.

Output

Let v_d be the satisfaction at the end of day d.
Print D integers v_d to Standard Output in the following format:

v_1
v_2
\vdots
v_D

Sample Input 1

5
86 90 69 51 2 96 71 47 88 34 45 46 89 34 31 38 97 84 41 80 14 4 50 83 7 82
19771 12979 18912 10432 10544 12928 13403 3047 10527 9740 8100 92 2856 14730 1396 15905 6534 4650 11469 3628 8433 2994 10899 16396 18355 11424
6674 17707 13855 16407 12232 2886 11908 1705 5000 1537 10440 10711 4917 10770 17272 15364 19277 18094 3929 3705 7169 6159 18683 15410 9092 4570
6878 4239 19925 1799 375 9563 3445 5658 19857 11401 6997 6498 19933 3848 2426 2146 19745 16880 17773 18359 3921 14172 16730 11157 5439 256
8633 15862 15303 10749 18499 7792 10317 5901 9395 11433 3514 3959 5202 19850 19469 9790 5653 784 18500 10552 17975 16615 7852 197 8471 7452
19855 17918 7990 10572 4333 438 9140 9104 12622 4985 12319 4028 19922 12132 16259 17476 2976 547 19195 19830 16285 4806 4471 9457 2864 2192
1
17
13
14
13

Sample Output 1

18398
35037
51140
65837
79325

Note that this example is a small one for checking the problem specification. It does not satisfy the constraint D=365 and is never actually given as a test case.

Next Step

We can build a solution (schedule) for this problem in the order of day 1, day 2, and so on. And for every partial solution we have built, we can calculate the goodness (satisfaction) by using the above score calculator. So we can construct the following algorithm: for each d=1,2,\ldots,D, we select the contest type that maximizes the satisfaction at the end of day d. You may have already encountered this kind of "greedy algorithms" in algorithm contests such as ABC. Greedy algorithms can guarantee the optimality for several problems, but unfortunately, it doesn't ensure optimality for this problem. However, even if it does not ensure optimality, we can still obtain a reasonable solution in many cases. Let's go back to Problem A and implement the greedy algorithm by utilizing the score calculator you just implemented!

Greedy methods can be applied to a variety of problems, are easy to implement, and often run relatively fast compared to other methods. Greedy is often the most powerful method when we need to process huge inputs.
We can further improve the score by changing the greedy selection criteria (evaluation function), keeping multiple candidates instead of focusing on one best partial solution (beam search), or using the output of greedy algorithms as an initial solution of other methods.
For more information, please refer to the editorial that will be published after the contest.



Sample input:
5
86 90 69 51 2 96 71 47 88 34 45 46 89 34 31 38 97 84 41 80 14 4 50 83 7 82
19771 12979 18912 10432 10544 12928 13403 3047 10527 9740 8100 92 2856 14730 1396 15905 6534 4650 11469 3628 8433 2994 10899 16396 18355 11424
6674 17707 13855 16407 12232 2886 11908 1705 5000 1537 10440 10711 4917 10770 17272 15364 19277 18094 3929 3705 7169 6159 18683 15410 9092 4570
6878 4239 19925 1799 375 9563 3445 5658 19857 11401 6997 6498 19933 3848 2426 2146 19745 16880 17773 18359 3921 14172 16730 11157 5439 256
8633 15862 15303 10749 18499 7792 10317 5901 9395 11433 3514 3959 5202 19850 19469 9790 5653 784 18500 10552 17975 16615 7852 197 8471 7452
19855 17918 7990 10572 4333 438 9140 9104 12622 4985 12319 4028 19922 12132 16259 17476 2976 547 19195 19830 16285 4806 4471 9457 2864 2192
1
17
13
14
13


Sample output:
18398
35037
51140
65837
79325

Submission status: Accepted

Solution code:
D = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for _ in range(D)]
t = [int(input()) for _ in range(D)]

last = [-1] * 26
S = 0
for d in range(D):
    S += s[d][t[d] - 1]

    last[t[d] - 1] = d
    for i in range(26):
        S -= c[i] * (d - last[i])

    print(S)

Submission status: Accepted

Solution code:
d=int(input())
c=list(map(int,input().split()))
s=[list(map(int,input().split())) for i in [0]*d]
t=[int(input()) for i in [0]*d]
k=0
n=1
p=[0]*26
for i in t:
    k+=s[n-1][i-1]
    p[i-1]=n
    k-=sum([(n-p[i])*c[i] for i in range(26)])
    print(k)
    n+=1

Submission status: Accepted

Solution code:
import sys
input=sys.stdin.readline
d=int(input())
c=list(map(int,input().split()))
s=[]
for i in range(d):
    sub=list(map(int,input().split()))
    s.append(sub)
t=[]
for i in range(d):
    t.append(int(input()))

h=0
last=[-1]*26

for i in range(d):
    h+=s[i][t[i]-1]
    last[t[i]-1]=i
    for j in range(26):
        h-=c[j]*(i-last[j])
    print(h)

Submission status: Accepted

Solution code:
D = int(input())
c = list(map(int, input().split()))
s = []
for i in range(D):
  S = list(map(int, input().split()))
  s.append(S)

ans = 0
blank = [0]*26
for i in range(D):
  t = int(input())
  ans += s[i][t-1]
  blank = [i+1 for i in blank]
  blank[t-1] = 0
  for j in range(26):
    ans -= blank[j] * c[j]
  print(ans)

Submission status: Accepted

Solution code:
from sys import stdin
from sys import setrecursionlimit
from collections import Counter, deque, defaultdict
from math import floor, ceil
from bisect import bisect_left
from itertools import combinations
setrecursionlimit(100000)

INF = int(1e10)
MOD = int(1e9 + 7)

def main():
    from builtins import int, map
    D = int(input())
    C = list(map(int, input().split()))
    S = [list(map(int, input().split())) for _ in range(D)]
    V = [int(input()) for _ in range(D)]
    last = [0] * 26
    score = 0
    for i in range(D):
        j = V[i] - 1
        last[j] = i + 1
        for k in range(26):
            score -= C[k] * (i + 1 - last[k])
        score += S[i][j]
        print(score)

if __name__ == '__main__':
    main()

Submission status: Accepted

Solution code:
def main():
    from sys import setrecursionlimit, stdin, stderr
    from os import environ
    from collections import defaultdict, deque, Counter
    from math import ceil, floor
    from itertools import accumulate, combinations, combinations_with_replacement
    setrecursionlimit(10**6)
    dbg = (lambda *something: stderr.write("\033[92m{}\033[0m".format(str(something)+'\n'))) if 'TERM_PROGRAM' in environ else lambda *x: 0
    input = lambda: stdin.readline().rstrip()
    LMIIS = lambda: list(map(int,input().split()))
    II = lambda: int(input())
    P = 10**9+7
    INF = 10**18+10

    D = II()
    C = LMIIS()
    S = []
    for i in range(D):
        S.append(LMIIS())
    T = [0]*D
    for i in range(D):
        T[i] = II()
    satisfaction = 0
    last_held = [-1] * 26
    for i,t in enumerate(T):
        last_held[t-1] = i
        for j,d in enumerate(last_held):
            satisfaction -= C[j] * (i-d)
        satisfaction += S[i][t-1]
        print(satisfaction)




    
main()

Submission status: Accepted

Solution code:
d=int(input())
c=list(map(int, input().split()))
s=[list(map(int, input().split())) for _ in range(d)]
t=[0]*26
ans=0
for i in range(1,d+1):
  tmp=int(input())
  t[tmp-1]=i
  ans+=s[i-1][tmp-1]
  for j in range(26):
    ans-=c[j]*(i-t[j])
  print(ans)

Submission status: Accepted

Solution code:
D = int(input())
C = [int(_) for _ in input().split()]
S = [[int(_) for _ in input().split()] for _ in range(D)]
T = [int(input()) - 1 for _ in range(D)]

ans = 0
last = [-1] * 26

for day, t in enumerate(T):
    ans += S[day][t]
    last[t] = day
    for t2, day2 in enumerate(last):
        ans -= C[t2] * (day - day2)
    print(ans)

Submission status: Accepted

Solution code:
D = int(input())
cl = list(map(int, input().split()))
sl = [list(map(int, input().split())) for _ in range(D)]
tl = [int(input()) for _ in range(D)]

def score(d, ti, last):
    s = sl[d-1][ti]
    for i in range(len(cl)):
        c = cl[i]
        s -= c*(d-last[i])
    return s

s = 0
last = [0]*26
for d in range(D):
    t = tl[d] - 1
    last[t] = d+1
    s += score(d+1,t,last)
    print(s)

Submission status: Accepted

Solution code:
from collections import defaultdict
D=int(input())
clist=list(map(int,input().split()))
smat=[[int(x) for x in input().split()] for _ in range(D)]
#print(clist)
#print(smat)

last_dic=defaultdict(int)
answer=0
for d in range(D):
  t=int(input())
  answer+=smat[d][t-1]
  last_dic[t]=d+1

  for i in range(26):
    answer-=clist[i]*(d+1-last_dic[i+1])
    
  print(answer)

Submission status: Accepted

Solution code:
import sys
input = lambda: sys.stdin.buffer.readline()

L = 26
d = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for _ in range(d)]

t = [int(input()) for _ in range(d)]

def evl(a):
    score = 0
    last = [-1] * L
    for i, x in enumerate(a):
        x -= 1
        score += s[i][x]
        last[x] = i
        for j, l in enumerate(last):
            score -= c[j] * (i - l)
        print(score)
    
    return score

evl(t)

Submission status: Accepted

Solution code:
import math
#import numpy as np
import queue
from collections import deque,defaultdict
import heapq as hpq
from sys import stdin,setrecursionlimit
#from scipy.sparse.csgraph import dijkstra
#from scipy.sparse import csr_matrix
ipt = stdin.readline
setrecursionlimit(10**7)

def main():
    d = int(ipt())
    c = [int(i) for i in ipt().split()]
    s = [[int(i) for i in ipt().split()] for _ in range(d)]
    t = [int(ipt())-1 for i in range(d)]
    ld = [-1]*26
    ans = 0
    for i,ti in enumerate(t):
        ans += s[i][ti]
        ld[ti] = i
        for j,ci in enumerate(c):
            ans -= ci*(i-ld[j])
        print(ans)

if __name__ == '__main__':
    main()

Submission status: Accepted

Solution code:
T = 26


def score(d, c, s, t):
    total = 0
    for i in range(d):
        total += s[i][t[i]]
    return max(10**6+total, 0)


def main():
    d = int(input())
    c = list(map(int, input().split()))
    s = [list(map(int, input().split())) for i in range(d)]
    t = [int(input()) for i in range(d)]
    last = [-1]*T
    total = 0
    for i in range(d):
        total += s[i][t[i]-1]
        last[t[i]-1] = i
        for j in range(T):
            total -= c[j] * (i - last[j])
        print(total)


if __name__ == "__main__":
    main()

Submission status: Accepted

Solution code:

D = int(input())
c = list(map(int,input().split()))
s = [list(map(int,input().split())) for i in range(D)]
t = [int(input()) for i in range(D)]

score = 0
last = [0]*26
for d in range(D):
    score += s[d][t[d]-1]
    last[t[d]-1] = d+1
    for i in range(26):
        score -= c[i]*(d+1-last[i])
    print(score)

Submission status: Accepted

Solution code:
import sys
sys.setrecursionlimit(300000)
from collections import defaultdict

def I(): return int(sys.stdin.readline())
def MI(): return map(int, sys.stdin.readline().split())
def MI0(): return map(lambda s: int(s) - 1, sys.stdin.readline().split())
def LMI(): return list(map(int, sys.stdin.readline().split()))
def LMI0(): return list(map(lambda s: int(s) - 1, sys.stdin.readline().split()))
MOD = 10 ** 9 + 7
INF = float('inf')


D = I()
C = LMI()
S = [LMI() for _ in range(D)]
T = [I() - 1 for _ in range(D)]

last = defaultdict(int)
ans = 0
for d in range(D):
    ans += S[d][T[d]]
    last[T[d]] = d + 1
    for i in range(26):
        ans -= C[i] * (d + 1 - last[i])
    print(ans)

Submission status: Accepted

Solution code:
# -*- coding: utf-8 -*-
import sys
sys.setrecursionlimit(10**9)
INF=10**18
MOD=10**9+7
input=lambda: sys.stdin.readline().rstrip()
YesNo=lambda b: bool([print('Yes')] if b else print('No'))
YESNO=lambda b: bool([print('YES')] if b else print('NO'))
int1=lambda x:int(x)-1

D=int(input())
c=tuple(map(int,input().split()))
s=[tuple(map(int,input().split())) for _ in range(D)]
last=[-1]*26

t=tuple([int1(input()) for _ in range(D)])
ans=0

for d,x in enumerate(t):
    ans+=s[d][x]
    last[x]=d
    for i in range(26):
        ans-=(d-last[i])*c[i]
    print(ans)

Submission status: Accepted

Solution code:
#!usr/bin/env python3
from collections import defaultdict, deque
from heapq import heappush, heappop
from itertools import permutations, accumulate
import sys
import math
import bisect
def LI(): return [int(x) for x in sys.stdin.readline().split()]
def I(): return int(sys.stdin.readline())
def LS():return [list(x) for x in sys.stdin.readline().split()]
def S():
    res = list(sys.stdin.readline())
    if res[-1] == "\n":
        return res[:-1]
    return res
def IR(n):
    return [I() for i in range(n)]
def LIR(n):
    return [LI() for i in range(n)]
def SR(n):
    return [S() for i in range(n)]
def LSR(n):
    return [LS() for i in range(n)]

sys.setrecursionlimit(1000000)
mod = 1000000007

def solve():
    d = I()
    c = LI()
    s = LIR(d)
    t = IR(d)
    n = 26
    v = 0
    minus = sum_ = sum(c)
    last = [0]*n
    for i in range(d):
        ti = t[i]-1
        minus -= (i+1-last[ti])*c[ti]
        v += s[i][ti]-minus
        minus += sum_
        last[ti] = i+1
        print(v)
    return

#Solve
if __name__ == "__main__":
    solve()

Submission status: Accepted

Solution code:
# import sys
# input = sys.stdin.readline

def mp(): return map(int, input().split())
def lmp(): return list(map(int, input().split()))

d = int(input())
c = lmp()
s = []
for i in range(d):
    s.append(lmp())
t = []
for i in range(d):
    t.append(int(input()))
last = [0] * 26
ful = 0
for i in range(d):
    tar = t[i]-1
    ful += s[i][tar]
    last[tar] = i+1
    for j in range(26):
        ful -= c[j]*(i+1-last[j])
    print(ful)

Submission status: Accepted

Solution code:
###==================================================
### import
#import bisect
#from collections import Counter, deque, defaultdict
#from copy import deepcopy
#from functools import reduce, lru_cache
#from heapq import heappush, heappop
#import itertools
#import math
#import string
import sys
### stdin
def input(): return sys.stdin.readline()
def iIn(): return int(input())
def iInM(): return map(int, input().split())
def iInM1(): return map(int1, input().split())
def iInLH(): return list(map(int, input().split()))
def iInLH1(): return list(map(int1, input().split()))
def iInLV(n): return [iIn() for _ in range(n)]
def iInLV1(n): return [iIn()-1 for _ in range(n)]
def iInLD(n): return [iInLH() for _ in range(n)]
def iInLD1(n): return [iInLH1() for _ in range(n)]
def sInLH(): return list(input().split())
def sInLV(n): return [input().rstrip("\n") for _ in range(n)]
def sInLD(n): return [sInLH() for _ in range(n)]
### stdout
def OutH(lst): print(*lst)
def OutV(lst): print(*lst, sep="\n")
### setting
sys.setrecursionlimit(10 ** 6)
### utils
int1 = lambda x: int(x) - 1
### constants
INF = int(1e9)
MOD = 1000000007
dx = (-1, 0, 1, 0)
dy = (0, -1, 0, 1)
###---------------------------------------------------

N_ALP = 26

D = iIn()
C = iInLH()
S = iInLD(D)
T = iInLV1(D)

score = 0
past = [0] * N_ALP
for i in range(D):
    score += S[i][T[i]]
    past = [x+1 for x in past]
    past[T[i]] = 0
    for j in range(N_ALP):
        score -= C[j] * past[j]
    print(score)

Submission status: Accepted

Solution code:
D = int(input())

C = list(map(int, input().split()))#下がりやすさ
S = [list(map(int, input().split())) for _ in range(D)]#増加分
T = [int(input()) for _ in range(D)]#コンテストの開催
Last = [-1] * 26

ans = 0

for i in range(D):
    t = T[i]
    ans += S[i][t-1]
    Last[t-1] = i 
    for k in range(26):
        ans -= C[k] * (i - Last[k])
    print(ans)

Submission status: Accepted

Solution code:
D=int(input())
c=list(map(int,input().split()))
s=[list(map(int,input().split())) for i in range(D)]
t=[int(input()) for i in range(D)]
last=[0]*26
m=0
for i in range(D):
    m+=s[i][t[i]-1]
    last[t[i]-1]=i+1
    for j in range(26):
        m-=c[j]*(i+1-last[j])
    print(m)

Submission status: Accepted

Solution code:
D = int(input())
C = list(map(int, input().split()))  # 低下
S = [list(map(int, input().split())) for _ in range(D)]
T = [int(input()) for _ in range(D)]
Last = [-1] * 27
def calc_pena(day):
    res = 0
    for l, c in zip(Last, C):
        res += (day-l)*c
    return res
ans = 0
for i, (s, t) in enumerate(zip(S, T)):
    t -= 1
    Last[t] = i
    ans += s[t]
    ans -= calc_pena(i)
    print(ans)

Submission status: Accepted

Solution code:
D = int(input())
Cs = list(map(int, input().split()))
Sss = []
for _ in range(D):
    Ss = list(map(int, input().split()))
    Sss.append(Ss)

Ts = []
for _ in range(D):
    Ts.append(int(input()))

lasts = [0] * 26
score = 0
for i, t in enumerate(Ts):
    d = i + 1
    t -= 1
    score += Sss[i][t]
    lasts[t] = d
    for j in range(26):
        score -= (d - lasts[j]) * Cs[j]
    print(score)

Submission status: Accepted

Solution code:
d=int(input())
c=list(map(int,input().split()))
s=[]
for i in range(d):
	r=list(map(int,input().split()))
	s.append(r)
def score(t):
	last=[0]*26
	happy=0
	for i in range(d):
		last[t[i]-1]=i+1
		happy+=s[i][t[i]-1]
		for j in range(26):
			happy-=(i+1-last[j])*c[j]
		print(happy)
score([int(input())for _ in range(d)])

Submission status: Accepted

Solution code:
d = int(input())
clst = list(map(int, input().split()))
s = [list(map(int, input().split())) for _ in range(d)]
t = [int(input()) for _ in range(d)]
last = [0 for _ in range(26)]

def score_minus(d):
    minus = 0
    for c, ls in zip(clst, last):
        minus += c * (d - ls)
    return minus
    
ans = 0
for d, num in enumerate(t, start = 1):
    ans += s[d - 1][num - 1]
    last[num - 1] = d
    ans -= score_minus(d)
    print(ans)

Submission status: Accepted

Solution code:
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 status: Accepted

Solution code:
d = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for i in range(d)]
t = [int(input()) for i in range(d)]
contest_types = 26


# 最後にコンテストを行った日
last_days = {}
for c_type in range(contest_types):
    last_days[c_type] = 0


ans = 0
for i in range(d):
    day = i + 1
    c_type = t[i] - 1
    ans += s[i][c_type]

    last_days[c_type] = day
    for c_type in range(contest_types):
        ans -= c[c_type] * (day - last_days[c_type])
        
    print(ans)

Submission status: Accepted

Solution code:
D = int(input())
c = [int(i) for i in input().split()]
s = []
for _ in range(D):
    s.append([int(i) for i in input().split()])
score = 0
days = [0]*26
for k in range(D):
    t = int(input())-1
    for i in range(26):
        if i == t:
            days[i] = 0
        else:
            days[i] += 1
    score += s[k][t]
    for i in range(26):
        score -= days[i]*c[i]
    print(score)

Submission status: Accepted

Solution code:

import sys
def input():return sys.stdin.readline()[:-1]
def N(): return int(input())
def NM():return map(int,input().split())
def L():return list(NM())
def LN(n):return [N() for i in range(n)]
def LL(n):return [L() for i in range(n)]
def Yes(x):print("Yes")if x==True else print("No")

d=N()
c=L()
s=[[0]+L() for i in range(d)]
ld=[-1]*(27)
score=0
for i in range(d):
    t=N()
    score+=s[i][t]
    ld[t]=i
    for j,k in zip(ld[1:],c):
        score-=(i-j)*k
    print(score)

Submission status: Accepted

Solution code:
d = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for i in range(d)]
sat = 0
last = [0]*26
for i in range(d):
  t = int(input())
  sat += s[i][t-1]
  last[t-1]=i+1
  for j in range(26):
    sat -= c[j]*(i+1-last[j])
  print(sat)

Submission status: Accepted

Solution code:
import sys, re
from collections import deque, defaultdict, Counter
from math import ceil, sqrt, hypot, factorial, pi, sin, cos, tan, asin, acos, atan, radians, degrees, log2, gcd
from itertools import accumulate, permutations, combinations, combinations_with_replacement, product, groupby
from operator import itemgetter, mul
from copy import deepcopy
from string import ascii_lowercase, ascii_uppercase, digits
from bisect import bisect, bisect_left, insort, insort_left
from heapq import heappush, heappop
from functools import reduce
def input(): return sys.stdin.readline().strip()
def INT(): return int(input())
def MAP(): return map(int, input().split())
def LIST(): return list(map(int, input().split()))
def ZIP(n): return zip(*(MAP() for _ in range(n)))
sys.setrecursionlimit(10 ** 9)
INF = float('inf')
mod = 10 ** 9 + 7
#import numpy as np
#from decimal import *

D = INT()
c = LIST()
s = [LIST() for _ in range(D)]

last = [0]*26
ans = 0
for i in range(D):
	t = INT()
	ans += s[i][t-1]
	last[t-1] = i+1
	decreasing = sum([c[j]*(i+1-last[j]) for j in range(26)])
	ans -= decreasing
	print(ans)

Submission status: Accepted

Solution code:
D = int(input())
C = list(map(int, input().split()))
S = [list(map(int, input().split())) for _ in range(D)]
T = [int(input()) for _ in range(D)]

res = 0
last = [0 for _ in range(26)]

for d in range(D):
    res += S[d][T[d] - 1]
    last[T[d] - 1] = d + 1
    for j in range(26):
        res -= C[j] * ((d + 1) - last[j])
    print(res)

Submission status: Accepted

Solution code:
D = int(input())
cs = list(map(int, input().split()))
ss = [list(map(int, input().split())) for _ in range(D)]
ts = [int(input()) for _ in range(D)]

def calc(ts):
    last = [-1] * 26
    sat = 0
    for i, t in enumerate(ts):
        sat += ss[i][t-1]
        last[t-1] = i
        sat -= sum(c * (i - l) for c, l in zip(cs, last))
            
    return sat

for i in range(len(ts)):   
    print(calc(ts[:i+1]))

Submission status: Accepted

Solution code:
D = int(input())
C = [int(a) for a in input().split()]
S = [[int(a) for a in input().split()] for _ in range(D)]
T = [int(input()) - 1 for _ in range(D)]

def calc():
    s = 0
    last = [-1] * 26
    for i, t in enumerate(T):
        s += S[i][t]
        last[t] = i
        for j, c in enumerate(C):
            s -= c * (i - last[j])
        print(s)
    return s

calc()

Submission status: Accepted

Solution code:
D = int(input())
*C, = map(int, input().split())
S = [list(map(int, input().split())) for _ in range(D)]
T = [int(input()) for _ in range(D)]
score = 0
last = [0] * 26
for d, t in enumerate(T, 1):
    score += S[d - 1][t - 1]
    last[t - 1] = d
    for c, l in zip(C, last): score -= c * (d - l)
    print(score)

Submission status: Accepted

Solution code:
import random

D=int(input())
c=list(map(int,input().split()))
s=[list(map(int,input().split())) for i in range(D)]

res=0
last=[0]*26
for i in range(D):
    t=int(input())
    res+=s[i][t-1]
    last[t-1]=i+1
    minus=sum(c[j]*(i+1-last[j]) for j in range(26))
    res-=minus
    print(res)

Submission status: Accepted

Solution code:
D=int(input())
C=list(map(int,input().split()))
S=[list(map(int,input().split())) for i in range(D)]
T=0
X=[0]*26
P=0
for i in range(D):
  T=int(input())
  P+=S[i][T-1]
  for j in range(26):
    X[j]+=C[j]
  X[T-1]=0
  P-=sum(X)
  print(P)

Submission status: Accepted

Solution code:
# でつoO(YOU PLAY WITH THE CARDS YOU'RE DEALT..)
import sys
def main(D, C, S, T):
    satisfaction = 0
    last = [-1] * 26
    for i, t in enumerate(T):
        t -= 1
        satisfaction += S[i][t]
        last[t] = i
        for j, c in enumerate(C):
            satisfaction -= c * (i - last[j])
        print(satisfaction)

if __name__ == '__main__':
    input = sys.stdin.readline
    D = int(input())
    *C, = map(int, input().split())
    S = [list(map(int, input().split())) for _ in range(D)]
    T = [int(input()) for _ in range(D)]
    main(D, C, S, T)

Submission status: Accepted

Solution code:
D = int(input())
c = [0] + list(map(int, input().split()))
s = [0] + [[0] + list(map(int, input().split())) for _ in range(D)]
last = [0]*27

SL = 0

for d in range(1, D+1):
    t = int(input())
    last[t] = d
    SL += s[d][t]
    for i in range(1, 27):
        SL -= c[i] * (d - last[i])
    
    print(SL)

Submission status: Accepted

Solution code:
D = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for _ in range(D)]
t = [int(input()) for _ in range(D)]

L = [0] * 26

ans = 0
for i in range(D):
  ans += s[i][t[i] - 1]
  for j in range(26):
    if t[i] - 1 == j:
      L[j] = 0
    else:
      L[j] += 1
      ans -= L[j] * c[j]
  print(ans)

Submission status: Accepted

Solution code:
from collections import defaultdict
ans = 0
d = int(input())
c = list(map(int,input().split()))
s = []
for i in range(d):
    s.append(list(map(int,input().split())))
dic = defaultdict(int)
for i in range(1,d+1):
    t = int(input())
    ans += s[i-1][t-1]
    dic[t-1] = i
    for j in range(26):
        ans -= c[j]*(i-dic[j])
    print(ans)

Submission status: Accepted

Solution code:
import bisect
import cmath
import heapq
import itertools
import math
import operator
import os
import random
import re
import string
import sys
from collections import Counter, defaultdict, deque
from copy import deepcopy
from decimal import Decimal
from functools import lru_cache, reduce
from math import gcd
from operator import add, itemgetter, mul, xor

import numpy as np

if os.getenv("LOCAL"):
    sys.stdin = open("_in.txt", "r")

sys.setrecursionlimit(10 ** 9)
INF = float("inf")
IINF = 10 ** 18
MOD = 10 ** 9 + 7
# MOD = 998244353


D = int(sys.stdin.buffer.readline())
C = list(map(int, sys.stdin.buffer.readline().split()))
S = [list(map(int, sys.stdin.buffer.readline().split())) for _ in range(D)]
T = [int(sys.stdin.buffer.readline()) for _ in range(D)]

S = [0] + S
T = [0] + T

lasts = [0] * 26
score = 0
for d in range(1, D + 1):
    t = T[d] - 1
    score += S[d][t]

    lasts[t] = d
    for j, c in enumerate(C):
        score -= (d - lasts[j]) * c
    print(score)

Submission status: Accepted

Solution code:
import sys
input = sys.stdin.readline

d = int(input())
C = list(map(int,input().split()))
S = [list(map(int,input().split())) for i in range(d)]
T = [int(input()) for i in range(d)]
c = 26

ans = 0
L = [-1] * c

for i in range(d):
    T[i] -= 1

for i in range(d):
    ans += S[i][T[i]]
    L[T[i]] = i
    for j in range(c):
        ans -= C[j] * (i - L[j])
    print(ans)

Submission status: Accepted

Solution code:
import sys
input = sys.stdin.readline

D=int(input())
C=list(map(int,input().split()))
S=[list(map(int,input().split())) for i in range(D)]
T=[int(input()) for i in range(D)]

ANS=0
LAST=[0]*26
for i in range(D):
    t=T[i]
    ANS+=S[i][t-1]
    LAST[t-1]=i+1

    for j in range(26):
        ANS-=C[j]*(i+1-LAST[j])

    print(ANS)

Submission status: Accepted

Solution code:
d = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for _ in range(d)]
t = [int(input()) for _ in range(d)]
ans = 0
k = [-1] * 26
for i in range(d):
    p = t[i]
    ans += s[i][p - 1]
    k[p - 1] = i
    for j in range(26):
        ans -= c[j] * (i - k[j])
    print(ans)

Submission status: Accepted

Solution code:
import sys
input = sys.stdin.readline
D = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for _ in range(D)]
t = [0] * D
for i in range(D): t[i] = int(input())

last = [0] * 26
def score(day, score):
  for i in range(26): score -= (day - last[i]) * c[i]
  return score
res = 0
for d in range(D):
  contest = t[d] - 1
  last[contest] = d + 1
  res = score(d + 1, res) + s[d][contest]
  print(res)

Submission status: Accepted

Solution code:
d = int(input())
c = list(map(int, input().split()))
s = [list(map(int, input().split())) for i in range(d)]
t = [int(input()) for i in range(d)]
last = [-1]*26

ans = 0
for i in range(d):
    op = t[i]-1
    ans += s[i][op]
    last[op]=i
    for j in range(26):
        ans -= c[j]*(i-last[j])
    print(ans)

Submission status: Accepted

Solution code:
import sys

readline = sys.stdin.readline
readall = sys.stdin.read
ns = lambda: readline().rstrip()
ni = lambda: int(readline().rstrip())
nm = lambda: map(int, readline().split())
nl = lambda: list(map(int, readline().split()))
prn = lambda x: print(*x, sep='\n')

def solve():
    m = 26
    D = ni()
    c = nl()
    s = [nl() for _ in range(D)]
    sche = [ni() - 1 for _ in range(D)]
    last = [-1] * m
    ans = 0
    for i in range(D):
        t = sche[i]
        ans += s[i][t]
        last[t] = i
        for j in range(m):
            ans -= c[j] * (i - last[j])
        print(ans)

    return

solve()

Submission status: Accepted

Solution code:
import sys
input=lambda: sys.stdin.readline().rstrip()
D = int(input())
C = [int(i) for i in input().split()]
S = [[int(i) for i in input().split()] for _ in range(D)]

result = 0
Last = [0]*26
for i in range(D):
    t = int(input())
    result += S[i][t - 1]
    Last[t-1] = i+1
    for j in range(26):
        result -= C[j] * (i + 1 - Last[j])
    print(result)

Submission status: Accepted

Solution code:
D = int(input())
c = list(map(int,input().split()))
s = [list(map(int,input().split())) for i in range(D)]
t = [int(input()) for i in range(D)]
cont = 0
last = [0]*26
for i in range(D):
    cont += s[i][t[i]-1]
    last[t[i]-1] = i+1
    for j in range(26):
        cont -= c[j]*(i+1-last[j])
    print(cont)

Submission status: Accepted

Solution code:
d=int(input())
c=list(map(int,input().split()))
s=[list(map(int,input().split()))for _ in range(d)]
t=[int(input())-1 for _ in range(d)]
x=[0]*26
ans=0
for i in range(d):
  v=t[i]
  ans+=s[i][v]
  for j in range(26):
    if j==v:x[j]=0
    else:x[j]+=1
    ans-=c[j]*x[j]
  print(ans)

Submission status: Accepted

Solution code:
mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    D = int(input())
    C = list(map(int, input().split()))
    S = []
    for _ in range(D):
        S.append(list(map(int, input().split())))
    ans = 0
    last = [-1] * 26
    for d in range(D):
        t = int(input()) - 1
        ans += S[d][t]
        last[t] = d
        for i in range(26):
            ans -= C[i] * (d - last[i])
        print(ans)


if __name__ == '__main__':
    main()