README.md

파이썬 기본 연산자 시간 복잡도

https://dev.plusblog.co.kr/42

itertools

from itertools

중복조합

combinations_with_replacement(iter, r) : A~E 5명의 후보가 있다. 중복을 허용해 2명의 대표를 뽑는 방법

from itertools import combinations_with_replacement

list(combinations_with_replacement('ABC', 2)) #결과 : 'AA' 'AB' 'AC' 'BB' 'BC' 'CC' 

중복순열

product(iter, r) : 1~5 숫자카드가 있다. 중복을 허용해 2자리수를 만드는 방법.

from itertools import product

list(product('ABC', repeat=2)) #결과 : 'AA' 'AB' 'AC' 'BA' 'BB' 'BC' 'CA' 'CB' 'CC'

heap

import heapq

기존 리스트를 힙으로 변환

heapq.heapify( 리스트 )

a = [4,1,3,10]
heapq.heapify(a)  #결과: a는 [1, 4, 3, 10]

-> 힙에서 인덱스 0이 최솟값이라고 해서, 인덱스 1에 두 번째, 인덱스 2에 세 번째로 작은 원소가 있는 것이 아니다. 두 번째로 작은 원소를 얻으려면 heappop()을 통해 최솟값을 삭제한 후 heap[0]로 접근하는 방법을 사용하거나, 인덱스 1을 인덱스 2와 비교하는 방법을 사용해야 한다. (최소값만 보장)

hash

from collections import defaultdict

defaultdict(int)
defaultdict(list)

딕셔너리에 items() 메서드를 사용해주면 {"key" : value}의 형태를 [(key, value)]의 형태로 만들어 준다.

기본문법

d[k] = v  # (k, v) 추가
d.pop(k)  # key가 k인 쌍 찾아 제거
k in d    # 현재 hashmap에 key가 k인 쌍이 있는지 확인

key 가능 type immutable한 값 -> int, char 등의 primitive type과 string, tuple 등 가능, mutable한 list, dict 등의 type은 가변적이기 때문에 불가

items, keys, values

print(d.items())   #dict_items([(1, 3), (2, 4), (-1, 6)])
print(d.keys())    #dict_keys([1, 2, -1])
print(d.values())  #dict_values([3, 4, 6])

key를 기준으로 딕셔너리 정렬

sorted(d)   #['blue', 'green', 'red']: list 형태로 return

value를 기준으로 딕셔너리 정렬

sorted(d.items(), key=lambda x : x[1])   #[('red', 3), ('blue', 3), ('green', 1)]: list 형태로 return

value가 list인 경우

for key in d:
    d[key].sort()

set

기본문법

s=set()
s.add(e)     # 데이터 e 추가
s.remove(e)  # 데이터 중 숫자 e 찾아 제거
e in s       # 현재 hashset에 숫자 e가 들어 있는지 확인

합집합, 교집합, 차집합

# a: {1,2,6}, b: {2,6,9}
a|b   #{1,2,6,9}
a&b   #{2,6}
a-b   #{1}

bisect

이진 탐색 모듈

특정 값 위치

해당 값이 위치한 마지막 index 표시

import bisect
mylist = [1, 2, 3, 3, 7, 9, 11, 33]
print(bisect.bisect(mylist, 3)) # 4 (bisect = bisect_right)

target 값 존재하지 않아도 index 출력하므로 target 값과 동일한지 check하는 로직 필요요

for _ in range(m):
    target = int(input())
    index = bisect.bisect_right(lst, target)
    if lst[index-1] == target:
        print(index)
    else:
        print(-1)

특정 값 갯수

import bisect
mylist = [1, 2, 3, 3, 7, 9, 11, 33]
print(bisect.bisect_right(mylist, 3) - bisect.bisect_left(mylist, 3))

Math

import math

최대공약수

math.gcd( 숫자들 )

math.gcd(12,8)  //결과: 4

Counter

요소가 딕셔너리 키로 저장되고 개수가 딕셔너리 값으로 저장되는 컬렉션

from collections import Counter

데이터 개수가 많은 순으로 정렬된 배열을 리턴

Counter( [리스트] or [문자열] ).most_common( 출력갯수 )

문자열

문자 등장 횟수

"별똥별".count("별") #결과: 2

문자열 정렬

문자열을 정렬하고 싶을 땐 sort한 다음에 join으로 묶어줘야 한다. (sort하면 리스트를 반환한다)

orders[i] = "".join(sorted(orders[i]))

문자열 수정

문자열 값을 변경하고 싶을땐 list로 변환 후 idx에 해당하는 값을 수정하고 join으로 묶어준다.

찾기

find() 함수는 왼쪽부터 찾고, rfind() 함수는 오른쪽부터 찾는다.

• 문자가 없다면 -1을 반환
• 문자가 여러 개 있다면 맨 마지막 인덱스 반환

"별똥별".find("별") #결과: 0
"별똥별".rfind("별") #결과: 2

대소문자 변환

• 문자열.swapcase() # 소문자는 대문자로, 대문자는 소문자로 서로 바꿔주는 메서드

"abc".upper() #결과: "ABC"
"ABC".lower() #결과: "abc"
"aBc".swapcase() #결과: "AbC"

공백 제거

• 문자열.strip()

문자 변경

• 문자열.replace(old, new)

아스키 코드 변환

• 문자 -> 아스키 코드 : ord('A')
• 아스키 코드 -> 문자 : chr(65)

진수 변환

10진수 -> n진수

#1. bin(x)[2:]): 내장함수 bin 사용
#2. 
def change_10to2(n):
    tmp = []
    while n > 0:
        tmp.append(str(n%2))
        n //= 2
    tmp = tmp[::-1]
    return tmp

n진수 -> 10진수

int(문자열 값, n)

zip 함수

각 iterables 의 요소들을 모으는 이터레이터를 만듭니다. 튜플의 이터레이터를 돌려주는데, i 번째 튜플은 각 인자로 전달된 시퀀스나 이터러블의 i 번째 요소를 포함합니다.

사용 예 #1 - 여러 개의 Iterable 동시에 순회할 때 사용

list1 = [1, 2, 3, 4]
list2 = [100, 120, 30, 300]
list3 = [392, 2, 33, 1]
answer = []
for number1, number2, number3 in zip(list1, list2, list3):
   print(number1 + number2 + number3)

사용 예 #2 - Key 리스트와 Value 리스트로 딕셔너리 생성하기

animals = ['cat', 'dog', 'lion']
sounds = ['meow', 'woof', 'roar']
answer = dict(zip(animals, sounds)) # {'cat': 'meow', 'dog': 'woof', 'lion': 'roar'}

List Comprehension

for 문과 if문을 한번에

mylist = [3, 2, 6, 7]
answer = [number**2 for number in mylist if number % 2 == 0]

for-else문

flag 대신 사용. for loop가 break없이 빠져나올 경우 else문 실행. continue는 break 처럼 동작하지 않음.

포매팅

a = 3
print(f'{a} 입니다')

inf

가장 큰 수 표현

inf = int(1e9)
print(inf > inf)   #False
print(inf+1 > inf) #True

sys

프로그램 종료 함수수

import sys
sys.exit(0)