프로그래머스 - 신고결과받기(Swift)

프로그래머스 - 신고결과받기(Swift)

 

 

 문제 설명

 

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

프로그래머스

 

 

 나의 풀이

 

닐 코얼레싱(nil일떄 ?? 연산자를 통해 기본값을 제공하는 것)에 익숙하지 않아서 if let을 도배한 흔적이 보인다..

 

import Foundation

func solution(_ id_list:[String], _ report:[String], _ k:Int) -> [Int] {
    
    var dictionaryReport: [String: [String]] = [:]
    var checkReported: [String: Int] = [:]
    
    for comp in report {
        let people = comp.components(separatedBy: " ")
        let reporter = people[0]
        let reportedPerson = people[1]
        if let res = dictionaryReport[reporter]?.contains(reportedPerson)
        {
            if res != false {
                continue
            }
        }
        
        
        if let res = dictionaryReport[reporter] {
            dictionaryReport[reporter]!.append(reportedPerson)
        } else {
            dictionaryReport[reporter] = [reportedPerson]
        }
        
        if let res = checkReported[reportedPerson] {
            checkReported[reportedPerson]! += 1 
        } else {
            checkReported[reportedPerson] = 1    
        }
    }
    
    return id_list.map { 
        if let arr = dictionaryReport[$0] {
            let cnt = arr.filter{ checkReported[$0]! >= k }.count
            return cnt
        }
        return 0
    }
}

 

if-let 부분만 닐 코얼레싱으로 바꿔도 코드가 훨씬 간결해진다.

 

import Foundation

func solution(_ id_list:[String], _ report:[String], _ k:Int) -> [Int] {
    
    var dictionaryReport: [String: [String]] = [:]
    var checkReported: [String: Int] = [:]
    
    for comp in report {
        let people = comp.components(separatedBy: " ")
        let reporter = people[0]
        let reportedPerson = people[1]
        if let res = dictionaryReport[reporter]?.contains(reportedPerson)
        {
            if res != false {
                continue
            }
        }
        // 변경된 코드
        dictionaryReport[reporter] = (dictionaryReport[reporter] ?? []) + [reportedPerson]
        checkReported[reportedPerson] = (checkReported[reportedPerson] ?? 0) + 1
    }
    
    return id_list.map { 
        if let arr = dictionaryReport[$0] {
            let cnt = arr.filter{ checkReported[$0]! >= k }.count
            return cnt
        }
        return 0
    }
}

 

 

 

 다른 사람의 풀이

 

배울게 많은 코드이다. 거의 모든 경우를 닐 코어레싱을 사용했고 reduce를 이용해서 신고한 사람이 어떤 사람을 신고했는지, 그 사람은 몇번 지목되었는지를 깔끔하게 코드로 작성하셨다.

 

import Foundation

func solution(_ id_list:[String], _ report:[String], _ k:Int) -> [Int] {
    
    var reported: [String: Int] = [:]
    var user: [String: [String]] = [:]
    
    for r in Set(report) {
        let results = r.components(separatedBy: " ")
        user[results[0]] = (user[results[0]] ?? []) + [results[1]]
        reported[results[1]] = (reported[results[1]] ?? 0) + 1
    }
    
    return id_list.map { id in
        return (user[id] ?? []).reduce(0) {
            $0 + ((reported[$1] ?? 0) >= k ? 1 : 0)
        }
    }
}

 

 피드백

 

1. if-let 바인딩은 값이 nil이 아닌 경우에 코드 블록에 들어오게 된다. 즉 false라는 값도 코드블록에 들어오기 때문에 if와 혼동하지 말자.

2. if-let 바인딩을 사용하기 전에 닐 코얼레싱으로 해결이 가능한지 고민해본다.

3. reduce는 기본적으로 배열을 숫자로 만들어주는 함수이다. 딕셔너리에 값이 없을 때는 빈 배열을 초기값으로 둬서 진행하면 된다.