[백준] 1339번: 단어 수학 - Kotlin[코틀린]

문제

 

1339번: 단어 수학

 

---

풀이

 

 이 문제에서는 각 알파벳에 가중치를 계산하고, 그 가중치에 따라 알파벳에 숫자를 할당해주어 계산하면 최대값이 계산된다.

 

 먼저 26개의 알파벳에 대한 가중치를 저장하는 배열을 선언한다. 입력받은 문자열에서 각 알파벳의 자릿수에 따라 10의 배수로 가중치를 부여한다. 예를 들어 예제 입력 2에서 GCF는 G: 100, C: 10, F: 1,의 가중치가 부여되고, ACDEB는 A: 10000, C: 1000, D: 100, E: 10, B: 1의 가중치를 부여되어 알파벳 배열에 저장한다.

 

 이후 알파벳 배열을 오름차순으로 정렬한다. 정렬된 배열을 기반으로 각 알파벳에 할당된 가중치를 순서대로 9부터 0까지 곱하여 더하면 최대값을 얻을 수 있다. 예제 입력 2의 경우에 가중치 순서로 정렬하면 A(10000), C(1010), D(100), G(100), E(10), B(1), F(1) 가 되고 순서대로 숫자를 할당하면 90000 + 8080 + 700 + 600 + 50 + 4 + 3 이 되어 99,437이 답이 된다. 또한 예제 입력의 문자열에서 GCF + ACDEB = 683 + 98754 = 99437이 되어 정답인 것을 확인할 수 있다.

 

---

코드

 

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()
 
    val num = br.readLine().toInt()
    val arr = Array(num) { br.readLine() }
    val alphabet = IntArray(26)
 
    for (i in 0 until num) {
        var tmp = 1
        for (j in arr[i].length - 1 downTo 0) {
            alphabet[arr[i][j].code - 65] += tmp
            tmp *= 10
        }
    }
 
    alphabet.sort()
 
    var idx = 9
    var sum = 0
 
    for(i in 25 downTo 0) {
        if(alphabet[i] == 0) break
        sum += alphabet[i]*idx--
    }
 
    bw.write("$sum")
    bw.flush()
    bw.close()
    br.close()
}