본문 바로가기

Coding Test/프로그래머스 연습

추억 점수

문제 설명
사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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.HashMap;
import java.util.Map;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
        Map<String,Integer> map = new HashMap<>();
        
        //map에 이름과 점수 매칭
        for(int i=0; i < name.length; i++){
            map.put(name[i],yearning[i]);
        }
        
        //photo에 있는 이름으로 점수 합산
        for(int j=0; j <  photo.length; j++){
            int num = 0; // 점수 기록용
            String[] arrString = photo[j];
            
            for(String person : arrString){
                num += map.getOrDefault(person,0); // map에 없는 경우 0점
            }
            answer[j] = num;
        }
        return answer;
    }
}

 

1. 배열 초기화하는 것 까먹지 말기

2. map에 put/getOrDefault 기억

 

map으로 점수와 이름을 매칭시켜준 뒤에

photo 배열을 하나씩 돌면서  arrString 배열을 가져와 그 안에서 향상된 for문을 이용하여 값을 더해줌.

 

🌈다른 풀이

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;
    }
}

 

 if(photo[i][j].equals(name[k])) answer[i] += yearning[k];

 

map을 쓰지않고 바로 equals로 글이 같은지 비교한뒤에 같으면 answer에 점수 추가