[백준/BOJ] 17952번: 과제는 끝나지 않아!

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

17952번: 과제는 끝나지 않아!

문제

성애는 이번 학기에 전공을 정말 많이 듣는다. 이로 인해 거의 매일을 과제를 하면서 보내고 있다. 그런데도 과제가 줄어들 기미가 보이지 않는데, 바로 분단위로 과제가 추가되고 있기 때문이다. 다행히 과제 제출 기한은 학기가 끝날 때까지이다. 너무나도 많은 과제를 하다가 미쳐버린 성애는 아래와 같은 규칙으로 과제를 해 나가고 있다.

1. 과제는 가장 최근에 나온 순서대로 한다. 또한 과제를 받으면 바로 시작한다.
2. 과제를 하던 도중 새로운 과제가 나온다면, 하던 과제를 중단하고 새로운 과제를 진행한다.
3. 새로운 과제가 끝났다면, 이전에 하던 과제를 이전에 하던 부분부터 이어서 한다. (성애는 기억력이 좋기 때문에 아무리 긴 시간이 지나도 본인이 하던 부분을 기억할 수 있다.)

성애는 과제를 받자마자 이 과제가 몇 분이 걸릴지 정확하게 알 수 있고, 성애가 제출한 과제는 무조건 만점을 받는다.

성애는 이번 학기에 자기가 받을 과제 점수를 예상해보고 싶다. 하지만 과제 점수를 예상하는 지금도 과제가 추가되고 있기에 여유를 부릴 수가 없다. 여러분이 성애가 받을 과제 점수를 구해주자!

입력

첫째 줄에 이번 학기가 몇 분인지를 나타내는 정수 N이 주어진다. (1 ≤ N ≤ 1,000,000)

두번째 줄부터 N줄 동안은 학기가 시작하고 N분째에 주어진 과제의 정보가 아래의 두 경우 중 하나로 주어진다.

• 1 A T: 과제의 만점은 A점이고, 성애가 이 과제를 해결하는데 T분이 걸린다. A와 T는 모두 정수이다. (1 ≤ A ≤ 100, 1 ≤ T ≤ 1,000,000)
• 0: 해당 시점에는 과제가 주어지지 않았다.

출력

성애가 받을 과제 점수를 출력한다.

해설

핵심을 집어보면 과제를 하던 도중 새로운 과제가 나오다면, 하던 과제를 중단하고 새로운 과제를 진행하는 것이다.

Stack을 이용하여 문제를 풀어보자.

 

우선 1일 때 만점과 해결하는데 걸리는 시간이 나온다.

이 둘을 스택에 넣어줄건데, 걸리는 시간은 입력을 받음과 동시에 1분을 사용하는 것이므로 1을 뺀 값을 넣어준다.

이 때 시간이 0이 된다면 문제를 해결한 것이므로 총 점수에 점수를 더해준다.

 

이제 0일 때를 보자.

해결하고 있던 문제가 있다면 그 문제를 계속하여 풀면 되므로 시간에서 1을 빼준다.

이도 1일 때 말했듯 시간이 0이 되면 총 점수에 만점을 더해주면 된다.

 

문제를 풀 때 Stack<Pair<Int, Int>>를 사용하였는데,

코틀린에서 Pair는 public final val이라서 시간에서 1을 뺄 때 peek()이 아닌 pop()을 통해 값을 뺀 후 계산을 하여

다시 Stack에 넣어주는 식으로 문제를 해결하였다.

소스 코드

import java.util.*

fun main()  = with(System.`in`.bufferedReader()) {
    val minute = readLine().toInt()
    var stack = Stack<Pair<Int, Int>>()
    var score = 0
    for(i in 0 until minute) {
        val st = StringTokenizer(readLine())
        if(st.nextToken().toInt() == 1) {
            stack.push(Pair(st.nextToken().toInt(), st.nextToken().toInt() - 1))
            if(stack.peek().second == 0) score += stack.pop().first
        }
        else {
            if(stack.empty()) continue
            val first = stack.peek().first
            val second = stack.pop().second - 1
            stack.push(Pair(first, second))
            if(stack.peek().second == 0) score += stack.pop().first
        }
    }
    println(score)
}