배열

• 1차원배열

#3가지 방법
arr = [0] * 6
arr = list(range(6))
arr= [0 for _ in range(6)]

• 2차원 배열 

arr = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
print(arr[2][3]) #12
arr = [[i]*4 for i in range(3)] # [[0,0,0,0],[1,1,1,1],[2,2,2,2]]

실제로는 오른쪽 처럼 저장된다.

 

배열 선택시 고려할 점

• 할당할 수 있는 메모리 크기를 확인
• 중간에 데이터 삽입이 많은지 확인

 

파이썬에서는 리스트로 배열 표현

 

리스트 기법

• 리스트 생성
• 리스트에 데이터 추가
  • append() 메서드: 리스트 끝에 새로운 데이터 추가
  • + 연산자: 두 리스트를 합쳐서 새로운 리스트 만든다.
  • insert() 메서드: 특정 위치에 데이터 삽입
• 리스트에서 데이터 삭제
  • pop() 메서드: 리스트의 마지막 요소  제거, 특정 위치 제거
  • remove() 메서드: 특정 데이터 삭제

my_list = [1, 2, 3, 4, 5]
my_list.append(6)  # [1, 2, 3, 4, 5, 6]
new_list = my_list + [7, 8]  # [1, 2, 3, 4, 5, 6, 7, 8]
my_list.insert(2, 10)  # [1, 2, 10, 3, 4, 5, 6]
my_list.pop()  # 마지막 요소 삭제 -> [1, 2, 10, 3, 4, 5]
my_list.pop(2)  # 인덱스 2의 요소 삭제 -> [1, 2, 3, 4, 5]
my_list.remove(4)  # 값 4를 삭제 -> [1, 2, 3, 5]

• 리스트 컴프리헨션 -> 기존 리스트를 기반으로 새 리스트를 만들거나 복잡한 리스트를 생성 가능

numbers = [1,2,3,4,5]
squares = [num ** 2 for num in numbers] #[1,4,9,16,25]

• 기타 메서드
  
  • sort() 메서드: 리스트 정렬
  • reverse() 메서드: 리스트 역순으로 뒤집기
  • len() 메서드: 리스트 길이 반환
  • index() 메서드: 특정 값의 첫 번째 위치(인덱스) 반환
  • count() 메서드: 리스트에서 특정 값이 몇 번 등장하는지 카운트
  • clear() 메서드: 리스트 모든 요소 삭제

my_list.sort()  # [1, 2, 3, 5]
my_list.reverse()  # [5, 3, 2, 1]
len(my_list)  # 4
my_list.index(1) #3 (첫 번째 1이 3에 위치)
m_list.count(1) #1 (1이 1번 등장)
my_list.clear()  # []

 

정렬 시 리스트를 arr.sort()는 기존 리스트가 바뀌지만, 그대로 남기고 싶다면 

sorted_list = list(sort(arr))

추가로 내림차순 및 오름차순을 하고싶다면

lst.sort(reverse=True) #내림차순
lst.sort(reverse=False) #오름차순

 

 

중복값 제거위한 함수

unique_list = list(set(lst))

set()함수는 해시 알고리즘으로 데이터를 저장하므로 시간 복잡도 O(N)

 

2차원행렬곱에 대한 기본 세팅

def solution(arr1, arr2):
    answer = [[]]
    r1,c1 = len(arr1),len(arr1[0])
    r2,c2 = len(arr2),len(arr2[0])
    ret = [[0]*c2 for _ in range(r1)]
    
    #arr1의 각 행과 arr2의 각 열에 대해서
    for i in range(r1):
        for j in range(c2):
            #두 행렬을 곱함
            for k in range(c1):
                ret[i][j]  += arr1[i][k] * arr[k][j]
    return answer

 

문제

'''
[1,-5,2,4,3] -> [5,1,2,3,4]
'''

def solution(s):
    s.sort()
    return s

# 시간 복잡도의 경우 NlogN이 된다.

print(solution([1,-5,2,4,3]))

• 파이썬에는 코딩테스트에 유용한 함수가 많다. 굳이 직접 만들지 말자!!

'''
s = [4,2,2,1,3,4]
return = [4,3,2,1]
'''

def solution(s):
    s.sort(reverse=True) #[4,4,3,2,2,1]
    #중복제거 -> set
    sets= set(s)

    return sets

print(solution([4,2,2,1,3,4]))
#이렇게 하면, 정렬이 다 날아간다. 주의할것

def solution(s):
    #중복제거 -> set
    sets= list(set(s)) #[1, 2, 3, 4]

    sets.sort(reverse=True)

    return sets

print(solution([4,2,2,1,3,4]))

 

https://school.programmers.co.kr/learn/courses/30/lessons/68644

'''
numbers = [2,1,3,4,1]
result = [2,3,4,5,6,7]

오름차순으로 정렬한다.
[1,1,2,3,4]
시간복잡도 널널하므로 이중 for문 돌리자, 포인터 두개잡고 반복
'''

def solution(numbers):
    numbers.sort()
    arr=[]
    for i in range(len(numbers)):
        for j in range(i+1,len(numbers)):
            arr.append(numbers[i]+numbers[j])


    return sorted(list(set(arr)))
print(solution([2,1,3,4,1]))
print(solution([5,0,2,7]))

이때, set은 자동으로 정렬해주지 않으므로 추가로 정렬을 명시적으로 해줘야 한다. 잊지말것!

 

https://school.programmers.co.kr/learn/courses/30/lessons/42840

'''
answrs = [1,2,3,4,5]
return = [1]

answers = [1,3,2,4,2]
return = [1,2,3]

answers = [1,2,3,4,5,1] -> 1은 6개, 2는 0개, 3은 0개
return = [1,2,3]
'''

def solution(answers):
    patterns = [
        [1,2,3,4,5],
        [2,1,2,3,2,4,2,5],
        [3,3,1,1,2,2,4,4,5,5]
    ]
    score=[0]*3
    for i in range(len(answers)):
        for j in range(len(patterns)):
            if patterns[j][i]==answers[i]:
                score[j]+=1

    maxs=max(score)
    lists=[]
    for i in range(len(patterns)):
        if maxs ==score[i]:
            lists.append(i+1)

    return lists

# print(solution([1,2,3,4,5]))
# print(solution([1,3,2,4,2]))
print(solution([1,2,3,4,5,1]))

여기서 문제는 answers의 크기가 patterns의 한개의 배열의 크기보다 클 경우 range 문제가 발생한다. -> %로 해결하자

따라서

if patterns[j][i]==answers[i]:
# 이 부분을 아래와 같이 변경하자
if patterns[j][i%len(patterns[j])]==answers[i]:

 

https://school.programmers.co.kr/learn/courses/30/lessons/12949