백준 1302) 베스트셀러

 

www.acmicpc.net/problem/1302

1302번: 베스트셀러

 

문제

김형택은 탑문고의 직원이다. 김형택은 계산대에서 계산을 하는 직원이다. 김형택은 그날 근무가 끝난 후에, 오늘 판매한 책의 제목을 보면서 가장 많이 팔린 책의 제목을 칠판에 써놓는 일도 같이 하고 있다.

오늘 하루 동안 팔린 책의 제목이 입력으로 들어왔을 때, 가장 많이 팔린 책의 제목을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 오늘 하루 동안 팔린 책의 개수 N이 주어진다. 이 값은 1,000보다 작거나 같은 자연수이다. 둘째부터 N개의 줄에 책의 제목이 입력으로 들어온다. 책의 제목의 길이는 50보다 작거나 같고, 알파벳 소문자로만 이루어져 있다.

출력

첫째 줄에 가장 많이 팔린 책의 제목을 출력한다. 만약 가장 많이 팔린 책이 여러 개일 경우에는 사전 순으로 가장 앞서는 제목을 출력한다.

 

 

---

 

해당 문제는 hashMap을 이용해서 각각의 책에 대한 권 수를 저장해서 가장 많은 수를 가진 책 이름을 출력한다.

시간이 다소 소모됐던 부분은 value가 동일한 책들을 어떻게 내림차순으로 정렬해서 key 값을 정렬하는지 관건이었다.

 

sortedBy 나 containsValue 등으로 접근했지만 원하는 결과가 출력되지 않았다.

결국 다양하게 제공되는 코틀린 함수를 써보면서 각각 어떤 기능을 가지고 있는지 알 수 있었고,

다음과 같이 풀어내게 됐다.

 

 

 

import java.io.*

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()
    val map = hashMapOf<String, Int>()
    
    repeat(n) {
        val book = br.readLine()
        if (map.containsKey(book)) {		//map에 book이 존재하면 +1
            map[book] = map[book]!! + 1
        } else {				//존재하지 않으면 1로 초기화
            map[book] = 1
        }
    }
    val max = map.values.max()			// max값을 추출
    
    val list = mutableListOf<String>()		// key를 저장하기 위한 리스트 생성
    map.forEach {
        if(it.value == max) {			// max값과 value가 같다면
            list.add(it.key)			// list에 key를 추가
        }
    }
    print(list.sorted()[0])			// 사전 순이므로 가장 앞에 있는 책 이름을 출력
}

 

 

테스트 케이스

2

b

a

결과 : a

 

5

ab

ba

ab

ba

aaaaa

결과 : ab