1157번: 단어 공부

문제 출처 : https://www.acmicpc.net/problem/1157

1157번: 단어 공부

언어 : Kotlin

 

문제 설명 :

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

• 첫째 줄에 알파벳 대소문자로 이루어진 단어가 주어진다. 주어지는 단어의 길이는 1,000,000을 넘지 않는다.
• 첫째 줄에 이 단어에서 가장 많이 사용된 알파벳을 대문자로 출력한다. 단, 가장 많이 사용된 알파벳이 여러 개 존재하는 경우에는 ?를 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 128MB

 

입출력 예 :

입력	출력
Mississipi	?
zZa	Z
z	Z
baaa	A

풀이 : 

import java.io.*

fun main(args: Array<String>) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val wordGroup = br.readLine().chunked(1).groupBy { it.uppercase() }
    val max = wordGroup.maxOf { it.value.size }

    bw.write(if (wordGroup.filter { it.value.size >= max }.size >= 2) "?" else wordGroup.filter { it.value.size >= max }.values.flatten()[0].uppercase())
    bw.flush()
    bw.close()
}