[프로그래머스][Java][Lv. 1] 가장 많이 받은 선물

문제 설명

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

---

제한 사항

---

입출력 예

---

풀이

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

class Solution {
    public int solution(String[] friends, String[] gifts) {
        int[] answer = new int[friends.length];
        HashMap<String, Gift> giftMap = new HashMap<>();
        
        // 주어진 대로 세팅
        for (String name : friends) giftMap.put(name, new Gift());
        for (String gift : gifts) {
            String[] names = gift.split(" ");
            giftMap.get(names[0]).addGivenList(names[1]);
            giftMap.get(names[1]).addReceivedList(names[0]);
        }
        
        for (int i = 0; i < friends.length - 1; i++) {
            Gift a = giftMap.get(friends[i]);
            
            for (int j = i + 1; j < friends.length; j++) {
                Gift b = giftMap.get(friends[j]);
                
                // 주고 받은 기록이 있을 때
                if (a.getGivenList().contains(friends[j]) 
                    || a.getReceivedList().contains(friends[j])) {
                    int aGivenCount = Collections.frequency(a.getGivenList(), friends[j]);
                    int bGivenCount = Collections.frequency(b.getGivenList(), friends[i]);
                    
                    if (aGivenCount < bGivenCount) {
                        answer[j]++;
                    } else if (aGivenCount > bGivenCount) {
                        answer[i]++;
                    } else {
                        if (a.getScore() < b.getScore()) {
                            answer[j]++;
                        } else if (a.getScore() > b.getScore()) {
                            answer[i]++;
                        }
                    }
                } else {  // 주고 받은 기록이 없을 때
                    if (a.getScore() < b.getScore()) {
                        answer[j]++;
                    } else if (a.getScore() > b.getScore()) {
                        answer[i]++;
                    }
                }
            }
        }
        return Arrays.stream(answer).max().getAsInt();
    }
    
    static class Gift {
        private ArrayList<String> givenList = new ArrayList<>();
        private ArrayList<String> receivedList = new ArrayList<>();
        private int score;
        
        public ArrayList<String> getGivenList() { return givenList; }
        public ArrayList<String> getReceivedList() { return receivedList; }
        public int getScore() { return givenList.size() - receivedList.size(); }
        
        public void addGivenList(String name) { givenList.add(name); }
        public void addReceivedList(String name) { receivedList.add(name); }
    }
}

---

후기

문제에서 요구하는 사항이 은근 많다. 그래서 구현할 때 빼먹은 사항은 없는지 잘 살펴야 한다. 나도 제대로 안 읽어서 중간에 몇 번 다시 고쳤다;; 푸는 방법은 일단 문제에서 주어진 대로 세팅을 하고, 주어진 것을 따로 클래스로 만든 곳에 저장해서 해당 클래스 내의 리스트 값을 비교해서 선물 카운트를 증가시킨다. 마지막에는 선물 카운트 중 제일 큰 것을 선택한다.