가이드라인

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

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


방법 1
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]

가이드라인

AI 튜터

배포

디자인

업로드

수업 노트

즐겨찾기

도움말