문제 설명
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); }
}
}
후기
문제에서 요구하는 사항이 은근 많다. 그래서 구현할 때 빼먹은 사항은 없는지 잘 살펴야 한다. 나도 제대로 안 읽어서 중간에 몇 번 다시 고쳤다;; 푸는 방법은 일단 문제에서 주어진 대로 세팅을 하고, 주어진 것을 따로 클래스로 만든 곳에 저장해서 해당 클래스 내의 리스트 값을 비교해서 선물 카운트를 증가시킨다. 마지막에는 선물 카운트 중 제일 큰 것을 선택한다.
'코딩테스트 (프로그래머스) > Java' 카테고리의 다른 글
| [프로그래머스][Java][Lv. 2] 도넛과 막대 그래프 (0) | 2024.02.07 |
|---|---|
| [프로그래머스][Java][Lv. 2] 큰 수 만들기 (0) | 2024.02.06 |
| [프로그래머스][Java][Lv. 2] 삼각 달팽이 (0) | 2024.02.01 |
| [프로그래머스][Java][Lv. 2] 두 큐 합 같게 만들기 (0) | 2024.01.31 |
| [프로그래머스][Java][Lv. 2] 소수 찾기 (0) | 2024.01.30 |