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

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

문제 풀이

프로그래머스에서 Level1 문제 중 제일 마지막으로 남았던 문제.. 뭔가 그래서 더 어려운? 느낌으로 다가오지 않았나.. 싶기도.. 1차 시도는 답이 나왔지만 계속 시간 초과가 나왔다. Dictionary, Set을 조만간 다시 정리해야겠다.

1차 시도

처음 시도한 코드는 시간 초과가 나왔다. 중복 신고를 막기 위해 report를 Set으로 변환하여 저장하고 처리했는데, 아무래도 신고 당한 횟수를 체크하면서 이메일 결과를 처리하는 과정에 있어서 뭔가 문제가 있지 않았나 싶다...

2차 시도

먼저 emailResult에 이메일 결과를 [String: Int] 형태로 저장하는데, String에는 유저 이름, Int에는 받을 이메일 수를 저장한다. for 문으로 유저 이름을 추가했고, 이메일 수는 모두 0으로 초기화했다.

 

var emailResult = [String: Int]() // 이메일 결과 저장
for i in 0..<id_list.count {
    emailResult.updateValue(0, forKey: id_list[i])
}

 

 

 

userInfoDic이라는 딕셔너리를 만들어서 Key에는 신고 당한 유저 ID를, Value에는 신고 당한 유저 ID를 신고한 모든 유저 ID를 저장한다.

var userInfoDic = [String: [String]]() // [신고 당한 유저 ID: [신고한 유저 ID]]

 

 

 

for 문을 돌면서 report를 통해 처리하는데, 여기서 userInfoDic을 구해준다.

먼저 report를 userId와 targetId로 나누고, 2가지 경우로 나눠서 체크한다.

 

 

userInfoDic에 targetId를 가진 키가 존재할 경우

• 한 유저가 동일한 유저를 여러 번 신고하는 것을 막기 위해, targetId를 가진 키가 값으로 userId를 갖고 있는지 확인한다. 갖고 있지 않을 경우 userId를 추가해준다.

 

userInfoDic에 targetId를 가진 키가 존재하지 않을 경우

• 해당 키 (targetId)를 추가하고, 값으로 userId (신고한 유저 ID)를 추가한다.

for i in report {
    let reportArr = i.components(separatedBy: " ").map { String($0) }
    let userId = reportArr[0] // 신고한 유저 ID
    let targetId = reportArr[1] // 신고 당한 유저 ID
    
    if userInfoDic[targetId] == nil {
        userInfoDic.updateValue([userId], forKey: targetId)
    } else {
        if !userInfoDic[targetId]!.contains(userId) {
            userInfoDic[targetId]!.append(userId)
        }
    }
}

 

 

그리고 이메일 수를 구하기 위해 userInfoDic.keys로 for문을 돌린다.

이 때 userInfoDic의 키가 갖고 있는 값이 k개 이상이라면, 해당 키를 갖는 유저들을 찾아서 emailResult의 이메일 수를 +1 해준다.

for targetId in userInfoDic.keys {
    if userInfoDic[targetId]!.count >= k {
        for n in userInfoDic[targetId]! {
            emailResult[n]! += 1
        }
    }
}

 

 

마지막으로, id_list에서 유저 목록을 차례대로 for 문으로 돌리면서 emailResult[id]의 값을 새로운 ans 배열에 차례대로 추가해서 ans를 리턴하면 정답이다.

var ans = [Int]()
for id in id_list {
    ans.append(emailResult[id]!)
}

 

소스 코드 (Swift)

//
//  신고결과받기.swift
//  Programmers
//
//  Created by 안상희 on 2022/02/06.
//

import Foundation

func solution(_ id_list:[String], _ report:[String], _ k:Int) -> [Int] {
    var emailResult = [String: Int]() // 이메일 결과 저장
    for i in 0..<id_list.count {
        emailResult.updateValue(0, forKey: id_list[i])
    }
    
    var userInfoDic = [String: [String]]() // [신고 당한 유저 ID: [신고한 유저 ID]]
    for i in report {
        let reportArr = i.components(separatedBy: " ").map { String($0) }
        let userId = reportArr[0] // 신고한 유저 ID
        let targetId = reportArr[1] // 신고 당한 유저 ID
        
        if userInfoDic[targetId] == nil {
            userInfoDic.updateValue([userId], forKey: targetId)
        } else {
            if !userInfoDic[targetId]!.contains(userId) {
                userInfoDic[targetId]!.append(userId)
            }
        }
    }
    
    for targetId in userInfoDic.keys {
        if userInfoDic[targetId]!.count >= k {
            for n in userInfoDic[targetId]! {
                emailResult[n]! += 1
            }
        }
    }
    
    var ans = [Int]()
    for id in id_list {
        ans.append(emailResult[id]!)
    }
    
    return ans
}