백준 - 단어수학(Swift)

백준 - 단어수학(Swift)

 

 

 문제 설명

 

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

1339번: 단어 수학

 

 나의 풀이

 

 

처음에는 각 문자열의 count에 따라서 높은 순서대로 값을 부여했다. 이렇게하면 예제들은 전부 맞지만 반례들이 존재했다.

예를 들어 ABC, D, D로 주어졌을때 내가 한 방식은 A:9, B:8, C:7, D:6이 된다. 하지만 D가 7인 경우가 최대의 수가 된다. 

즉 이 방식은 정답이 될 수 없는 풀이이다.(문제를 간과했다.)

 

그래서 다른사람의 풀이를 보고 알게 된 솔루션은 각 자리의 수를 딕셔너리에 저장하고 큰 값부터 숫자를 부여하면 되는 것이다.

 

예시로 ABA라는 값이 들어오게 된다면 첫 번째 A는 10의 2승 B는 10의 1승 세 번째 A는 10의 3승이 되고 A는 101, B는 10이된다.

A는 B보다 크기 때문에 9를 B는 8을 주어 101*9 + 10*8 = 989가 된다.

 

 

import Foundation

let n = Int(readLine()!)!
var dic: [String: Int] = [:]
for _ in 0..<n {
    var word = readLine()!
    var val = Int(pow(10, Double(word.count-1)))
    for x in word {
        if dic[String(x)] == nil {
            dic[String(x)] = val
        } else {
            dic[String(x)]! += val
        }
        val = val/10
    }
}
var result = 0
var cnt = 9
for num in dic.values.sorted(by : >) {
    result += cnt * num
    cnt -= 1
}

print(result)

 

 

 피드백

 

처음 내 풀이(틀린 풀이)를 하면서 몇 가지 알게 된 점이 있다.

 

문자열 배열에서 문자열을 하나씩 줄여나갈때 마지막은 없어지는 것이 아닌 ""인 채로 남아있는 것이다. 그래서 따로 ""가 되었을때 배열 전체에서 remove할 수 있도록 설정했다.

 

[["abcd", "efg"]]라는 2차원 배열이 있을때 flatMap을 하면 ["abcd", "efg"]가 된다.

 

readLine으로 문자열을 받을때 꼭 map으로 변하려고 노력하지 말자. 문자열을 배열로 만드는게 아닌 문자열 그대로 푸는 문제일 수도 있다. 너무 사고의 폭을 좁히지 말아야 겠다.