본문 바로가기
코딩테스트/이것이 취업을 위한 코딩테스트다(Python)

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

by cogito21_cpp 2024. 7. 19.
반응형
 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

 

반응형