[Python]파이썬 알고리즘 풀 때 기초적인 skill들

알고리즘 초보자로서 흔히 백준 알고리즘 문제를 풀 때 기본적인 스킬 혹은 문법을 몰라 계속 에러가 나는 경우가 허다했다.. 그래서 쌩기초이면서 특히 알고리즘 문제 풀 때 필요한 파이썬 문법들을 정리하고자 한다.

(풀다가 계속 추가될 수 있음 주의!)

1. split()

주로 여러 변수에 값을 할당할 때 자주 쓰인다. 괄호 안을 공백으로 두고 공백을 기준으로 변수를 나눈다.

a, b = input().split()
print(int(a)+int(b)

2. map(function, iterable data)

a, b = map(int,input().split())
print(a+b)

3. end="" sep=""

end는 말 그대로 끝에, sep은 seperation의 약자로 분리하여 출력한다.

year=2022
month=01
date=02
print(year,month,date,sep="/" end=" ")
print("happy new year")
 
#2022/01/02 happy new year

4. cf. 백준 15552

주로 입력을 할 때 input()을 쓰는 경우가 많은데, 파이썬은 느린 언어(다른 언어에 비해)라 코딩테스트를 볼 때 런타임에러에 걸릴 수 있다.

그럴 때 시간 단축을 위해서 sys 모듈 안의 sys.stdin.readline()을 이용한다.

import sys #모듈 임포트
 
x=int(sys.stdin.readline()) #input()대신 시간 효율화
 
for i in range(x):
    a,b=map(int,sys.stdin.readline().split())
    print(a+b)