백준 22113번: 창영이와 버스

문제

문제 바로가기> 백준 22113번: 창영이와 버스

풀이

2차원 배열을 이용하면 문제를 쉽게 해결할 수 있다.

def solution():
    import sys
    input = sys.stdin.readline
    n, m = map(int, input().split())
    bus = list(map(int, input().split()))
    fee = []
    for i in range(n): fee.append(list(map(int, input().split())))
    ans = 0
    for i in range(m-1):
        ans+=fee[bus[i]-1][bus[i+1]-1] # 배열의 index는 0부터 시작
    print(ans)
solution()