문제
https://school.programmers.co.kr/learn/courses/30/lessons/92334
문제 설명
신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.
- 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
- 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
- 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
- k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
- 유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.
다음은 전체 유저 목록이 ["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 함수를 완성해주세요.
- 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 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.
입출력 예id_listreportkresult
["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 합니다.
- 정확성 테스트 : 10초
풀이방법
문제에서 요구한대로 풀어주면되는데, HashSet과 HashMap을 함께 써야하는게 나한테는 좀 까다롭게 느껴졌다.
우선, 아래와 같은 표 모양을 만들기위해 HashMap에 HashSet을 넣어주었다.(같은 사용자를 신고할시, 한번에 신고로 취급되기 때문에.)
report를 공백단위로 잘라서 신고한 사람을 key값으로 삼아 넣어주었다.
이때 HashMap은 put으로, HashSet은 add으로 값을 넣어주어야한다.
HashMap<String, HashSet<String>> map=new HashMap<>();
for(int i=0; i<id_list.length; i++){
map.put(id_list[i], new HashSet<String>());
}
for(int i=0; i<report.length; i++){
String[] a=report[i].split(" ");
map.get(a[0]).add(a[1]);
}
그 다음 아래와 같은 구조를 만들어주기 위해, String과 Integer를 갖는 HashMap(map2)을 만들어서 값을 넣어준다.
앞서 넣어준 map에 있는 HashSet을 돌리면서, map2에 있는 value값을 높여주면된다.
HashMap<String, Integer> map2=new HashMap<>();
for(int i=0; i<id_list.length; i++){
map2.put(id_list[i], 0);
}
for(int i=0; i<id_list.length; i++){
HashSet<String> set=map.get(id_list[i]);
for(String s : set) {
map2.put(s,map2.get(s)+1);
}
}
k보다 큰 횟수로 신고당한 id들을 찾아서 stopUser에 넣어주고 최종적으로 map을 돌리면서, HashSet에 StopUser가 있으면, 값을 1씩 늘려줘서 출력한다.
HashMap<String, Integer> map2=new HashMap<>();
for(int i=0; i<id_list.length; i++){
map2.put(id_list[i], 0);
}
for(int i=0; i<id_list.length; i++){
HashSet<String> set=map.get(id_list[i]);
for(String s : set) {
map2.put(s,map2.get(s)+1);
}
}
전체코드는 아래와 같다.
코드
import java.util.*;
class Solution {
public int[] solution(String[] id_list, String[] report, int k) {
HashMap<String, HashSet<String>> map=new HashMap<>();
for(int i=0; i<id_list.length; i++){
map.put(id_list[i], new HashSet<String>());
}
for(int i=0; i<report.length; i++){
String[] a=report[i].split(" ");
map.get(a[0]).add(a[1]);
}
HashMap<String, Integer> map2=new HashMap<>();
for(int i=0; i<id_list.length; i++){
map2.put(id_list[i], 0);
}
for(int i=0; i<id_list.length; i++){
HashSet<String> set=map.get(id_list[i]);
for(String s : set) {
map2.put(s,map2.get(s)+1);
}
}
Set<String> stopUser=new HashSet<>();
for(int i=0; i<id_list.length; i++){
if(map2.get(id_list[i])>=k){
stopUser.add(id_list[i]);
}
}
int[] answer = new int[id_list.length];
for(int i=0; i<id_list.length; i++){
HashSet<String> set=map.get(id_list[i]);
for(String s : set) {
for(String s1: stopUser){
if(s.equals(s1)){
answer[i]++;
}
}
}
}
System.out.println(Arrays.toString(answer));
return answer;
}
}
추가
HashMap, HashSet이 헷갈려서 정리를 좀 해보자면,
- 선언
- HashMap<String> h=new HashMap<>();
- Set<String> s=new HashSet<>();
- 값 넣기
- h.put("");
- s.add("");
- 값 얻기
- HashMap은 키로 얻는다.
- map.get("A");
- HashSet은 값을 하나하나 가져오려면 아래와같이 쓴다.
- for(String eachs : s) {};
- HashMap은 키로 얻는다.
댓글