[백준] 1157번: 단어 공부

문제 링크

 

문제

알파벳 대소문자로 된 단어가 주어지면, 이 단어에서 가장 많이 사용된 알파벳이 무엇인지 알아내는 프로그램을 작성하시오. 단, 대문자와 소문자를 구분하지 않는다.

 

입력

첫째 줄에 알파벳 대소문자로 이루어진 단어가 주어진다. 주어지는 단어의 길이는 1,000,000을 넘지 않는다.

 

출력

첫째 줄에 이 단어에서 가장 많이 사용된 알파벳을 대문자로 출력한다. 단, 가장 많이 사용된 알파벳이 여러 개 존재하는 경우에는 ?를 출력한다.

 

소깡이 풀이

import Foundation

extension Character {
    var asciiIndex: Int {
        return Int(self.asciiValue!) - 97
    }
}

var charDic: [Int] = [Int](repeating: 0, count: 26)

readLine()!.map{ Character($0.lowercased()) }.forEach { char in
    charDic[char.asciiIndex] += 1
}

let max = charDic.max()
var resultArray: [Int] = []
var index: Int = 0

for i in charDic.indices {
    if charDic[i] == max {
        resultArray.append(i)
        index = i
    }
}

if resultArray.count == 1 {
    let answer = String(UnicodeScalar(65 + index)!) // 대문자
    print(answer)
} else {
    print("?")
}

 

이 문제의 핵심은 .. 아스키 코드가 아니라 .. 딕셔너리를 잘 활용할 수 있는가? 이다.

딕셔너리를 활용한 풀이의 경우, 아래와 같다.

var check: [Character: Int] = [:]
let word = readLine()!.uppercased()

for char in word {
    if let value = check[char] {
        check[char] = value + 1
        
    } else {
        check[char] = 1
    }
}

let max = check.values.max()!
let most = check.filter { $0.value == max }

if most.count != 1 {
    print("?")
} else {
    print(most.first!.key)
}

 

알파벳 모든 문자에 대한 경우를 관리하는 것이 아니라, 입력 받은 문자와 그 문자의 개수만 딕셔너리로 관리하고

값을 비교해서 가장 큰 값일 때의 키를 반환하면 된다.