[JAVA] 프로그래머스 신고 결과 받기 | HashMap | HashSet

 

~목차~

문제

문제 해결 포인트

작성 코드

---

 

문제

https://school.programmers.co.kr/learn/courses/30/lessons/92334

프로그래머스

 

 

문제 해결 포인트

- 파이썬처럼 배열에 다른 타입을 넣을 수 없기 때문에 헤맸으나 Map 클래스의 value에 Set 타입을 넣음

- 각 신고이력을 돌며 신고 당한 사람이 key값, 신고 한 사람들이 value

- 각 유저 별 정지 메일을 카운트 하기 위해 LinkedHashMap을 정의(단순 HashMap으론 순서를 장담받을 수 없음)

- 신고받은 사람들을 순회하며 k보다 많은 사람에게 신고 받았다면 신고한 사람들의 메일 개수를 하나씩 추가한다.

 - 최종적으로 stream을 이용하여 int형으로 변환 후 배열을 반환

 

 

작성 코드

import java.util.*;
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
        // 유저 별 누구에게 신고받았는지
        Map<String, Set<String>> map = new HashMap<>();
        for(String repo: report){
            String[] singo = repo.split(" ");
            Set<String> user = map.getOrDefault(singo[1], new HashSet<String>());
            user.add(singo[0]);  // 주의
            map.put(singo[1], user);
        }
        Map<String, Integer> countmap = new LinkedHashMap();
        for(String id: id_list){
            countmap.put(id, 0);
        }
        for(Map.Entry<String, Set<String>> entry: map.entrySet())
        {
            if(entry.getValue().size() >= k){
                for(String name: entry.getValue()){
                    countmap.put(name, countmap.get(name)+1);
                }
            }
        }
        
        return countmap.values().stream().mapToInt(i->i).toArray();
    }
}

 

 

 

도움이 되셨다면 좋아요 눌러주세요💚

 

---

REFERENCE

https://lnsideout.tistory.com/entry/JAVA-%EC%9E%90%EB%B0%94-%ED%8A%B9%EC%A0%95-%EB%AC%B8%EC%9E%90%EC%97%B4-%ED%8F%AC%ED%95%A8-%ED%99%95%EC%9D%B8-%EB%B0%8F-%EC%B0%BE%EA%B8%B0containsindexofmatches

 

https://recordsoflife.tistory.com/314