프로그래머스_카펫

1. 문제

 

출처 : 프로그래머스

2. 소스코드

import math   def solution(brown, yellow):     answer = []     all = brown + yellow     for col in range(3, int(math.sqrt(all))+1):         if all % col: continue         row = all // col         if (row-2) * (col-2) == yellow:             return [row, col]

 

3. 고찰

1) 알고리즘

전체 카펫 격자 개수를 all에 담았다.

세로 < 가로 이기 때문에 세로에 대해 for문을 돌렸다. 가로에 대해서 돌리려면 루트(전체 격자개수) ~ 전체격자개수로 돌리면 된다.

가로, 세로의 최소 개수는 3이다. yellow의 상하좌우에 brown이 1줄씩 있기 때문.

세로col을 3~루트(전체 격자 개수)까지 loop를 돌리면서 확인하였다.

    나눗셈의 나머지가 0이라면 (가로, 세로는 전체의 약수)

        전체÷세로 = 가로 로 가로 격자 개수를 구하고

        (가로-2)*(세로-2) = yellow인지 확인하였다.

        맞다면 answer에 가로, 세로를 담아 리턴하였다.