[Codeforces] 431A: Black Square

https://codeforces.com/problemset/problem/431/A

Problem - 431A - Codeforces

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(bf.readLine());
		int[] arr = new int[4];
		for(int i=0; i<arr.length; i++)
			arr[i] = Integer.parseInt(st.nextToken());
		
		String str = bf.readLine();
		int score = 0;
		
		// 문자열의 각 문자를 인덱스로 하는 arr 배열,
		for(int i=0; i<str.length(); i++) 
			score += arr[str.charAt(i) - 49];
		
		System.out.println(score);
		bf.close();
	}

}

풀이

문자열이 주어질 때 각 문자열에 맞는 배열을 더하면 됨.

배열의 시작은 0이므로, 아래 아스키코드 값에서 48이 아닌 49를 빼야한다.

 

input

1 2 3 4

123214

 

arr[] = [1, 2, 3, 4]

score = arr[1] + arr[2] + arr[3] + arr[2] + arr[1] + arr[4] = 13

https://namu.wiki/w/%EC%95%84%EC%8A%A4%ED%82%A4%20%EC%BD%94%EB%93%9C