이차원 배열 대각선 순회 - 문제 풀이
이차원 배열의 대각선 요소들을 리스트로 반환하는 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 튜터
배포
디자인
업로드
수업 노트
즐겨찾기
도움말