[python]string에서 입력 받기

알고리즘 문제를 푸는 과정에서 실행할 때마다 테케를 콘솔로 입력 받아서 처리하기가 여간 귀찮은게 아니다.

이번 포스트에서는 입력의 방향을 keyboard에서 str로 변경해서 입력의 편의를 도모하는 방법을 살펴보자.

java에서는..

java에서는 테스트 과정에서 StringReader를 사용하고 제출할 때는 InputStreamReader로 변경해주면 된다.

static String src = "입력 내용"
static BufferedReader input = new BufferedReader(new InputStreamReader(System.in));

public static void main(String[] args) throws IOException {
	// 제출 시 주석 처리
	input = new BufferedReader(new StringReader(src));
	// do something..
}

 

python에서는

python에서는 StringIo를 이용해서 동일한 처리를 할 수 있다.

import sys
from io import StringIO

src = """테케 내용"""

input = sys.stdin
# 제출 시 주석 처리
input = StringIO(src)

total = int(input.readline())
for i in range(9):
    total-=int(input.readline());

print(total)

즉 위와 같이 작성 후 제출 시점에는 다시 StringIO 대신 sys.stdin을 이용하면 된다.