[백준] (Swift) 20291번 - 파일 정리 (구현)

🟠 문제

https://www.acmicpc.net/problem/20291

20291번: 파일 정리

 

🟠 나의 풀이

문제는 뭐 주절주절 길지만, 확장자별로 갯수를 세어주고, 확장자명 오름차순으로 출력해주면 되는 간단한 문제다. 나는 특정 확장자별로 갯수를 세어주는 방식을 dictionary로 구현했다.

dict안에 내가 원하는 확장자명이 있다면 value에 + 1을 해주었고, 없다면 dict에 새로 생성해주는 방식이다. 그리고 dict를 key기준으로 정렬해준뒤, key와 value를 출력해주면 끝!

🟠 정답 코드

import Foundation

let n = Int(readLine()!)!
var dict = [String:Int]()

for _ in 0..<n {
    let temp = readLine()!.split(separator: ".").map{ String($0) }

    if !dict.keys.contains(temp[1]) {
        dict[temp[1]] = 1
    } else {
        dict[temp[1]]! += 1
    }
}

let sortedDict = dict.sorted { $0.0 < $1.0 }
for i in 0..<sortedDict.count {
    print("\(sortedDict[i].key) \(sortedDict[i].value)")
}