[Kotlin, B1] 백준 25593번 근무 지옥에 빠진 푸앙이 (Small)

출처: unsplash.com

 

근무 지옥에 빠진 푸앙이 (Small)(25593번)

Bronze 1

#구현 #자료 구조 #문자열 #해시를 사용한 집합과 맵

 

문제 내용

 

 

문제 접근

문제를 이해하기만 하면 된다.

사진 1-1

입력 형식이 주의 개수인 N에서 4를 곱한 줄 만큼 입력이 주어지고 있다.

위 그림에서 눈치가 빠르신 분들은 대강 눈치를 챌 것이다.

첫 줄의 주의 개수를 입력받고 문자열들이 주의 개수(4)의 4를 곱한 만큼의 수(16)만큼 입력된다는 사실을 볼 수 있다.

 

또한 한 줄에 공백을 기준으로 7개로 나뉘어서 입력된다. 또한 문제 설명에서 4 교대 근무를 선다고 되어 있다.

오전 8 ~ 오후 12, 오후 12 ~ 오후 6, 오후 6 ~ 오후 10, 오후 10 ~ 오전 8 해서 4번의 교대가 이루어진다.

- - - - - pangyo puang
- - - - - sally boss
- - - - - leonard brown
- - - - - edward edward

입력값에서 4개의 줄을 떼왔다. 이 값들이 첫 주의 근무표를 나타내는 것이다.

왜 이 값들이 첫 주의 데이터가 되는지는 힌트를 보면 알 수 있다.

5번째 줄의 첫 번째 근무자가 puang이다. 힌트를 보면 둘째 주 첫날이라고 언급되어 있다. 다음 시간대의 근무자는 이전 시간대의 다음 줄에 입력되는 것을 볼 수 있다.

이를 통해 2가지 사실을 알 수 있다.

 

1. 입력은 1주가 4줄로 입력된다.

2. 1줄에 각 근무에 해당하는 시간대에 근무하는 근무자들을 입력받는다. 첫 주의 입력값에서 1번째 줄은 첫 주의 오전 8시부터 오후 12시까지의 근무자 목록을 입력받는 것이다.

다시 입력 예제를 이용해 정리를 해 보겠다.

// 입력되는 근무자는 공백을 기준으로 7개를 입력받는다(각 요일에 근무하는 사람을 의미함)
- - - - - pangyo puang      // 오전 8시 ~ 오후 12시 근무자(4시간)
- - - - - sally boss        // 오후 12시 ~ 오후 6시 근무자(6시간)
- - - - - leonard brown     // 오후 6시 ~ 오후 10시 근무자(4시간)
- - - - - edward edward     // 오후 10시 ~ 오전 8시 근무자(10시간)

 

입력값들을 이용해 이제 문제를 해결하기 위해서는 N의 * 4만큼 반복해서 값을 입력받고 각 줄에 맞는 근무 타임을 지정해 어떤 사람이 얼마만큼 근무를 했는지 더해가면서 계산하면 된다.

Map을 이용해서 근무자와 근무 시간을 정의해 주면 각 근무자에 대한 근무 시간을 편하게 설정할 수 있다.

 

줄에 맞는 근무 타임을 지정하려면 각 근무 타임에 몇 시간만큼 일하는지 배열로 저장하고, 입력 순서에 각 배열을 4로 나눈 나머지값을 index로 설정해서 적절한 근무 타임을 설정해 준다.

그전에 Map의 key가 존재하는지 확인해야 한다. Map에 key가 데이터가 존재하는지 확인하기 위해서는 containsKey()를 사용한다.

if (!worker.containsKey(week[j])) {
    worker[week[j]] = hour[i % 4]
} else {
    worker[week[j]] = worker[week[j]]!! + hour[i % 4]
}

 

 

그리고 가장 적게 일한 사람과 가장 많이 일한 사람의 차이가 12보다 크면 공평하지 않은 근무가 된다. 조건은 아래와 같이 설정할 수 있다.

if (maxWork - minWork > 12) println("No")
else println("Yes")

그리고 근무자가 존재하지 않을 때는 Map이 비어 있으면 근무자가 존재하지 않다는 뜻이다. isEmpty()로 비어 있는지 확인해 준다.

 

 

문제 해결 코드

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

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val weeks = readLine().toInt()
    val worker = mutableMapOf<String, Int>()
    val hour = listOf(10, 4, 6, 4)

    for (i in 1 .. weeks * 4) {
        val week = readLine().split(" ")

        for (j in week.indices) {
            if (week[j] == "-") continue
            else {
                if (!worker.containsKey(week[j])) {
                    worker[week[j]] = hour[i % 4]
                } else {
                    worker[week[j]] = worker[week[j]]!! + hour[i % 4]
                }
            }
        }
    }

    if (worker.isEmpty()) {
        println("Yes")
        return
    }

    val maxWork = worker.values.max()
    val minWork = worker.values.min()

    if (maxWork - minWork > 12) println("No")
    else println("Yes")
}

 

문제 풀이

입력받은 주의 개수에 * 4를 해서 입력받아줘야 한다.

근무자 정보를 입력받아 근무자가 없는지 확인한다.

if (week[j] == "-") continue

근무자가 없으면 넘긴다.

 

근무자가 있으면 Map에 key가 존재하는지 확인해야 한다.

위에서 언급했다시피 containsKey()를 이용해야 한다.

if (!worker.containsKey(week[j])) {
    worker[week[j]] = hour[i % 4]
} else {
    worker[week[j]] = worker[week[j]]!! + hour[i % 4]
}

키가 없으면 키를 새로 생성해 초기값을 근무 시간으로 설정해 준다.

그렇지 않으면 기존 키에 근무 시간을 더해 준다.

 

근무 시간이 저장된 Map에서 value들의 최댓값과 최솟값을 구해 준다.

최댓값은 가장 근무를 많이 한 사람의 근무 시간이고,

최솟값은 가장 근무를 적게 한 사람의 근무 시간이다.

 

두 값의 차가 12를 넘어가면 공평하지 않은 근무가 된다. 문제 접근을 참고하자.

또한 입력을 받았는데도 Map이 비어 있으면 근무자가 존재하지 않는다는 뜻이다. 문제 설명에서는 아무도 근무하지 않으면 공평한 것으로 간주한다.

if (worker.isEmpty()) {
    println("Yes")
    return
}

 

 

문제 해결 과정

문제 풀이 시간보다 문제 이해 시간이 더 오래 걸렸던 문제이다.

예제 1의 입력값의 첫 번째 주의 값과 4번째 주의 값의 빈 근무자에만 빈 근무자(-)가 있어서 이 값이 무슨 관계가 있나? 해서 헛다리를 짚었었다.

이해를 한 번 하니까 왜 그런 생각을 했을까 하는 생각이 들긴 했다.

구현은 그냥 문자열로도 가능하긴 한데, Map을 사용하는 게 정신건강에 이로울 것이다.