[ps] 여러 테스트 케이스 입력 받기

spring·2024년 11월 17일
0

PS문제에서 테스트케이스 개수를 입력받지 않고, 개별로 입력을 받는 경우 테스트가 번거로운 단점이 있다.

예를들어 입력n이 들어오면 2xn을 출력하는 문제의 예시가 아래와 같을 때,

예제 입력예제 출력
12
612
100200

아래와 같은 코드는 매번 입력을 바꿔주어야 한다. (input.txt를 리다이렉션 설정시)

#include<iostream>
using namespace std;
int main(){
	int n;
    cin >> n;
    cout << 2*n << endl;
    return 0;
}

반면, 아래와 같이 작성시 여러 테스트케이스들을 한번에 돌려볼 수 있다.

#include<iostream>
using namespace std;
int main(){
	int n;
    while(cin >> n){
    	cout << 2*n << endl;
    }
    return 0;
}

kotlin은 아래와 같이 사용할 수 있다.

fun main() {
    while (true) {
        val input = readLine() ?: break
        val n = input.toInt()
        println(2 * n)
    }
}
profile
Researcher & Developer @ NAVER Corp | Designer @ HONGIK Univ.

0개의 댓글