[프로그래머스][Lv. 1] 신고 결과 받기 Swift 풀이

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

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

풀이는 추후에 차차 올리도록 하겠습니다...

 

 

 

import Foundation
 
func solution(_ id_list:[String], _ report:[String], _ k:Int) -> [Int] {
    var user_dict: [String : Int] = [String : Int]()
    for id in id_list { 
        user_dict.updateValue(0, forKey: id)
    }
 
    let report_proc = Array(Set(report))    
    for rpt in report_proc {
        let slice: [String] = rpt.components(separatedBy: " ")
        var count: Int = user_dict[slice[1]]!
        user_dict.updateValue(count + 1, forKey: slice[1])
    }
    
    var report_count: [Int] = Array(repeating: 0, count: user_dict.count)
    for i in 0..<id_list.count {
        if !(k > user_dict[id_list[i]]!) {
            report_count[i] += 1
        }
    }
    
    var result_count: [Int] = Array(repeating: 0, count: user_dict.count)
    for rpt in report_proc {
        let slice: [String] = rpt.components(separatedBy: " ")
        if !(k > user_dict[slice[1]]!) { // slice[1]: count
            result_count[id_list.index(of: slice[0])!] += 1 // slice[0]: reporter 
        }
    }
 
    return result_count
}