이차원 배열 대각선 순회 - 문제 풀이
이차원 배열의 대각선 요소들을 리스트로 반환하는 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]
다음 내용이 궁금하다면?
코드프렌즈 PLUS 멤버십 가입 or 강의를 등록해 주세요!