[Swift/Algorithm] dictionary default value 사용하기

다음 글은 간단한 코드 조각을 남기기 위한 글이다. 알고리즘 문제를 풀거나 개발을 하다 보면 dictionary를 사용하는 순간이 생각보다 많다. (..는 예전에는 array만 주구장창 썼었지.. 적절한 자료구조를 적재적소에 쓰는 것도 실력인 것을 날이 갈수록 느낀다..) 뭐 여튼 dictionary를 사용하게 되는 가장 기본적인 예제가 아마 빈도를 계산하는 부분이 아닐까 싶다.

 

예를 들어서 다음과 같이 영화 장르가 있다고 생각해보자.

let genres = ["comedy", "romance", "comedy" , "comedy" , "action"]

읽어보면 코미디 장르가 3편, 로맨스 장르가 1편, 액션 장르가 1편 있다는 것을 초등학생도 알 것이다. 근데 컴퓨터는 모른다. 알았다면 문제를 좀 더 편하게 해결할 수 있었겠지.. 각 장르가 몇 편 존재하는지 알기 위해서 로직을 짜야 할텐데 직관적으로 dictionary 자료구조를 써야겠다는 느낌이 온다. (array도 쓸 수 있겠지만 dictionary가 더 적합한 것 같다. 순서 고려할 필요 없이 그냥 개수만 세면 되니깐)

 

개수를 세기 위해 이전에 짰던 코드를 잠깐 들고 와보겠다.

let genres = ["comedy", "romance", "comedy" , "comedy" , "action"]
var dict = [String: Int]()

genres.forEach {
    if let count = dict[$0] {
        dict[$0] = count + 1
    } else {
        dict[$0] = 1
    }
}

// Prints
// ["action": 1, "comedy": 3, "romance": 1]

빈 dictionary를 만들어주고 genre 배열을 돌면서 key가 존재하지 않았던 것은 새로 추가해주면서 value를 1로 만들어주고, 기존에 값이 존재하던 것은 value를 1씩 증가시켜준다. dictionary를 출력해보면 각 genre에 해당하는 영화의 개수를 알 수 있다.

 

이렇게만 해도 되겠지만, 분기 처리를 하기가 싫다. 그냥 새롭게 key를 추가할때는 기본값이 있으면 좋을 것 같다. 역시나 Subscript에 default 값을 정해주는 형태가 있었다. (역시 Apple.. 🍎 ... 생각해보니까 Python에도 있다 ㅋ..)

 

key가 존재하지 않는 값에 접근하려고 하면 default 값으로 정해둔 값으로 그냥 key를 생성해주는 것이다.

// 개선 1: dictionary default 값 활용
let genre = ["comedy", "romance", "comedy" , "comedy" , "action"]
var dict = [String: Int]()

genre.forEach { dict[$0, default: 0] += 1 }

print(dict)

// Prints
// ["action": 1, "comedy": 3, "romance": 1]

코드가 깔끔해졌고 내 마음도 편해졌다. 최근에 Swift의 주어진 문법을 잘 활용해서 코드를 잘 작성하도록 연습 중인데 재미있는 것 같다. 고차함수를 잘 활용할 수 있는 사람이 되고 싶다. 물론 코드 스타일을 개선하는데에만 치우치지는 말자.