5026번: 박사 과정

문제 출처 : https://www.acmicpc.net/problem/5026

5026번: 박사 과정

언어 : Kotlin

 

문제 설명 :

동혁이는 박사 학위 논문을 쓰던 중 두 수를 더하는 방법을 까먹었다. 동혁이는 덧셈 문제와 컴퓨터 과학 문제로 이루어진 문제지를 풀어야 군면제를 받을 수 있다.

문제지의 덧셈 문제는 "a+b"와 같은 형식이고, 컴퓨터 과학 문제는 "P=NP" 하나이다. 동혁이의 문제지가 주어졌을 때, 답을 모두 구하는 프로그램을 작성하시오. 

 

입력 :

첫째 줄에 문제의 개수 N이 주어진다. (1 ≤ N ≤ 1000) 다음 N개 줄에는 "a+b"형식의 덧셈 문제나 "P=NP"가 주어다. a,b ∈ [0,1000]이며 a와 b는 정수이다.

 

출력 :

P=NP가 문제인 경우에는 skipped를, 덧셈 문제인 경우에는 덧셈한 결과를 출력한다.

 

제한 사항 :

• 시간 제한 : 1초
• 메모리 제한 : 128MB

입출력 예 :

입력	출력
4 2+2 1+2 P=NP 0+0	4 3 skipped 0

 

풀이 : 

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val sb = StringBuilder()
    val n = readLine().toInt()
    repeat(n) {
        readLine().also {
            if (it == "P=NP") {
                sb.appendLine("skipped")
            } else {
                val (a, b) = it.split("+").map { it.toInt() }
                sb.appendLine("${a + b}")
            }
        }
    }

    bw.write(sb.toString())
    bw.flush()
    bw.close()
}