[프로그래머스/파이썬] Lv.0 정사각형으로 만들기 (16')

Programmers 레벨 0_코딩 기초 트레이닝 문제 풀이

https://school.programmers.co.kr/learn/challenges/training?order=recent

코딩 기초 트레이닝 | 프로그래머스 스쿨

 

 

풀이 날짜 : 2023-09-21

소요 시간 : 16min

 

---

 

문제 설명

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

 

 

 

나의 풀이

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

 

 

 

다른 풀이

def solution(arr):
    if len(arr) > len(arr[0]):
        return [i + [0] * (len(arr) - len(arr[0])) for i in arr]
    elif len(arr) < len(arr[0]):
        return arr + [[0] * len(arr[0]) for i in range(len(arr[0])-len(arr))]
    else:
        return arr