[프로그래머스] 카펫 (Level 2) (파이썬 python)

 

 

문제 링크

 

 

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

코딩테스트 연습 - 카펫

 

 

문제

 

 

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고

테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

 

 

Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만,

전체 카펫의 크기는 기억하지 못했습니다.

 

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때

카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

 

 

제한사항

 

 

- 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.

- 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.

- 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

 

 

입출력 예

brown / yellow / return

10	2	[4, 3]
8	1	[3, 3]
24	24	[8, 6]

 

 

 

문제 풀이

 

 

def solution(brown, yellow):
    answer = []

    x = []
    for i in range(brown):
        if i * i - (brown * 1/2 + 2) * i + brown + yellow == 0:
            x.append(i)

    for i in range(len(x)):
        if x[i] >= (- x[i] + brown * 1/2 + 2):
            answer.append(x[i])
            answer.append(int(- x[i] + brown * 1/2 + 2))

    return answer

 

 

오랜만에 수학을 접하니 머리가 아프다 ㅋㅋㅋ

이렇게 푸는게 맞나 의구심이 들었는데, 다른 사람의 풀이를 보니 다들 수학적 개념을 활용해 풀이를 한 것 같았다

나는 조금 더 복잡스럽게 문제를 푼 것 같아 아쉽지만..

 

일단 첫 번째 for문에서는 가로의 길이를 찾으려하였는데

가로의 길이가 갈색 격자의 수보다 클 수 없기 때문에 범위를 갈색 격자 수까지로 하였다.

(다른 사람의 풀이를 보니 brown의 제곱근 + 1까지로 두었다.

그렇게 두어도 범위에 문제가 없기 때문에 탐색 수를 줄이기 위해 그렇게 하는게 맞을 것 같다.)

 

일단 나는 갈색 격자의 가로, 세로를 A, B로 두고

노란색 격자가 가로, 세로 모두 갈색 격자의 가로, 세로보다 2만큼 작기 때문에

노란색 격자의 가로, 세로를 A-2, B-2로 두고 식을 세웠다.

 

A * B - (A - 2) * (B - 2) 를 한 결과가 갈색 격자의 수, 즉 입력받은 변수 brown이며

(A - 2) * (B - 2) 를 한 결과가 노란색 격자의 수, 즉 입력받은 변수 yellow이다.

 

두 식을 정리하여 하나의 변수 A로 정리하니 아래 식이 나왔다

 

A * A - (brown * 1/2 + 2) * A + brown + yellow = 0

 

이 식을 통해 검사한 i를 배열 x에 집어넣고

 

가로가 세로보다 길거나 같아야한다는 조건에 맞추어 한 번 더 x를 검사해주었다.

검사에 사용된 식 역시 위의 관계에 대한 식을 사용했다.