코딩테스트 시 필요한 파이썬 문법

간단한 문법 + array / list(stack) / dictionary(map)

# for문과 if문의 scope 범위는 :와 들여쓰기로 구분한다
for 0 in range(n):
	print('0부터 n까지 반복')

for i in range(n):
	print('i부터 n까지 반복')
    
for i in range(2, n):
	print('i부터 n까지 반복하는데 2부터 시작')


# 입력
in_int = int(input())
in_str = str(input())

# 빠른 입력
##맨 상단에 import sys
in_rapid = int(sys.stdin.readline())


# array가 2차원 배열이라면
for x, y in array:
    print(x) # array[x][0]
    print(y) # array[x][1]
    

# 문자열 찾기
text = 'goodness'
word1 = 'good'
word2 = 'ness'

if word1 in text:
	print('goodness에서 good이라는 문자열 찾음')
    
if text.find(word2) == 0:
	print('goodness에서 ness이라는 문자열이 위치한 index 번호 출력. 없으면 -1')


# split 비슷한거
# array[j][0:len(array[i])]
# array의 j번째항을 0:array[i] 길이에 맞춰 가져온다

# 배열/리스트 선언
array = [] # 1차원
array = [[]] # 2차원
array = [[] for _ in range(N)]
# 2차원 배열/리스트 선언 및 초기화
array = [ [0 for _ in range(N)] for _ in range(N)]

array.append('a') # 추가
array.remove('a') # 삭제
# 활용
if array[0] == array[1][0:3]]:
    print('배열의 0번째 index 길이에 1번째 index 문자열 길이를 맞춰서 가져와 비교')
    print('두번째 []가 의미하는 것은 문자열의 길이를 의미')
 
# 배열 3인자 사용법 - ex) Array[::]
# https://blog.wonkyunglee.io/3
 
# map은 파이썬에서 dictionary
dic = {} # 선언
dic['a'] = a # 추가
b = dic['a'] # value 가져오는 문법

 

자료구조 관련

# 힙
import heapq

heapq = []
heapq.heappush(hp, item) # item을 hp에 추가.
heapq.heappop(hp) # hp에서 가장 작은 원소를 pop 한 후, return. empty일 경우, IndexError 호출.
heapq.heapify(x) # list x를 heap으로 변환.

 

주기적으로 추가합니다.

마지막 수정일 :  22.07.18