[프로그래머스] (Swift) Kakao 신고 결과 받기

 

문제 링크

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

코딩테스트 연습 - 신고 결과 받기

 

문제 풀이

우선 신고 당한사람을 기준으로 신고 한사람에게 메일을 보내는 것이기 때문에 dictionary를 설정할 때 "신고당한사람을 Key"로 두고, "신고 한 사람을 Value"로 두었다. 그래서 내가 선언하 변수들은 아래와 같다.

• reported_report 딕셔너리 : 신고당한사람key, 신고한사람value
• 얼마나 신고를 당했는지 세어줄 count 딕셔너리 → 이게 곧 정답
• 주어진 report 는 "신고한사람 신고당한사람"의 형태이기 때문에 나누어서 딕셔너리의 key와 value 로 각각 저장해줄 것임

이것을 코드로 구현하면 다음과 같다.

아, 그리고 가장 중요한사실 중 하나는, 내가 A라는사람을 6번 신고했더라도, count는 1만 된다는 것이다. 바로 이건 중복이 안된다는 사실! 그럼 Array 가 아니라 set으로 설정하여 중복을 제거해줘야한다. 그래서 reported_report 의 value는 set으로 설정했다.

1. 입력받은 것들을 변수로 정리

var reported_report: [String:Set<String>] = [:]
var idIdx:[String:Int] = [:]
var count = Array(repeating: 0, count: id_list.count)

 

2. report를 쪼개서 각자의 위치의 dic에 저장

for (i,id) in id_list.enumerated() {
        idIdx[id] = i
}

for r in report {
    let split = r.split(separator: " ")
    let reported = String(split[1])
    let report = String(split[0])

    if reported_report[reported] == nil {
        reported_report[reported] = [report]
    } else {
        reported_report[reported]!.insert(report)
    }
}

 

3. 주어진 K와 비교하여 count

for r in reported_report {
        if r.value.count >= k {
            for i in r.value {
                count[idIdx[i]!] += 1
            }
        }
    }

 

 

최종 코드

func solution(_ id_list:[String], _ report:[String], _ k:Int) -> [Int] {

    var reported_report: [String:Set<String>] = [:]
    var idIdx:[String:Int] = [:]
    var count = Array(repeating: 0, count: id_list.count)
    
    for (i,id) in id_list.enumerated() {
            idIdx[id] = i
        }
    
    for r in report {
        let split = r.split(separator: " ")
        let reported = String(split[1])
        let report = String(split[0])
        
        if reported_report[reported] == nil {
            reported_report[reported] = [report]
        } else {
            reported_report[reported]!.insert(report)
        }
    }
    
    for r in reported_report {
        if r.value.count >= k {
            for i in r.value {
                count[idIdx[i]!] += 1
            }
        }
    }
    return count
}