백준공부/java

[백준] 22113번 창영이와 버스 (bronze 1

gomduri43 2023. 6. 25. 00:42

import java.io.*;
import java.util.*;

public class Main{
    public static void main(String[] args) throws IOException{
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st=new StringTokenizer(br.readLine());
        int n=Integer.parseInt(st.nextToken());
        int m=Integer.parseInt(st.nextToken());

        int[][] map=new int[n+1][n+1];
        int[] bus=new int[m];
        st=new StringTokenizer(br.readLine());
        for(int i=0; i<m; i++){
            bus[i]=Integer.parseInt(st.nextToken());
        }

        for(int i=1; i<=n; i++){
            st=new StringTokenizer(br.readLine());
            for(int j=1; j<=n; j++){
                map[i][j]=Integer.parseInt(st.nextToken());
            }
        }
        int total=0;
        int temp=bus[0];
        for(int i=1; i<m; i++){
            total+=map[temp][bus[i]];
            temp=bus[i];
        }
        bw.write(total+"");
        bw.flush();
    }
}