Py) 프로그래머스 42842 카펫

https://programmers.co.kr/learn/courses/30/lessons/42842

코딩테스트 연습 - 카펫

 

def solution(brown, yellow):
    answer = []
    candi = []
    
    width = brown//2-1
    total = brown+yellow
    for i in range(1, width+1):  # 가로
        for j in range(1, i+1):  # 세로
            if i*j == total and i >= j:
                candi.append([i,j])
                
    if len(candi) ==1:
        answer.append(candi[0])
    else:
        for i in candi:
            if i[0]*2 + ((i[1]-2)*2) == brown:
                answer.append(i)
                break
    
    return answer[0]

뿌듯- 기분 좋게 오후 할 일 하겠당