개발 창고/Algorithm

[프로그래머스] 추억 점수 - JAVA

로이제로 2024. 1. 31. 07:00
반응형


문제

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제 내용은 지적 재산 보호 차원에서 가져오지 않고 풀이만 공유드리도록 하겠습니다.


풀이

 제 풀이가 무조건적으로 맞는 것도 최적의 답변도 아니지만, 이런 풀이도 있다는 차원에서 작성해 보며, 좀 더 나은 방법이 있다면 이야기해 주셔도 도움 될 것 같습니다.

import java.util.HashMap;

class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length];
        
        // Step. 친구별 추억 점수를 HashMap으로 변환
        HashMap<String, Integer> friends = new HashMap<>();
        for(int i = 0; i < name.length; i++){
            String friend   = name[i];      // 친구 이름
            int score       = yearning[i];  // 친구 추억 점수
            
            friends.put(friend, score);
        }
        
        // Step. 사진별 추억 점수 계산
        for(int i = 0; i < photo.length; i++){
            int score = 0;
            for(int j = 0; j < photo[i].length; j++){
                String friend = photo[i][j];
                // Step. 추억 점수가 있는 친구 목록만 가산
                if(friends.get(friend) != null){
                   score += friends.get(friend);
                }
            }
            answer[i] = score;
        }
        
        return answer;
    }
}

코드 실행 결과
제출 결과

반응형