카펫→ 약수는 제곱근까지만 구하기, 2차원 리스트 요소2개 x,y로 접근★★

나의 풀이

from math import *
def solution(brown, yellow):
    divisor = []

    for i in range(1, int(sqrt(yellow))+1):
        if yellow % i ==0:
            divisor.append([yellow//i, i])

    for y_width, y_height in divisor:
        if brown ==(y_width+2)*(y_height+2)-yellow:
              a, b=y_width+2, y_height+2
    return a,b

▶ 일단 yellow의 약수의 순서쌍을 구한다. 다만, sqrt(n)까지만 구한다.

 

▶ 2차원 리스트의 각 행의 요소가 2개인 경우 y_width, y_height 2개의 임시변수로 접근할 수 있다. 

 

▶ if brown ==(y_width+2)*(y_height+2)-yellow: 

 

▶ (y_width+2)*(y_height+2) : 전체 격자의 개수   yellow : 노란색 격자의 개수 

 

▶ 전체 격자의 개수 - 노란색 격자의 개수 = 갈색격자의 수

 

 

 

다른 사람의 풀이 

def solution(brown, yellow):
    for i in range(1, int(yellow**(1/2))+1):
        if yellow % i == 0:
            if 2*(yellow//i +i) == brown-4:
                return [yellow//i+2, i+2]

▶ (브라운 가로 +브라운 세로)*2 - 4(브라운겹치는 부분제거) = 브라운 갯수

▶ (노란색 가로 + 노란색 세로)*2 + 4(부족한 부분 추가)= 브라운 갯수 

 

▶ yellow // i는 노란색의 가로, i는 노란색의 세로를 의미한다.