[코테 공부(Kotlin) - 프로그래머스] 신고 결과 받기
카테고리: CodingTest
태그: CodingTest, kakao, Kotlin, lv1, programmers
문제
문제 출처
프로그래머스 : https://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_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 합니다.
풀이
나의 풀이
문제 풀이 흐름은 다음과 같다.
- 중복된 신고 제거
- 신고 당한사람 - 신고한 사람 목록 쌍을 만든다.
- id - 메일 수신 횟수 쌍을 만든다.
- 신고한 사람 수가 k명 보다 많다면 신고한 사람들의 메일 수신 횟수 + 1
- id 순서에 맞게 배열에 담아 출력
HashSet을 이용해 중복 신고를 제거하고 HashMap을 이용해 2,3의 쌍들을 만들었다.
작성한 코드는 다음과 같다.
fun solution(id_list: Array<String>, report: Array<String>, k: Int): IntArray {
val answer = arrayListOf<Int>()
val hashSet = HashSet<String>()
val map = HashMap<String, ArrayList<String>>()
val mailMap = HashMap<String, Int>()
for (re in report){
hashSet.add(re)
}
for (re in hashSet){
val repo = re.split(" ")
if (map.containsKey(repo[1])){
map[repo[1]]?.add(repo[0])
}else{
val arraylist = ArrayList<String>()
arraylist.add(repo[0])
map[repo[1]] = arraylist
}
}
for (m in map){
val value = m.value
if (value.size >= k){
for (v in value){
val v_cnt = mailMap.getOrDefault(v,0)
mailMap[v] = v_cnt + 1
}
}
}
for (id in id_list){
val cnt = mailMap.getOrDefault(id,0)
answer.add(cnt)
}
return answer.toIntArray()
}
다른 사람의 풀이
풀이 방식은 비슷하긴 하다만.. 그저 감탄
코드를 읽어보면서 나름의 주석을 달아보았다.
fun solution(id_list: Array<String>, report: Array<String>, k: Int): IntArray =
report.map { it.split(" ") } //신고를 분할
.groupBy { it[1] } //신고 당한 id로 그룹화
.asSequence() //시퀀스로 변경(컬렉션 크기 클 때 연산 속도 up)
.map { it.value.distinct() } //중복 신고 제거
.filter { it.size >= k } // 신고 횟수가 k번 이상인
.flatten() //시퀀스 안의 List<List<String>> 을 하나의 List<String>으로 변환
.map { it[0] } //신고 한 id로 새로운 시퀀스 Sequence<String> 생성
.groupingBy { it } // 같은 id끼리 묶기
.eachCount() // 각 id의 개수 map 생성
.run { id_list.map { getOrDefault(it, 0) }.toIntArray() } //id 순서 맞게 배열로 만듬