[JAVA] 추억 점수

문제설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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]의 원소들은 중복된 값이 들어가지 않습니다.

---

입출력 예

입출력 예 #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

설명 생략

---

코드 설명

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

 

💡 저는 이 문제를 보고 Map을 이용하여  name 배열의 각 요소를 키로 사용하고, yearning 배열의 해당 요소를 값으로 사용하여 점수를 구하자라는 생각이 들었습니다.

Map<String,Integer> map = new HashMap<String,Integer>();

먼저 map이라는 HashMap을 생성합니다.
name 배열의 각 요소를 키(key)로 사용하고, yearning 배열의 해당 요소를 값(value)으로 사용합니다.

이렇게 하면 이름과 그에 해당하는 점수가 맵에 저장됩니다 .

따라서 위의 코드처럼 키에는 String을 값에는 Integer로 선언해주었습니다.

int[] answer = new int[photo.length];

이제 모든 점수가 더해진 값을 넣어줄 answer를 photo의 길이에 맞춰 선언해주었습니다.

name 배열과 yearning 배열의 값을 사용하여 map에 데이터를 추가할 것입니다.

for(int i = 0; i < name.length; i++)

name.length는 name 배열의 길이를 나타냅니다.

따라서 i 변수는 0부터 name 배열의 길이보다 작을 때까지 반복문이 실행됩니다.

반복문의 각 반복에서 name[i]은 name 배열에서 i 인덱스에 해당하는 요소를 가져옵니다.
yearning[i]는 yearning 배열에서 동일한 인덱스에 해당하는 요소를 가져옵니다.

map.put(name[i], yearning[i]);

반복문을 실행하는 동안 map에 새로운 키-값 쌍을 추가합니다.

name[i]을 키(key)로 사용하고, yearning[i]을 값(value)으로 사용합니다.
이렇게 하면 map에 이름과 해당하는 추억점수가 저장됩니다.

 

이제 photo 배열을 반복하면서 각 사진에 있는 사람들의 점수를 계산하고, 이를 answer 배열에 저장하도록 하겠습니다.

for(int i = 0; i < photo.length; i++)

photo.length는 photo 배열의 길이를 나타냅니다.

따라서 i 변수는 0부터 photo 배열의 길이보다 작을 때까지 반복문이 실행됩니다.

String[] persons = photo[i];

각 반복에서 photo[i]는 photo 배열에서 i 인덱스에 해당하는 요소를 가져옵니다. 

int score = 0;

현재 사진에 있는 사람들의 총 점수를 저장하기 위해 score 변수를 초기화해줍니다.

for(int j = 0; j < persons.length; j++)

내부 반복문을 이용해 persons 배열을 반복하면서 각 사람의 점수를 계산합니다. 

j 변수는 0부터 persons 배열의 길이보다 작을 때까지 반복해줍니다.

String person = persons[j];

각 반복에서 persons[j]는 persons 배열에서 j 인덱스에 해당하는 사람의 이름을 가져옵니다. 

if(map.containsKey(person))

이 이름을 사용하여 map.containsKey(person)를 통해 map에 해당 이름이 포함되어 있는지 확인합니다.

containsKey(key)는 주어진 키 값이 포함되어 있는지를 확인하는 메서드입니다.

score += map.get(person);

만약 map에 해당 이름이 포함되어 있다면, get()메소드를 이용해 해당 이름에 대한 추억 점수를 가져옵니다.
이 값을 score에 더할 것입니다.

answer[i] = score;

내부 반복문이 종료되면 score에는 현재 사진에 있는 사람들의 총 점수가 저장됩니다.

이 값을 answer[i]에 할당하여 해당 사진의 점수를 answer 배열에 저장해주었습니다.

---

소스코드&결과

 

코드

import java.util.*;
class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        Map<String,Integer> map = new HashMap<String,Integer>();
        int[] answer = new int[photo.length];
        for(int i = 0; i < name.length; i++){
            map.put(name[i], yearning[i]);
        }
        
       for(int i = 0; i < photo.length; i++){
            String[] persons = photo[i]; 
            int score = 0;
            for(int j = 0; j < persons.length; j++){
                String person = persons[j];
                if(map.containsKey(person)){ 
                    score += map.get(person); 
                }
            }
            answer[i] = score;
        }
        return answer;
    }
}

결과

---

다른 사람이 짠 코드

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

위의 코드는 삼중 for문을 사용해 풀어주었네요!

---