Python 코딩 테스트 기본 템플릿(백준, 프로그래머스, SWEA)

백준

import sys

# 재귀 depth를 늘려줌
sys.setrecursionlimit(10 ** 9)
# 파일 입출력
r = sys.stdin.readline
# int 하나 입력 받기
length = int(r())
# list 입력 받기
list1 = list(map(int, r().split()))
# 문자열 입력받기
string1 = r().rstrip()

백준은 입력부터 출력까지 전부 다 작성해줘야 한다. 하지만 표준 라이브러리에 해당하는 모든 것을 사용할 수 있다.

표준 입출력보단 속도가 훨씬 빠른 파일 입출력으로, 재귀 depth도 늘려준 뒤 사용한다.

 

프로그래머스

# 여기에 사용할 라이브러리를 선언해줌

def solution(answers):
    answer = []
    return answer

프로그래머스는 답이 되는 값을 반환하는 함수인 solution을 작성하는 것이 기본 골자이다. 함수 위에 사용할 라이브러리를 작성해줘도 되고, 전역 변수나 PS에 필요한 함수를 직접 작성해도 된다. 백준과 다르게 값을 입력받을 필요 없이 변수로 받아오기 때문에 파일 입출력을 따로 사용해줄 필요는 없고, 필요에 따라서 재귀 depth만 늘려주고 사용하면 된다.

 

SWEA

T = int(input())
for test_case in range(1, T + 1):
    # ///////////////////////////////////////////////////////////////////////////////////
    '''

        이 부분에 여러분의 알고리즘 구현이 들어갑니다.

    '''
    # ///////////////////////////////////////////////////////////////////////////////////

입력부터 코드를 작성해야 하는 것은 백준과 같다. 하지만 SWEA는 sys 라이브러리를 사용할 수 없어 파일 입출력, 재귀 depth를 수정하는 것이 불가능하다. 일반적으로 test case 개수를 입력받고, 그 안에 코드를 작성하게 된다.