[프로그래머스] Swift 오픈채팅방 2019 kakao

문제 :

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

코딩테스트 연습 - 오픈채팅방

분석 :

확실히 swift는 c++보다 문자열 파싱에 있어서 유용한것 같다 ! 그래서 아이디어만 생각나면 구현하기 수월하다는 생각을 다시한번 하게 되었다. swift 짱 ..👍

이 문제는 문자열 파싱 및 구현 문제로, 오픈 채팅방에 들어왔을때, 나갔을때, 이름을 변경했을때의 명령어를 처리해주면 된다. 기존 톡방 참여자가 이름을 변경하고 다시 들어오게 되면 톡방에는 전 이름 -> 현재 이름으로 같이 변경이 이루어져야 한다. 

따라서 처음엔 for문을 돌면서 enter와 change 명령어 시, inputDictionary에 저장되어 있는 유저라면 출력 배열을 돌며 하나씩

replacingOccurrences(of:\(예전 이름), with: \(현재 이름))으로 대체하고자 하였다. 하지만 25부터 주르륵 시간초과가 난다... 

그래서 다른 블로그를 참고해 아이디어를 얻었다.

for문을 돌기 전에 inputDictionary에 마지막 이름을 초기값으로 가지게 한 뒤, 명령어를 처리해주면 enter, change시 출력문 내 같은 아이디 가진 사람을 찾아 대체해주지 않아도 된다는 점 ..!

이를 해결하니 시간 초과가 해결 되었다. 해시 맵 방식이 for문보다 빠름을 느낄 수 있었다.

 

Swift 시간초과 코드 :

//
//  main.swift
//  SOMA👩🏻‍💻
//
//  Created by JoSoJeong on 2022/05/02.
//

import Foundation
var inputDic:[String: String] = [:]
var answer = [(String, String)]()

func reName(_ id: String, _ name: String) {
    if inputDic.keys.contains(id) { // id가 저장된 사람이라면
        for (idx, i) in answer.enumerated() {
            if i.1 == id {
                answer[idx].0 = i.0.replacingOccurrences(of: "\(inputDic[i.1]!)", with: "\(name)")
            }
        }
    }
    inputDic[id] = name
    
}

func solution(_ record:[String]) -> [String]{
    for line in record {
        let l = line.components(separatedBy: " ")
        let command = l[0]
        let id = l[1]

        switch command {
        case "Enter":
            let name = l[2]
            reName(id, name)
            answer.append(("\(name)님이 들어왔습니다.", id))
        case "Leave":
            
            answer.append(("\(inputDic[id]!)님이 나갔습니다.", id))
        case "Change":
            let name = l[2]
            reName(id, name)
        default:
            print("")
        }
    }
    return answer.map { $0.0 }
}

 

Swift 해결 코드 :

//
//  main.swift
//  SOMA👩🏻‍💻
//
//  Created by JoSoJeong on 2022/05/02.
//

import Foundation

func solution(_ record:[String]) -> [String]{
    var inputDic:[String: String] = [:]
    var answer = [String]()
    for line in record {
        let l = line.components(separatedBy: " ")
        if l.count > 2 {
            inputDic[l[1]] = l[2]
        }
    }
    
    for line in record {
        let l = line.components(separatedBy: " ")
        let command = l[0]
        let id = l[1]
        
        switch command {
        case "Enter":
            answer.append("\(inputDic[id]!)님이 들어왔습니다.")
        case "Leave":
            answer.append("\(inputDic[id]!)님이 나갔습니다.")
        default:
            print("")
        }
    }
    return answer
}

결과 :