코딩 퀴즈 - 이차원 배열 대각선 순회

이번 코딩 퀴즈는 2차원 배열의 대각선 요소를 조회하고, 이 요소들로 구성된 리스트를 반환하는 함수를 작성합니다.

입력으로 주어지는 2차원 배열은 정사각형 배열이며, 배열의 왼쪽 상단부터 오른쪽 하단까지의 대각선에 위치한 요소들로 리스트를 생성해야 합니다.

Copy
def solution(matrix):
    # 여기에 코드를 작성하세요
    return

---

• 입력으로 주어지는 2차원 배열은 항상 정사각형 형태입니다 (NxN).

• 배열의 각 요소는 정수입니다.

• 배열의 크기는 최소 1x1, 최대 100x100입니다.

---

• 입력: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

• 출력: [1, 5, 9]

• 입력: [[1, 2], [3, 4]]

• 출력: [1, 4]