[Kotlin, B1] 백준 20632번 유니대전 퀴즈쇼

출처: unsplash.com

 

유니대전 퀴즈쇼(20632번)

Bronze 1

#구현 #문자열

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

 

문제 내용

 

 

문제 접근

정답자의 채팅이 올라오고 그 이전에 정답자와 같은 채팅을 친 사람들이 몇 명인지 구해야 한다.

입력은 시간 순서대로 주어지고 정답자가 입력되고 나서 이전에 채팅을 친 사람들 중 정답자와 같은 채팅을 친 사람을 구해야 한다.

그러면 입력을 받을 때 정답자가 입력되면 정답자가 몇 번째 채팅에서 입력되었는지, 정답은 무엇인지 확인을 해줘야 한다.

 

그리고 입력받은 채팅들과 사용자 이름을 모두 List로 2차원 List에 저장해 준다.

2차원 List를 거꾸로 돌게 되면 반복의 마지막이 가장 먼저 입력된 값이 되고 가장 빨리 올라온 채팅이 되기 때문에 정답자 이전에 정답을 올린 사용자가 몇 명인지 알 수 있다.

 

 

문제 해결 코드

import java.io.BufferedReader
import java.io.InputStreamReader

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val (case, correctPerson) = readLine().split(" ")
    val chattingInfo = mutableListOf<List<String>>()
    var correctIndex = 0
    var correct = ""

    for (i in 0 until case.toInt()) {
        val (person, chatting) = readLine().split(" ")

        if (person == correctPerson) {
            correctIndex = i
            correct = chatting
        }
        chattingInfo.add(listOf(person, chatting))
    }

    var disappoint = 0
    for (i in correctIndex - 1 downTo 0) {
        if (chattingInfo[i].last() == correct) disappoint++
    }

    println(disappoint)
}

 

문제 풀이

채팅을 입력된 순서대로 chattingInfo에 저장한다. 그리고 저장하다가 도중에 정답자가 입력되면 정답 채팅(correct)과 몇 번째 채팅(correctIndex)인지 저장한다.

마지막에 반복을 역순으로 진행해서 정답자 이전의 채팅 중 정답자와 같은 채팅을 친 사람만 구해준다.

 

 

문제 해결 과정

단순 구현 문제.

map을 이용하면 더 쉬울 것 같긴 했지만 map이 입력 순으로 잘 들어오는지 알 수 없어서 2중 List를 사용했다.