[백준][코틀린] 1700 멀티탭 스케줄링

문제 내용

문제

기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.

예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면, 

키보드
헤어드라이기
핸드폰 충전기
디지털 카메라 충전기
키보드
헤어드라이기
키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다. 

입력

첫 줄에는 멀티탭 구멍의 개수 N (1 ≤ N ≤ 100)과 전기 용품의 총 사용횟수 K (1 ≤ K ≤ 100)가 정수로 주어진다. 두 번째 줄에는 전기용품의 이름이 K 이하의 자연수로 사용 순서대로 주어진다. 각 줄의 모든 정수 사이는 공백문자로 구분되어 있다. 

출력

하나씩 플러그를 빼는 최소의 횟수를 출력하시오. 

예제 입력	예제 출력
2 7 2 3 2 3 1 2 7	2

문제 풀이

plug 리스트에서 하나씩 읽어들여서 문제를 푸는데 

1. 이미 꽂혀 있으면 다음 플러그를 확인한다.
2. 꽂혀 있지 않다면 비어 있는 곳이 있는지 확인한다.
   1. 비어 있는 곳이 있다면 그 곳에 넣어준다.
   2. 비어있는 곳이 없다면 다음으로 올 플러그들을 검사하면서, 나오는 순서대로 리스트에 넣는다. (이때, 한번 리스트에 넣었으면 더 넣지 않는다.)
      1. 이때, 다시는 안나오는 플러그가 있다면, 그 플러그를 제거해준다. 
      2. 지금 꽂혀있는 플러그 모두가 다시 나온다면, 가장 마지막에 사용될 플러그를 제거해준다. 
      3. 위의 조건문이 끝이나면 새로운 콘센트의 사용을 표시해주고, result 값 (플러그 빼는 횟수)을 1증가시킨다.

</>̆̈ 코드

import java.util.*

fun main() {
    val br = System.`in`.bufferedReader()
    val (n, k) = br.readLine().split(" ").map { it.toInt() }
    val plugs = StringTokenizer(br.readLine()).run { IntArray(k) {nextToken().toInt()} }
    val used = BooleanArray(k+1)
    var result = 0

    for ((index, plug) in plugs.withIndex()) {
        if (used[plug]) continue // 이미 꽂혀있음
        if (used.count{it} < n) used[plug] = true // 비어 있는 곳이 있다면 넣어준다.
        else {
            val list = mutableListOf<Int>()
            for (next in index + 1 until k) {
                if (used[plugs[next]] && plugs[next] !in list) list.add(plugs[next])
            }

            if (list.size != n) {
                // 나중에 사용되지 않을 콘센트를 제거해준다.
                for ((nextPlug, use) in used.withIndex()) {
                    if (use && nextPlug !in list) {used[nextPlug] = false; break}
                }
            } else used[list.last()] = false // 가장 마지막에 사용될 콘센트 제거

            used[plug] = true
            result += 1
        }
    }

    print(result)
}

링크

1700번: 멀티탭 스케줄링