[Python] 파이썬 코딩테스트를 위한 개념 정리

방학동안 코딩테스트 공부를 하기 위해서 두고두고 볼 겸

파이썬에 관련된 간단한 개념들을 한 페이지에 정리해보았다!

 

입력과 출력

입력

- input() : 사용자로부터 문자열 한 줄을 입력받는다.

- map() : 모든 원소에 전달한 함수 하나씩 적용한 객체 반환

 

# 튜플의 형태의 객체이므로 다음과 같이 언패킹 가능
a, b, c = map(int, input().split())
# list로 변환 가능
data = list(map(int, input().split()))

 

입력량으로 인한 수행시간 초과 오류 발생시

시스템 라이브러리에 정의된 readline() 이용

import sys
str = sys.stdin.readline().rstrip() # rstrip()으로 줄바꿈 삭제
print(str)

 

출력

print() 이용, end를 이용해 끝문자 지정가능

print("HELLO") # 출력 후 줄바꿈
print("HELLO",end=" ") #출력 후 한 칸 공백

---

코드 수행시간 측정

import time
start_time = time.time() # 측정 시작

# 프로그램 소스코드

end_time = time.time() # 측정 종료
print(f"time: {end_time - start_time}") # 수행 시간 출력
# print(f"time: {end_time - start_time:.6f}") # 수행 시간 출력

코테에서 통상적으로 5초 이내에 수행하는 것이 시간복잡도 측면에서 효율적인 코드라고 한다!

---

자료형

정수형 실수형 복소수형 문자열 리스트 튜플 딕셔너리

 

실수형

실수형 주의!
컴퓨터 시스템 내의 데이터 처리에 사요되는 것은 2진수 체계로, 실수를 정확히 표현못함

round()를 이용하기

a = .3+.6
if round(a, 4) == .9: # round() 사용하지 않으면 False
    print(True)
else:
    print(False)
    
>>> True

 

지수표현법

e 혹은 E로 지수를 표현할 수 있다. e뒤에 오는 수는 10의 지수를 표현하는데

만약 1e2 라면 10^2 = 100 이고 2e-2 라면 2*10^-2 = 0.02를 의미한다.

 

수 자료형 연산자

+ - *

/ : 나눈 결과 실수형으로 반환

// : 나눈 결과의 몫 반환

% : 나눈 결과의 나머지 반환

** : 멱수를 반환

 

문자열

- 인덱싱과 슬라이싱이 가능하다. 그러나 해당 인덱스 변경은 불가

- '+' : 문자열이 연결된다.

- '*' : 문자열과 정수와 곱하면 같은 문자열 반복

문자열 메소드

count, find, index, join, upper, lower, strip(+ l,r), replace, split

",".join("abcd") #a,b,c,d
",".join(['a', 'b', 'c', 'd']) #a,b,c,d

# 리스트 --> 문자열 변환
arr = ['a', 'b', 'c', 'd']
result1 = str(arr)
result2 = ''.join(arr)
print(type(result1), result1) # <class 'str'> ['a', 'b', 'c', 'd'] # 형태 그대로 문자열로
print(type(result2), result2) # <class 'str'> abcd

 

리스트

- 여러개의 데이터를 다루기 위한 자료형

- 하나의 리스트에 서로 다른 자료형 데이터 공존가능

- 인덱스와 슬라이싱 이용가능

- 내부의 값을 인덱스를 이용하여 접근 후 변경 가능

 

리스트 인덱싱&슬라이싱

arr = [1,2,3,4]

# 인덱싱
arr[0] # 1
arr[-1] # 4

# 슬라이싱
arr[1:3] # [2, 3]
arr[1:3:1] # [2, 3] # 위와 같은 결과
arr[::2] # [1, 3] # 전체 대상 앞부터 2칸 씩
arr[::-1] # [4, 3, 2, 1] # 역순으로 슬라이싱
arr[::-2] # [4, 2]

 

리스트 컴프리헨션

리스트 초기화시 사용하는 방법

arr = [0 for i in range(0, 4)]
print(arr) # [0, 0, 0, 0]

arr = [i for i in range(10) if i % 2 == 0]
print(arr) # # 0~9사이 짝수

# 2차원 리스트
n = 3
m = 2
arr = [n * [0] for i in range(m)]

print(arr) # [[0, 0, 0], [0, 0, 0]]

리스트 메소드

append, sort, sort(reverse=True), reverse, insert, count, remove

arr = [1,2,3,4]

arr.append(5);print(arr) 
arr.sort();print(arr) 
arr.sort(reverse=True);print(arr) 
arr.reverse();print(arr) 
arr.insert(10,1);print(arr) # 범위 넘으면 자동으로 계산
print(arr.count(1))
arr.remove(1);print(arr) # 조건만족 원소 중 가장 앞 하나 삭제

 

튜플

- 한 번 선언된 값은 변경 불가

- 리스트에 비해 공간효율적

- 활용 예시

- 서로 다른 성질의 데이터 묶어서 관리할 때  - 최단 경로 알고리즘에서 (W, Num) 형태로 사용할 때  - 해싱할 때 (데이터, 해싱 키)  - 리스트보다 메모리 효율적으로 사용할 때

 

딕셔너리

- 비순차적 자료형

- Key와 value의 쌍을 데이터로 가짐

- '변경 불가능한 자료형'을 Key로 사용 가능

- 내부적으로 해시 테이블을 이용하므로 데이터의 '조회 및 수정이 O(1)'의 시간복잡도

# 방법 1
name = '이름'

info = {
  name: 'Kassid',
  '나이': 25,
  5 : 25
}
print(info) # {'이름': 'Kassid', '나이': 25, 5: 25}

# 방법 2
info1 = dict()
info1['이름'] = 'Kassid'
info1['나이'] = 25

Key 데이터만 뽑기 : keys()

값 데이터만 뽑기 : values()

 

집합 자료형

- 중복을 허용하지 않는다.

- 순서가 없다.

- 데이터의 '조회 및 수정이 O(1)'의 시간복잡도

data = set([1, 2, 3, 4, 4, 4]) # {1, 2, 3, 4, 5}. 4중복 제거
data = {1, 1, 2, 2, 3, 4, 4, 5, 5 } # {1, 2, 3, 4, 5}

집합 자료형 연산자

a = {1, 2, 3, 4 }
b = {1, 4, 5, 6 }
# 합
a | b # {1, 2, 3, 4, 5, 6}

#교
a & b # {1, 4}

#차
a - b # {2, 3}

---

global 키워드

- 함수 내부에서 함수 밖의 변수를 사용하고자 할 때

- 단순 출력or 참조는 global 없이 참조 가능하다.

- list의 내부 메서드를 호출해 사용하는 것 또한 오류 미발생

a = 0
def func():
    global a
    a += 1
    
func();print(a) # a의 값 변경
# 1

b = 10
def func2():
    print(b+20)
func2();print(b) # 단순 값만 참조 # b의 값 미변경
# 30 10

arr = [1, 2 ,3, 4, 5]
def func3():
    arr.append(10)
func3();print(arr)
# [1, 2, 3, 4, 5, 10]

---

함수

함수의 반환 여러 개 한번에 가능

아래와 같은 형태로 여러 값을 한 번에 반환할 수 있다.

def func(a,b):
	return a+b, a-b, a*b, a/b

A,B,C,D = func(10,3)

 

lambda

다음과 같은 경우에 활용한다.

- 함수의 간단한 작성 - 함수가 한번만 쓰이고 마는 경우 가능에 한줄로 표현 가능 - 함수 자체를 입력으로 받는 함수에서 유용함

print((lambda a, b: a + b)(1, 2)) # 3

list1 = [1, 2, 3]
list2 = [4, 5, 6]
# 각각의 원소를 더하기
result = list(map(lambda a,b: a + b, list1, list2))
# [5, 7, 9]

 

---

---

유용한 팁

setrecursionlimit 설정

파이썬에서 재귀활용 문제에서 필수적으로 필요한 코드

재귀 깊이 제한을 늘려준다.

import sys
sys.setrecursionlimit(10 ** 6)