👉 문제 설명
신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.
- 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
- 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
- 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
- k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
- 유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.
👉 제한 사항
- 2 ≤ id_list의 길이 ≤ 1,000
- 1 ≤ id_list의 원소 길이 ≤ 10
- id_list의 원소는 이용자의 id를 나타내는 문자열이며 알파벳 소문자로만 이루어져 있습니다.
- id_list에는 같은 아이디가 중복해서 들어있지 않습니다.
- 1 ≤ report의 길이 ≤ 200,000
- 3 ≤ report의 원소 길이 ≤ 21
- report의 원소는 "이용자id 신고한id"형태의 문자열입니다.
- 예를 들어 "muzi frodo"의 경우 "muzi"가 "frodo"를 신고했다는 의미입니다.
- id는 알파벳 소문자로만 이루어져 있습니다.
- 이용자id와 신고한id는 공백(스페이스)하나로 구분되어 있습니다.
- 자기 자신을 신고하는 경우는 없습니다.
- 1 ≤ k ≤ 200, k는 자연수입니다.
- return 하는 배열은 id_list에 담긴 id 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.
우선 문제를 읽고 다음과 같은 것들이 필요하다고 생각했다.
1. 유저 목록을 기반으로 각 유저가 신고한 사람 목록 (userReportList)
2. 유저 목록을 기반으로 신고 당한 횟수 (reportedUserCount)
3. 이용 정지 당한 유저 목록 (banList)
각 유저가 서로 다른 유저를 계속해서 신고할 수 있고(HashMap), 동일한 유저에 대한 신고 횟수는 제한(HashSet)되므로 1번의 목록은 HashMap<String, HashSet<String>> 타입으로 선언했다.
👀 HashMap
- Map 인터페이스를 구현한 대표적인 Map컬렉션
- 키와 값으로 구성
- null값과 null 키를 허용
👀 HashSet
- 해시 테이블이 지원하는 Set 인터페이스를 구현하는 클래스.
- 집합의 반복 순서를 보장하지 않으며 중복된 원소를 허용하지 않고 null값을 허용한다.
코드를 작성한 순서는 다음과 같다.
1. 유저 목록(id_list)을 기반으로 각 변수에 유저 목록 세팅
2. 공백으로 분리되어있는 "이용자id 신고한id"를 각각 분리하여 userReportList에 신고 목록을 기록하고, reportedUserCount에 신고 당한 횟수 기록
3. reportedUserCount에서 k번 이상 신고당한 유저를 추려내어 banList에 기록
4. banList에 기록되어있는 정지 유저를 신고한 이용자들에게 answer 숫자 1 추가
작성 후 코드를 실행하니 두번째 테스트 케이스를 실패하였다.
한 유저가 같은 유저를 반복적으로 신고한 경우, 2번 순서에서 신고 목록을 기록함과 동시에 신고 당한 횟수를 기록하다보니 해당 유저가 여러번 신고한 것으로 기록되었기 때문이었다. (신고 목록의 경우 Set으로 선언되어 중복이 제거되었음)
따라서 다음과 같이 코드를 변경해주었다.
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
class Solution {
public int[] solution(String[] id_list, String[] report, int k) {
int[] answer = new int[id_list.length];
// 신고한 목록 Map<유저 아이디, 해당 유저가 신고한 사람>
HashMap<String, HashSet<String>> userReportList = new HashMap<>();
// 신고 당한 목록 Map<유저 아이디, 신고 당한 횟수>
HashMap<String, Integer> reportedUserCount = new HashMap<>();
// 이용 정지당한 유저 목록
ArrayList<String> banList = new ArrayList<>();
// Map 세팅
for(int i = 0; i < id_list.length; i++) {
answer[i] = 0;
HashSet nullSet = new HashSet();
userReportList.put(id_list[i], nullSet);
reportedUserCount.put(id_list[i], 0);
}
for(int i = 0; i < report.length; i++) {
String reporterUser = report[i].split(" ")[0];
String reportedUser = report[i].split(" ")[1];
userReportList.get(reporterUser).add(reportedUser);
// reportedUserCount.replace(reportedUser, reportedUserCount.get(reportedUser) + 1);
}
for(int i = 0; i < id_list.length; i++) {
for(String id : id_list) {
if(userReportList.get(id_list[i]).contains(id)) reportedUserCount.replace(id, reportedUserCount.get(id) + 1);
}
}
for(int i = 0; i < id_list.length; i++) {
if(reportedUserCount.get(id_list[i]) >= k) {
banList.add(id_list[i]);
}
}
for(int i = 0; i < id_list.length; i++) {
for(int j = 0; j < banList.size(); j++) {
if(userReportList.get(id_list[i]).contains(banList.get(j))) {
answer[i]++;
}
}
}
return answer;
}
}
다른 사람의 풀이를 통해 새로운 방법을 알게 되었다.
바로 한 줄로 리스트의 중복을 제거한 방식이었는데, 다음과 같다.
List<String> list = Arrays.stream(report).distinct().collect(Collectors.toList());
Stream클래스에서 제공하는 distinct()메서드를 통해 중복 요소를 제거한 뒤 반환된 결과를 List로 변환한 방법인데, 몰랐던 방법이라서 기억해두려고 살짝 기록 (●'◡'●)
알고리즘 문제를 풀 때 마다 매번 느끼는 거지만, 정답 제출 후 다른 사람의 풀이를 볼 때 마다 쪼끔 부끄러워진다
그리고 지금 작성된 코드도 뭔가,, 분명 조금 더 개선할 수 있을 것 같은데 시간이 좀 더 지난 뒤 다시 풀어봐야겠다 히히
더 더 열심히 공부해야겠다... ╰(‵□′)╯🔥🔥
'Basic > 알고리즘' 카테고리의 다른 글
소수점 자르기, 백분율 구하기 / 백준 4344번 (0) | 2021.08.02 |
---|---|
JAVA sort() , copyOfRange() / 프로그래머스 K번째 수 (0) | 2021.07.28 |
백준알고리즘 15552번 / Scanner, BufferedReader (0) | 2021.07.18 |