<완전탐색> 프로그래머스 카펫

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

프로그래머스

 

def solution(brown, yellow):
    stack = []
    answer = []
    for i in range(1,brown+yellow+1):
        if (brown+yellow) % i == 0:
            stack.append(i)
    if len(stack) % 2 == 0:
        answer.append(stack[len(stack)//2])
        answer.append(stack[len(stack)//2-1])
    else :
        answer.append(stack[len(stack)//2])
        answer.append(stack[len(stack)//2])

     
    return answer

내가 생각했던 풀이이다. 

우선 brown+yellow 값의 약수들을 구해주고, 그 절반 index를 출력해주면 된다고 생각했는데 76.9점밖에 받지 못하였다.

brown+yellow 값의 약수들을 다 구해주다보니 값이 클 때, 시간 초과가 일어난 것 같다. 

 

해결하지 못해 다른 사람의 풀이를 참고하였는데,

갈색-4) // 2 = 노란색의 col+row가 나오게 된다.

어떻게 이런 식을 생각해내는 것일까? 진짜 나는 부족하고 부족하구나 깨달았다.

이를 참고하여 다시 코드를 작성해보았다.

def solution(brown, yellow):
  ab = brown + yellow
  for a in range(ab, 2, -1): #ab부터 2까지 반복
    b = ab // a
    if ab % a == 0: #즉, ab가 a로 나눠진다면? 그럼 b가 나올 수 있으니깐
      if yellow == (a-2)*(b-2): ##yellow 개수 조건 충족하는지
        return [a,b]