[Java] 추억 점수 (programmers)

Haeun Noh·2023년 6월 7일

programmers

목록 보기

50/64

0607

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["may", "kein", "kain"]이고 각 인물의 그리움 점수가 [5점, 10점, 1점]일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 됩니다. 다른 사진 속 인물의 이름이 ["kali", "mari", "don", "tony"]이고 ["kali", "mari", "don"]의 그리움 점수가 각각 [11점, 1점, 55점]]이고, "tony"는 그리움 점수가 없을 때, 이 사진의 추억 점수는 3명의 그리움 점수를 합한 67(11 + 1 + 55)점입니다.

그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성해주세요.

제한사항

3 ≤ name의 길이 = yearning의 길이≤ 100
3 ≤ name의 원소의 길이 ≤ 7
name의 원소들은 알파벳 소문자로만 이루어져 있습니다.
name에는 중복된 값이 들어가지 않습니다.
1 ≤ yearning[i] ≤ 100
yearning[i]는 i번째 사람의 그리움 점수입니다.
3 ≤ photo의 길이 ≤ 100
1 ≤ photo[i]의 길이 ≤ 100
3 ≤ photo[i]의 원소(문자열)의 길이 ≤ 7
photo[i]의 원소들은 알파벳 소문자로만 이루어져 있습니다.
photo[i]의 원소들은 중복된 값이 들어가지 않습니다.

입출력 예

`name`	`yearning`	`photo`	`result`
`["may", "kein", "kain", "radi"]`	`[5, 10, 1, 3]`	`[["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"], ["kon", "kain", "may", "coni"]]`	`[19, 15, 6]`
`["kali", "mari", "don"]`	`[11, 1, 55]`	`[["kali", "mari", "don"], ["pony", "tom", "teddy"], ["con", "mona", "don"]]`	`[67, 0, 55]`
`["may", "kein", "kain", "radi"]`	`[5, 10, 1, 3]`	`[["may"],["kein", "deny", "may"], ["kon", "coni"]]`	`[5, 15, 0]`

입출력 예 설명

입출력 예 #1
첫 번째 사진 속 "may", "kein", "kain", "radi"의 그리움 점수를 합치면 19(5 + 10 + 1 + 3)점 입니다. 두 번째 사진 속 그리워하는 사람들인 "may"와 "kein"의 그리움 점수를 합치면 15(5 + 10)점입니다. 세 번째 사진의 경우 "kain"과 "may"만 그리워하므로 둘의 그리움 점수를 합한 6(1 + 5)점이 사진의 추억 점수입니다. 따라서 [19, 15, 6]을 반환합니다.

입출력 예 #2
첫 번째 사진 속 그리워하는 사람들인 "kali", "mari", "don"의 그리움 점수를 합치면 67(11 + 1 + 55)점입니다. 두 번째 사진 속엔 그리워하는 인물이 없으므로 0점입니다. 세 번째 사진 속 그리워하는 사람은 "don"만 있으므로 55점입니다. 따라서 [67, 0, 55]를 반환합니다.

입출력 예 #3
설명 생략

풀이 방법

저는 이 문제를 보자마자 이런 생각이 들었습니다.
아, 삼중 for문을 써야 하겠구나.

이 문제는 각 이름에 추억 점수라는 것이 주어집니다.
그리고 이중 배열에 들어있는 이름들의 묶음에서 같은 이름이 있을 경우 그 이름에 해당하는 추억 점수를 더하여 반환해야 합니다.

        for ( int i = 0; i < name.length; i++) {
            for ( int j = 0; j < photo.length; j++) {
                for ( int k = 0; k < photo[j].length; k++) {
                    if ( name[i].equals(photo[j][k])) {
                        answer[j] += yearning[i];
                    }
                }
            }
        }

위의 코드를 보면 for문이 세 개가 중첩되어 있습니다.

첫 번째 for문은 name배열을 돌고 있습니다. photo[j]에 들어있는 이름들과 비교하여 같은 이름이 있는지를 비교합니다.

또한 첫 번째 for문은 yearning배열을 돌고 있습니다. yearning[i]은 name[j]의 추억점수와 같습니다. 이름과 추억점수의 인덱스가 동일하니 이를 활용할 수 있겠군요!

두 번째 for문은 photo배열을 돌고 있습니다. photo가 2차원 배열이므로 for문 두 개가 필요하겠네요!

세 번째 for문은 photo[j]배열을 돌고 있습니다. 드디어 이 for문에서 조건을 비교할 수 있습니다.
만약 name[i]와 photo[j][k]의 문자열이 같다면 name[i]에 해당하는 추억점수를 answer[i]방에 누적해줍니다.

(주의) 배열은 new를 사용하여 생성이 될 때 자동적으로 초기값이 넣어지며 초기화가 됩니다. 굳이 for문을 하나 더 사용하여 실행시간을 늦출 필요가 전혀 없다는 말입니다.

짜잔 그럼 문제가 해결되었습니다.
아래의 소스코드와 풀이 방법을 비교하며 찬찬히 뜯어보세요.

소스 코드

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
        for ( int i = 0; i < name.length; i++) {
            for ( int j = 0; j < photo.length; j++) {
                for ( int k = 0; k < photo[j].length; k++) {
                    if ( name[i].equals(photo[j][k])) {
                        answer[j] += yearning[i];
                    }
                }
            }
        }
        return answer;
    }
}