[Python Algorithm] 카펫 Programmers(Lv.2)

문제

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

코딩테스트 연습 - 카펫

 

내 문제 풀이

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

    for i in range(3, brown):
        if (i - 2) * ((brown - (i * 2)) / 2) == yellow:
            answer = [(brown - (i * 2)) / 2 + 2, i]
            break

    return answer

👉 완전탐색 문제로, 카펫이 깔리는 규칙을 생각해서 풀어주면 된다.

가장 자리의 딱 한칸씩만 갈색이 깔리고, 나머지는 노란색이므로 전체 카펫의 최소 한 변 길이는 3이다.

따라서 3부터 for문을 돌려주었고, 세로 길이를 3부터 늘려주면서, 반복마다 노란색이 들어갈 수 있는 칸 수를 구했다.

 

💡 피드백

• 갈색과 노란색의 상관 관계를 잘 생각해주면 쉽게 풀 수 있는 문제였다.