알고리즘/프로그래머스

[파이썬][프로그래머스] 완전탐색 카펫

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

반응형