[프로그래머스] 정사각형으로 만들기 - 파이썬

Programmers

정사각형으로 만들기

문제

코드

<틀린풀이>

def solution(arr):
    for i in range(len(arr)):
        for j in range(len(arr[i])):
            if len(arr) > len(arr[i]):
                arr[i].append(0)
            elif len(arr) < len(arr[i]):
                arr.append([0] * len(arr[i]))
            else:
                break
    return arr

 

<정답풀이>

def solution(arr):
    answer = []

    row = len(arr)
    col = len(arr[0])

    if row > col:
        for i in arr:
            answer.append(i + [0] * (row - col))
    elif row < col:
        for _ in range(col - row):
            arr.append([0] * col)
        answer = arr
    else:
        answer = arr

    return answer

Reference

• https://school.programmers.co.kr/learn/courses/30/lessons/181830