[프로그래머스, Java] 추억 점수

사진을 클릭하면 해당 문제로 이동

💡풀이

- HashMap 활용

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        // 이름과 그리움 점수를 매핑하는 맵 생성
        Map<String,Integer> map = new HashMap<>();

        // 이름 배열을 순회하며 맵에 이름과 그리움 점수 저장
        for (int i = 0; i < name.length; i++) {
            map.put(name[i], yearning[i]);
        }

        // 결과를 저장할 리스트
        ArrayList<Integer> answer = new ArrayList<>();

        // 각 사진에 대해 그리움 점수 계산
        for (String[] arr : photo) {
        
            int total = Arrays.stream(arr)
                    .mapToInt(e -> map.getOrDefault(e, 0)) // 맵에 없으면 0점
                    .sum();
                    
            answer.add(total);
        }

        // ArrayList를 int 배열로 변환하여 반환
        return answer.stream().mapToInt(Integer::intValue).toArray();
    }
}

---

 

GitHub - okjunghyeon/Programmers_CodingTest: 프로그래머스 관련 코딩테스트 문제를 풀이한 저장소입니다.