05. 배열

🧩 <코딩테스트 합격자 되기: 파이썬 편> 1독 스터디 기록입니다.

 

 

선언

일반적인 방법

arr = [0, 0, 0, 0, 0]
arr = [0] * 5

리스트 생성자를 사용하는 방법

arr = list(range(5))
# 결과: arr = [0, 1, 2, 3, 4]

List Comprehension을 사용하는 방법

List Comprehension이란, [expression for item in iterable if condition] 구조로 리스트를 간결하고 효율적으로 생성할 수 있는 문법이다. 코드의 가독성을 높이고 일반적인 for 루프보다 빠르게 실행할 수 있다.

arr = [0 for _ in range(5)]
# 결과: arr = [0, 0, 0, 0, 0]

 

 

시간 복잡도

맨 뒤에 삽입하는 경우

해당 위치에 바로 접근할 수 있고 다른 데이터의 위치에 영향을 주지 않기 때문에 $ O(1) $이다.

맨 앞에 삽입하는 경우

기존에 있던 데이터들을 모두 한 칸씩 미루기 위한 연산이 필요하므로 $ O(N) $이다.

중간에 삽입하는 경우

삽입 위치의 뒤에 있는 데이터들을 모두 한 칸씩 미루는 연산이 필요하므로 마찬가지로 $ O(N) $이다.

 

 

고려할 점

배열은 데이터의 인덱스를 이용해서 바로 접근할 수 있다는 장점이 있다. 따라서 데이터에 자주 접근해야 한다면 효율적이다. 반면 메모리 낭비가 발생할 수도 있으므로 아래 사항을 고려해야 한다.

• 할당 가능한 메모리 크기를 확인한다. 일반적으로 정수형 1차원 배열은 1000만, 2차원 배열은 3000*3000 크기를 최대로 설정한다.
• 맨 앞 또는 중간에 데이터를 삽입하는 경우가 많은지 확인한다. 
• 맨 앞에 데이터를 삽입해야 하는 경우 deque를 활용하면 $ O(1) $의 성능을 보장할 수 있다.

 

 

유용한 기법

아래와 같이 선언된 리스트 my_list를 가지고 자주 사용되는 리스트 기법을 정리해보자.

my_list = [1, 2, 3]

데이터 추가

맨 끝에 데이터 삽입

my_list.append(4)
# 결과: my_list = [1, 2, 3, 4]

' + ' 연산자 사용

my_list = my_list + [4, 5]
# 결과: my_list = [1, 2, 3, 4, 5]

특정 위치에 데이터 삽입

my_list.insert(2, 17)
# 결과: my_list = [1, 2, 17]

 

데이터 삭제

특정 위치의 데이터 팝

pop은 리스트에서 데이터를 삭제할 뿐만 아니라 삭제한 데이터의 값을 반환한다.

popped_number = my_list.pop(2)
# my_list = [1, 2]
# popped_number = 3

특정 데이터 삭제

특정 위치를 기준으로 데이터를 삭제하는 것이 아닌, 특정 값이 처음 등장하는 위치의 데이터를 삭제한다. 

my_list.remove(3)
# 결과: my_list = [1, 2]

List Comprehension

아래의 방법으로 기존 리스트의 모든 데이터에 제곱연산을 한 새로운 리스트를 생성할 수 있다. 기존 리스트는 변경되지 않는다.

square_list = [num**2 for num in my_list]
# my_list = [1, 2, 3]
# square_list = [1, 4, 9]

 

기타

특정 값의 데이터가 처음 등장한 인덱스를 반환

해당 값의 데이터가 리스트에 존재하지 않으면 -1을 반환한다.

my_list.index(2) # 1
my_list.index(0) # -1

정렬

인수를 전달하지 않으면 오름차순으로 데이터를 정렬한다. 아래는 내림차순으로 데이터를 정렬하는 방법이다.

my_list.sort(reverse=True)
# my_list = [3, 2, 1]

특정 값의 데이터 개수 세기

my_list.count(1) # 1

 

 

---

📕 Reference

코딩 테스트 합격자 되기: 파이썬 편, 박경록, 골든래빗.

코딩 테스트 합격자 되기: 파이썬 편 | 박경록 - 교보문고