신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.
다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고, k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다.
유저 ID | 유저가 신고한 ID | 설명 |
---|---|---|
"muzi" | "frodo" | "muzi"가 "frodo"를 신고했습니다. |
"apeach" | "frodo" | "apeach"가 "frodo"를 신고했습니다. |
"frodo" | "neo" | "frodo"가 "neo"를 신고했습니다. |
"muzi" | "neo" | "muzi"가 "neo"를 신고했습니다. |
"apeach" | "muzi" | "apeach"가 "muzi"를 신고했습니다. |
각 유저별로 신고당한 횟수는 다음과 같습니다.
유저 ID | 신고당한 횟수 |
---|---|
"muzi" | 1 |
"frodo" | 2 |
"apeach" | 0 |
"neo" | 2 |
위 예시에서는 2번 이상 신고당한 "frodo"와 "neo"의 게시판 이용이 정지됩니다. 이때, 각 유저별로 신고한 아이디와 정지된 아이디를 정리하면 다음과 같습니다.
유저 ID | 유저가 신고한 ID | 정지된 ID |
---|---|---|
"muzi" | ["frodo", "neo"] | ["frodo", "neo"] |
"frodo" | ["neo"] | ["neo"] |
"apeach" | ["muzi", "frodo"] | ["frodo"] |
"neo" | 없음 | 없음 |
따라서 "muzi"는 처리 결과 메일을 2회, "frodo"와 "apeach"는 각각 처리 결과 메일을 1회 받게 됩니다.
이용자의 ID가 담긴 문자열 배열 id_list, 각 이용자가 신고한 이용자의 ID 정보가 담긴 문자열 배열 report, 정지 기준이 되는 신고 횟수 k가 매개변수로 주어질 때, 각 유저별로 처리 결과 메일을 받은 횟수를 배열에 담아 return 하도록 solution 함수를 완성해주세요.
id_list | report | k | result |
---|---|---|---|
["muzi", "frodo", "apeach", "neo"] | ["muzi frodo","apeach frodo","frodo neo","muzi neo","apeach muzi"] | 2 | [2,1,1,0] |
["con", "ryan"] | ["ryan con", "ryan con", "ryan con", "ryan con"] | 3 | [0,0] |
입출력 예 #1
문제의 예시와 같습니다.
입출력 예 #2
"ryan"이 "con"을 4번 신고했으나, 주어진 조건에 따라 한 유저가 같은 유저를 여러 번 신고한 경우는 신고 횟수 1회로 처리합니다. 따라서 "con"은 1회 신고당했습니다. 3번 이상 신고당한 이용자는 없으며, "con"과 "ryan"은 결과 메일을 받지 않습니다. 따라서 [0, 0]을 return 합니다.
먼저 한 유저를 여러번 신고하면, 동일한 유저에 대한 신고 횟수는 1회로 처리되고 result는 신고한 id에 대해 메일로 받을 횟수를 따로 카운트 해야 하는 부분이 복잡할 것 같다는 생각이 들었다.
또 단순하게 신고 당한 id에 카운트를 세는 것이 아닌 신고한 id, 신고 당한 id, 신고 횟수, 결과로 받을 메일 개수 까지 이어서 생각해야 하는 것 때문에 생각하는 시간이 좀 길어졌다.
문제를 보고 생각 했던 것
- 중복된 신고 기록(report)은 1회로 처리되니까, 같은 내용은 HashSet으로 아예 중복을 없애버리자
- 유저 ID를 ArrayList로 변환해서 해당하는 ID의 index에 카운트 처리가 되도록 하자
- HashMap.getOrDefault로 카운트해보자
- report가 String에다 공백으로 구분되어 있으니 split으로 나누자
// 1. 신고 기록 중 중복 건 제거
Set<String> set = new HashSet<>(Arrays.asList(report));
report = set.toArray(new String[0]);
List<String> id_Array = new ArrayList<>(Arrays.asList(id_list));
// 2. 신고 당한 user 카운팅
HashMap<String, Integer> hm = new HashMap<>();
for (int i = 0; i < report.length; i++) {
String user = report[i].split(" ")[1];
hm.put(user, hm.getOrDefault(user, 0) + 1);
}
// 3. 신고 당한 user의 카운트가 k 이상일 경우, 신고 당한 user가 있는 신고 기록에서
// 신고한 id를 ++
for (String user : hm.keySet()) {
if (hm.get(user) >= k) {
for (String reportUser : report) {
if (reportUser.split(" ")[1].equals(user)) {
answer[id_Array.indexOf(reportUser.split(" ")[0])]++;
}
}
}
}
알고리즘 문제에서 반복문이 자주 나오면 뭔가 불안해지는 사람.. 은 나 🙋♂️
report에서 중복된 건은 아예 없애고 시작했다. 없애고 시작하면 불필요한 건에 대한 반복 자체는 없지 않을까 ?! 라는 생각 (사실 이렇게 해 놓고 중복 건이 하나도 없으면 그 자체로 불필요한 코드가 추가된 것이라는 생각을 나중에 하게 되었다.)
신고한 유저는 id, 신고당한 유저는 user로 구분해놓고 먼저 user의 신고 횟수를 HashMap에 넣어 카운트했다.
그 후 user의 카운트가 k 이상일 경우에 answer에 id별로 ++해줘야 하는데, [신고한 유저+신고당한 유저]와 [신고한 유저 index = answer index]를 묶어서 생각해야 되기 때문에 또 split으로 나누는 과정을 추가했다.
지금 내 생각으론 제일 최적의 방법이다, 라고 생각하고 제출했지만 테스트케이스 3, 21에서 시간 초과 발생 💀
이곳 저곳 바꿔봐도 개선이 안돼서 결국 다른 분의 코드를 참고했다.
public int[] solution(String[] id_list, String[] report, int k) {
int[] answer = {};
answer = new int[id_list.length];
// 1. 각 user에 신고한 id를 추가할 map 생성
Map<String, HashSet<String>> reportMap = new HashMap<>();
List<String> id_Array = new ArrayList<>(Arrays.asList(id_list));
// 2. Map 초기화
for (int i = 0; i < id_list.length; i++) {
HashSet<String> user = new HashSet<>();
reportMap.put(id_list[i], user);
}
// 3. 신고 기록에서 신고 당한 user에 신고한 id 추가 (HashSet에 id add)
for (String str : report) {
String id = str.split(" ")[0];
String user = str.split((" "))[1];
reportMap.get(user).add(id);
}
// 4. 신고 당한 user의 카운트(신고한 id 수)가 k 이상일 경우
// 해당 user를 신고한 id에게 보낼 메일 수 ++
for (String user : reportMap.keySet()) {
HashSet<String> id = reportMap.get(user);
if (id.size() >= k) {
for (String userId : id) {
answer[id_Array.indexOf(userId)]++;
}
}
}
return answer;
}
당장 내가 작성한 코드가 가장 최선의 방법이라고 생각하지만, 더 최적화된 코드를 보면서 아직 더 다양하게 생각할 수 있는 스킬과 적용의 경험이 부족하다고 생각이 든다. 아직 갈 길이 너무 멀구만 🥳
출처 - https://school.programmers.co.kr/learn/courses/30/lessons/92334
참고 - https://eunoia3jy.tistory.com/166