[Lv.1] 추억 점수

https://school.programmers.co.kr/learn/courses/30/lessons/176963

프로그래머스

[ 문제설명 ]

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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 함수를 완성해주세요.

[ 풀이과정 ]

import java.util.*; 

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length]; 
        //정답배열 크기 photo배열 행길이와 동일하게 설정
        HashMap<String, Integer> map = new HashMap<>();
        //HashMap선언
        //key = name, value = yearning
        
        for (int i=0; i<name.length; i++) { //for문으로 돌면서
            map.put(name[i], yearning[i]); 
            //map에 name, yearning배열 원소들을 key와 value로 쌍을지어 넣어줌
        }
        
        for (int i=0; i<photo.length; i++) { //photo배열 행순회
            int sum = 0; 
            //그리움점수합을 담을 sum 선언
            //i가 증가할때마다(photo배열 다음행으로 넘어갈때마다) sum=0으로 초기화
            
            for (int j=0; j<photo[i].length; j++) { //photo배열 열순회
                if (map.containsKey(photo[i][j])) { //map에 name(key)이 있으면
                    sum += map.get(photo[i][j]); 
                    //name(key)에 해당하는 yearning(value)값 가져와서
                    //sum에 더해주기
                }
            }
            answer[i] = sum; 
            // 정답배열에 sum 담아주기
            // 이후 sum=0으로 다시 초기화
        }
        return answer;
    }
}

[ 다른풀이 ]

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
        
        //삼중for문 이용
        for(int i = 0; i < photo.length; i++){ //photo배열 행순회
            for(int j = 0; j < photo[i].length; j++){ //photo배열 열순회
                for(int k = 0; k < name.length; k++){ //name배열 순회
                    if(photo[i][j].equals(name[k])) answer[i] += yearning[k];
                    //photo배열에 name이 있으면
                    //name에 해당하는 yearning을 더해서 answer에 담아줌
                }
            }
        }
        return answer;
    }
}

[ 문법정리 ]

1. HashMap<key, value>
요소 추가 : map.put(key, value)
요소 접근 : map.get(key)
요소 삭제 : map.remove(key)
크기 확인 : map.size()
요소 존재여부 확인 :
map.containsKey(key)
map.containsValue(value)