Python 으로 코딩테스트 볼 때 유용한 code 정리

BFS 문제에서의 queue 구현 

from collections import deque 

def bfs():
	queue = deque([])
    ...

 

sort 내림차순

a = [1,3,2,6,5,10,8]
a.sort(reverse=True)

 

sort with lambda 조건 

a = [(100,4),(70,3), (80,1), (50,5),(60,2)]
a.sort(key= lambda x:x[1])

'''
결과 
[(80, 1), (60, 2), (70, 3), (100, 4), (50, 5)]
'''

 

Permutation (순열) 구현하기 

import itertools

a = ['A', 'B', 'C']
nPr = itertools.permutations(a,2)

'''
결과 
[('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'C'), ('C', 'A'), ('C', 'B')]
'''

 

Combination (조합) 구현하기 

import itertools

b = ['A', 'B', 'C']
nCr = itertools.combinations(b, 2)

'''
결과 
[('A', 'B'), ('A', 'C'), ('B', 'C')]
'''