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

 

내가 직전에 한 그림 확대와 같이 count와 while을 사용하여 해결했다!

 

def solution(arr):
    for i in range(len(arr)):           
        for l in range(len(arr[i])):    
            if len(arr) > len(arr[i]):      #행의 수가 열의 수보다 많으면
                count = 0                   #각 행의 끝에 0을 추가하기 위해 새로운 변수 설정
                while len(arr) > len(arr[i]):   #행의 수가 열의 수보다 많으면 계속 반복
                    arr[i].append(0)            #0을 하나씩 추가
                    count += 1
            elif len(arr) < len(arr[i]):    #행의 수가 열의 수보다 적으면
                arr.append([0]*len(arr[i])) #열의 끝에 0을 추가한 이차원 배열
    return arr

 

처음에 내가 접근한 방법이 있었는데...어디서 막혀서 앞으로 나아가지 못했다...

 

그러나 다른 사람의 풀이를 보니 내가 나아가려고 했던 방법이 있었다.

 

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