[Python] 백준, 파이썬 입력받기 (feat.input, split, map)

반짝반짝 알고리즘, 인프준

 
 
 
 
 
 
 
 

파이썬 입력받기

프로그래머스는 오롯이 문제에 집중할 수 있고 UI가 깔끔한 반면 백준 온라인 저지 사이트는 문제를 유형별로 묶어서 풀 수 있으며 다양한 장점이 있지만 직접 입출력 처리를 해줘야 한다는 단점이 있습니다.
 
오늘은 파이썬 입력에 대해 참고하기 위해 글을 작성해보려 합니다. 한 단어, 구분자가 있는 한 줄, 여러 줄 등 다양한 입력 케이스들에 대해서 소개해보겠습니다.
 
 
 
 
 
 
 

---

 
 
 
 
 
 
 

한 단어 입력

A = int(input())

print(A)

 
 
 
 

한 줄 입력 - 구분자, 띄어쓰기 미포함

# input = 54321, 모두 더해서 출력
print( sum(map(int, input())) ) # 15

 
 
 
 

한 줄 입력 - 구분자, 띄어쓰기 포함

# splits at space
A, B = map(int, input().split())

# splits at comma
A, B = map(int, input().split(','))

print(A + B)

 
 
 
 

두 줄 입력

A = int(input())
B = int(input())

print(A + B)

 
 
 
 

여러 줄 입력, 몇 줄이 입력될지 아는 경우

N = int(input)

for i in range(N):
    A, B = map(int, input().split())
    
    print(A + B)

 
 
 
 

여러 줄 입력, 몇 줄이 입력될지 모르는 경우

while True:
    try:
        A, B = map(int, input().split())
        
        print(A + B)
    except:
        break

 
 
 
 

여러 줄 입력, 마지막 행을 아는 경우

# 마지막 행은 0이 두 개이다.
While True:
	A, B = map(int, input().split())
    
    if A == 0 and B == 0:
    	break
    else:
    	print(A + B)

 
 
 
 
 
 
 

---

 
 
 
 
 
 
 

Reference

백준 온라인 저지 사이트, roum02님의 블로그