728x90
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
728x90
반응형
'Algorithm > 프로그래머스' 카테고리의 다른 글
[프로그래머스] 가위 바위 보 - 파이썬 (0) | 2024.04.01 |
---|---|
[프로그래머스] n의 배수 고르기 - 파이썬 (0) | 2024.04.01 |
[프로그래머스] 이차원 배열 대각선 순회하기 - 파이썬 (0) | 2024.03.31 |
[프로그래머스] 특별한 이차원 배열 1 - 파이썬 (1) | 2024.03.29 |
[프로그래머스] I로 만들기 - 파이썬 (1) | 2024.03.29 |