[Python/프로그래머스]코딩테스트 정사각형으로 만들기

 

 

Python 코딩테스트 정사각형으로 만들기

코딩테스트 정사각형으로 만들기

 

 

 

문제 설명

이차원 정수 배열 arr이 매개변수로 주어집니다. arr의 행의 수가 더 많다면 열의 수가 행의 수와 같아지도록 각 행의 끝에 0을 추가하고, 열의 수가 더 많다면 행의 수가 열의 수와 같아지도록 각 열의 끝에 0을 추가한 이차원 배열을 return 하는 solution 함수를 작성해 주세요.

 

 

 

제한사항

• 1 ≤ arr의 길이 ≤ 100
• 1 ≤ arr의 원소의 길이 ≤ 100
  • arr의 모든 원소의 길이는 같습니다.
• 1 ≤ arr의 원소의 원소 ≤ 1,000

 

 

Python 코딩테스트 정사각형으로 만들기 코드

def solution(arr):
    max_size = max(len(arr), len(arr[0]))
    answer = [[0]*max_size for _ in range(max_size)]

    for i in range(len(arr)):
        for j in range(len(arr[i])):
            answer[i][j] = arr[i][j]

    return answer