[알고리즘] 2. 코딩테스트를 위한 파이썬 라이브러리

Index

1. 내장 함수

2. itertools

3. heapq

4. bisect

5. collections/math

6. 참고자료

---

1. 내장 함수 

내장 함수

- 기본 입출력 함수부터 정렬 함수까지 기본적인 함수들을 제공

- sum(list), min(list), max(list), eval(문자열 수식), sorted(list, reverse=True, key=lambda x: x[1])

 

2. itertools

itertools

- 반복되는 형태의 데이터를 처리하기 위한 기능 제공

- 순열과 조합 기능 제공

- 순열: 서로 다른 n개에서 서로 다른 r개를 선택하여 일렬로 나열하는 것

- 조합: 서로 다른 n개에서 순서에 상관 없이 서로 다른 r개를 선택하는 것

# 순열
import itertools import permutations
permutations(list, repeat)

# 조합
import itertools import combinations
combinations(data, repeat)

# 중복 순열과 중복 조합
from itertools import product
product(data, repeat)

from itertools import combinations_with_replacement
combinations_with_replacement(data,repeat)

 

3.  heapq

heapq

- 우선 순위 큐를 구현하기 위해 사용

 

4.  bisect

bisect

- 이진 탐색 기능을 제공

 

5. collections / math

collections

- deque, counter등의 자료구조 포함

- counter는 등장 횟수를 세는 기능을 제공

from collections import Counter
result = Counter(data)
result['key']

import math
math.gcd(num,num)

---

참고 자료

[Video: 동빈나의 이코테 2021] https://www.youtube.com/watch?v=m-9pAwq1o3w&list=PLRx0vPvlEmdAghTr5mXQxGpHjWqSz0dgC&index=3