[Python] 프로그래머스 lv.0 정사각형으로 만들기

 

정답 코드

def solution(arr):
    answer = [[]]
    r = len(arr)
    c = len(arr[0])
    
    if r == c : 
        return arr
    elif r > c : 
        for i in range(r) : 
            arr[i] += [0] * (r-c)
    else : 
        for j in range(c-r) : 
            arr += [[0] * c]
                
    return arr

 

다른 사람 코드 1

def solution(arr):
    n=len(arr)
    m=len(arr[0])
    if n>m:
        for i in range(n):
            for j in range(n-m):
                arr[i].append(0)
    else:
        for i in range(m-n):
            arr.append([0]*m)

    return arr

 

다른 사람 코드 2

def solution(arr):
    answer = []
    x = len(arr)
    y = len(arr[0])
    m = max(x, y)

    answer=[[0]*m for i in range(m)]

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

    return answer