프로그래머스 Lv2 | 완전탐색 | 카펫

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

프로그래머스

20260401

훨씬 간단하게 풀었음

import math
def solution(brown, yellow):
    # 가로길이 >= 세로길이
    # yellow의 가로&세로 경우의 수 반복문 -> 올바른 brown 찾기
    
    for i in range(1,int(math.sqrt(yellow))+1):
        if yellow % i == 0:
            yellow_w, yellow_h = yellow//i, i

            cal_brown = (yellow_w+2)*2 + yellow_h*2
            if cal_brown == brown:
                break
    
    return [yellow_w+2, yellow_h+2]

 

 

---

if yellow % i 이 부분을 안 써서 처음에 틀렸었음

isqrt하면 정수 제곱근을 돌려준다는 사실도 알게됨. 그러면 math.floor할 필요가 없음

import math
def expect_brown(xy):
    xx, yy = xy[0], xy[1]
    
    return 2*(xx+2)+2*yy

def return_xy_list(yellow):
    xy_list = []
    for i in range(1, math.isqrt(yellow)+1):
        if yellow % i == 0:
            x = yellow/i; y = i
            xy_list.append([x,y])
    return xy_list
    
def solution(brown, yellow):
    xy_list = return_xy_list(yellow)
    for xy in xy_list:
        if expect_brown(xy) == brown:
            return [xy[0]+2, xy[1]+2]