본문 바로가기
코딩 테스트

[Python] 코딩테스트 고득점Kit | 완전탐색3-카펫

by 카프리썬_ 2021. 5. 26.
728x90

아래의 문제는 프로그래머스 코딩테스트 고득점 Kit 내용이며 코드는 직접 푼 내용입니다.


카펫

문제상황

Leo는 카펫을 사러 갔다가

아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

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

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

 

요구사항

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

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

 

제한사항

  • 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
  • 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
  • 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

입출력


풀이과정

구하고자 하는 카페의 가로와 세로를 a와 b라고 한다.

그리고 전체 카펫에 쓰여진 타일(?)의 총 갯수(s)는 brown과 yellow를 더한값이겠지?

 

이제 직접 가로와 세로의 길이를 찾아 들어가는 거야. 

가로(a) 는 타일의 전체 개수부터 최소2개일테니 for문을 돌아.

즉, brown+yellow = a *b = 카펫 타일총개수(s) 이니까 a는 s로 나누어져. 그리고 이때 b는 s//a값이 되.

그리고 yellow 타일의 개수는 위에 아래에서 하나씩 뺀 길이니까 (a-2)*(b-2) 값일꺼야. 

이걸 만족하는 a,b를 리턴한다.   

 

ㅂㄷㅂㄷ그냥 사실 결국 수학수식이다..

 

 

 

결과

내가 푼 정답코드결과

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

 

배운점

  • 수학적 사고력..? 수학 너무 어려운데?

참고 

참고한 블로그 링크

반응형