이차원 배열 대각선 순회 - 문제 풀이

이차원 배열의 대각선 요소들을 리스트로 반환하는 3가지 방법을 확인해 보세요.

복사
def solution(matrix):
    # 단계 1: 대각선 요소를 저장할 빈 리스트 생성
    diagonal_elements = []

    # 단계 2: 2차원 배열의 각 행에 대해 반복
    for i in range(len(matrix)):
        # 단계 3과 4: 대각선 요소를 찾아 리스트에 추가
        diagonal_elements.append(matrix[i][i])

    # 단계 5: 완성된 리스트 반환
    return diagonal_elements

이 함수를 사용하면 입력된 이차원 배열을 대각선으로 순회하여 요소를 반환할 수 있습니다.

복사
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

print(solution(matrix))  # 출력: [1, 5, 9]