[프로그래머스(Java)] 신고 결과 받기 - HashMap LinkedHashMap Set Map순회하는법 Stream

 

Level. 1

 

문제

 

신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.

• 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
  • 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
  • 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 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 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.

 

 

풀이

import java.util.*;

class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        
        int[] answer = new int[id_list.length];
        
        HashMap<String, Set<String>> users = new HashMap<>();
        Map<String, Integer> count = new LinkedHashMap<>();
        
        for(int i=0; i<report.length; i++){
            String[] arr = report[i].split(" ");
            Set<String> set = users.getOrDefault(arr[1], new HashSet<>());
            set.add(arr[0]);
            users.put(arr[1], set);
        }
        
        for(String id : id_list){
            count.put(id, 0);
        }
        
        for (Map.Entry<String, Set<String>> entry : users.entrySet()) {
            if(entry.getValue().size() >= k){
                for(String s : entry.getValue()){
                    count.put(s, count.get(s) + 1);
                }
            }
        }
        
        int idx = 0;
        for (Map.Entry<String, Integer> entry : count.entrySet()) {
            answer[idx++] = entry.getValue();
        }
       
        //return count.values().stream().mapToInt(Integer::intValue).toArray();
        return answer;
    }
}

 

사용한 자료구조 

• HashMap : Key로 신고당한 유저를, value로 신고한 유저를 저장하기 위함.
• Set : 신고한 유저를 중복 없이 저장하기 위함
• LinkedHashMap : Key가 유저 id, value로 받게 되는 메일의 개수를 저장하기 위함. 순서가 있길 원했기 때문에 LinkedHashMap를 사용했다. 

해결방법

1. report배열을 탐색하며 신고받은 유저를 Key, 신고한 유저를 value로 하여 users 데이터에 추가한다. 이때 신고한 유저는 Set에 담아 추가한다.
2. LinkedHashMap 자료구조를 사용한 count에 id순으로 데이터를 저장한다. key는 id, value는 메일을 받을 횟수
3. users 데이터를 순회하며 해당 유저를 신고한 사람이 k명 이상일 경우, 유저를 신고한 사람(users.value)은 메일을 받아야 하므로 count 데이터에서 해당 key(users.value)의 value값을 +1 한다. 
4. count데이터를 answer배열에 넣는다. 

 

Map.getOrDefault

Map.getOrDefault(Object key, V defaultValue)

• 찾는 키가 존재하면 찾는 키의 값을 반환하고, 없다면 기본 값을 반환하는 메서드이다.
• key : 값을 가져와야 하는 요소의 키값
• defaultValue : 저장된 키로 매핑된 값이 없는 경우 반환되어야 하는 기본값

 

Map 순회하기

 

1. Iterator을 통해 접근하기

Iterator<String> keys = map.keySet().iterator();
while (keys.hasNext()) {
    String key = keys.next();
    map.get(key);
}

 

2. entrySet을 사용해 접근하기

for (Map.Entry<String, String> entry : map.entrySet()) {
    String key = entry.getKey();
    String value = entry.getValue();
}

 

3. keySet을 사용해 접근하기

for (String key : map.keySet()) {
    map.get(key);
}

 

 

 

 

---

다른 풀이

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.stream.Collectors;

class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        List<String> list = Arrays.stream(report).distinct().collect(Collectors.toList());
        HashMap<String, Integer> count = new HashMap<>();
        for (String s : list) {
            String target = s.split(" ")[1];
            count.put(target, count.getOrDefault(target, 0) + 1);
        }

        return Arrays.stream(id_list).map(_user -> {
            final String user = _user;
            List<String> reportList = list.stream().filter(s -> s.startsWith(user + " ")).collect(Collectors.toList());
            return reportList.stream().filter(s -> count.getOrDefault(s.split(" ")[1], 0) >= k).count();
        }).mapToInt(Long::intValue).toArray();
    }
}

 

1. stream의 distinct()를 사용해 중복을 제거한 후, collect를 사용해 List형으로 자료형을 변환한다. 
2. count 데이터에 유저와 그 유저가 신고당한 횟수를 저장한다.
3. list를 stream으로 변환하여 map을 사용해 함수가 적용된 새로운 요소로 매핑한 후, mapToInt를 사용해 스트림을 IntStream으로 변환, toArray()를 통해 배열로 변환한다. 
4. 함수에서는 startWith을 사용하여 user(신고한 유저)로 시작하는 문자열을 가진 요소만 list로 만든 후, 
   user에게 신고당한 유저 중 k번 이상 신고당한 유저의 count를 리턴한다.  

>> 나름 해석해 봤는데 stream은 정말 유용하지만 해석하기 어려운 것 같다.. 아직 Java에 익숙하지 않아서 그런가 보다~

 

 

Stream.distinct()

Stream에서 distinct() 메서드는 스트림에서 중복되는 요소들을 모두 제거해주고 새로운 스트림을 반환해준다. 

 

Stream.collect()

Stream에서 collect() 메소드는 Stream의 데이터를 변형 등의 처리를 하고 원하는 자료형으로 변환해 준다. 

• Stream의 아이템들을 List 또는 Set 자료형으로 변환
• Stream의 아이템들을 joining
• Stream의 아이템들을 Sorting하여 가장 큰 객체 리턴
• Stream의 아이템들의 평균 값을 리턴

startsWith()

boolean startsWith(String prefix)

• 대상 문자열이 특정 문자 또는 문자열로 시작하는지 체크한다. 
• 해당 문자열로 시작되는지 여부를 확인하여 true/false 값을 리턴한다.

map과 filter

• map : 요소들을 특정조건에 해당하는 값으로 변환한다.
• filter : 요소들을 조건에 따라 걸러내는 작업을 한다. 

 

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/92334?language=java 

프로그래머스

https://junghn.tistory.com/entry/JAVA-Map-getOrDefault-%EC%9D%B4%EB%9E%80-%EC%82%AC%EC%9A%A9%EB%B2%95-%EB%B0%8F-%EC%98%88%EC%A0%9C

[JAVA] Map - getOrDefault 이란? 사용법 및 예제

https://pangtrue.tistory.com/232

[Java] Map을 순회하는 방법

https://codechacha.com/ko/java8-stream-collect/

Java - Stream.collect() 사용 방법 및 예제

https://devmoony.tistory.com/136

[Java] 스트림(Stream) 개념 및 사용법 / filter, map / iterator 비교

https://dpdpwl.tistory.com/81

[Java]자바 스트림Stream(map,filter,sorted / collect,foreach)