[파이썬 풀이] 백준 15552.빠른 A+B

문제   

본격적으로 for문 문제를 풀기 전에 주의해야 할 점이 있다. 입출력 방식이 느리면 여러 줄을 입력받거나 출력할 때 시간초과가 날 수 있다는 점이다.

C++을 사용하고 있고 cin/cout을 사용하고자 한다면, cin.tie(NULL)과 sync_with_stdio(false)를 둘 다 적용해 주고, endl 대신 개행문자(\n)를 쓰자. 단, 이렇게 하면 더 이상 scanf/printf/puts/getchar/putchar 등 C의 입출력 방식을 사용하면 안 된다.

Java를 사용하고 있다면, Scanner와 System.out.println 대신 BufferedReader와 BufferedWriter를 사용할 수 있다. BufferedWriter.flush는 맨 마지막에 한 번만 하면 된다.

Python을 사용하고 있다면, input 대신 sys.stdin.readline을 사용할 수 있다. 단, 이때는 맨 끝의 개행문자까지 같이 입력받기 때문에 문자열을 저장하고 싶을 경우 .rstrip()을 추가로 해 주는 것이 좋다.

또한 입력과 출력 스트림은 별개이므로, 테스트케이스를 전부 입력받아서 저장한 뒤 전부 출력할 필요는 없다. 테스트케이스를 하나 받은 뒤 하나 출력해도 된다.

 

입력   

첫 줄에 테스트케이스의 개수 T가 주어진다. T는 최대 1,000,000이다. 다음 T줄에는 각각 두 정수 A와 B가 주어진다. A와 B는 1 이상, 1,000 이하이다.

 

출력   

각 테스트케이스마다 A+B를 한 줄에 하나씩 순서대로 출력한다.

 

코드   

import sys

t=int(sys.stdin.readline())
sumn=[]

for i in range(t):
    sumn.append(list(map(int,sys.stdin.readline().split())))

for j in range(t):
    print(sumn[j][0]+sumn[j][1])

 

관련 개념 정리   

sys.stdin.readline()

input함수와 같은 기능을 하는 코드입니다.

단, 여러줄을 입력 받아야 하는 경우 input함수보다 더 적은 시간으로 구동이 가능하기 때문에 해당 코드를 사용하는 것이 더 좋습니다.

해당 코드에 대해서는 추후에 따로 포스팅을....!

 

 

split

또한 값을 입력받을 때, 한 번에 여러 값을 입력받을 수 있습니다.
그때 사용할 수 있는 함수가 바로 split 함수입니다.

split할 변수 이름.split('구분자')

 

만약 split()일 경우, space, tap, enter와 같은 공백을 기준으로 문자열을 나눕니다.

 

 

리스트

리스트란 여러개의 숫자나 문자를 하나의 변수에 저장할 때 사용합니다.

리스트이름=[데이터1,데이터2,데이터3...]
#b=[]와 같은 비어있는 리스트도 생성 가능!

또한 인덱싱이 가능합니다. 인덱스값은 0부터 시작합니다.

d=[1,2,3]
print(d[0]) #1
print(d[1]) #2
print(d[2]) #3

 

 

append

append함수는 리스트의 마지막에 사용자가 원하는 요소를 추가하는 함수입니다.

리스트.append(추가 할 요소)

위와 같은 형태로 사용 가능합니다.

 

 

map

map함수는 입력받은 자료형의 각 요소를 함수가 적용된 값으로 변형 시켜주는 함수입니다.

map(함수, 반복 가능한 자료형)

위와 같은 형태로 사용 가능합니다.

예시)

def dddd(x):
    return x+3

print(list(map(dddd,[2,5,1,8])))

# [5, 8, 4, 11]

 

 

list

list함수는 반복 가능한 자료형을 받아 리스트로 만들어 주는 함수입니다.

list(반복 가능한 자료형)

위와 같은 형태로 사용 가능합니다.

print(list("배고파"))
#['배', '고', '파']

print(list((1,6,3)))
#[1, 6, 3]

print(list([3,5,2]))
#[3, 5, 2]

문자열을 입력 받은 경우에는 한글자씩 쪼개서 리스트를 만듭니다.

숫자로 된 리스트를 만들기 위해서는 ,(쉼표) 로 구분해서 ()괄호 안에 입력하면 됩니다.

list에 list를 입력한 경우에는 아무런 변화 없이 입력한 list를 그대로 retrun합니다.

 

 

for문

특정 코드들을 원하는 횟수 만큼 반복 시킬 때 사용하는것이 for문입니다.

for문의 기본 구조는 아래와 같습니다.

for i in 리스트(or 문자열, 튜플):
	실행할 코드

리스트,문자열,튜플의 첫 글자나 요소부터 마지막 글자나 요소까지 차례대로 변수에 대입됩니다.

 

for문의 범위와 변수값을 숫자로 정하고 싶을때는 아래와 같이 range함수를 사용합니다.

for i in range(시작숫자,종료숫자,step):
	실행할 코드

step은 얼만큼씩 변수를 증가 시킬지 정하는 것 입니다.

시작숫자와 step은 생략되어도 됩니다.

 

 

산술 연산자

x+y	x 더하기 y
x-y	x 빼기 y
x*y	x 곱하기 y
x/y	x 나누기 y
x//y	x 나누기 y의 몫
x%y	x 나누기 y의 나머지
x**y	x의 y제곱