알고리즘/프로그래머스
[파이썬][프로그래머스] 완전탐색 카펫
SBOX Learning by doing
2022. 11. 28. 21:22
반응형
def solution(brown, yellow):
answer = []
# 가로 * 세로 = brown + yellow
areas = findRowCol(brown + yellow)
# (가로 + 세로) * 2 - 4 = brown
for area in areas:
if ((area[0] + area[1]) * 2 - 4) == brown:
return area
return answer
def findRowCol(data):
answer = []
for i in range(data//2 + 1):
if i in (0, 1, 2):
pass
else:
if data % i == 0:
if i < data//i:
if [data//i, i] not in answer:
answer.append([data//i, i])
else:
if [i, data//i] not in answer:
answer.append([i, data//i])
return answer
반응형