Algorithm/📊 Problem Solving

[프로그래머스/Programmers] N으로 표현

posted by sangmin

N으로 표현

📌 문제

아래와 같이 5와 사칙연산만으로 12를 표현할 수 있습니다.

12 = 5 + 5 + (5 / 5) + (5 / 5)
12 = 55 / 5 + 5 / 5
12 = (55 + 5) / 5

5를 사용한 횟수는 각각 6,5,4 입니다. 그리고 이중 가장 작은 경우는 4입니다.
이처럼 숫자 N과 number가 주어질 때, N과 사칙연산만 사용해서 표현 할 수 있는 방법 중 N 사용횟수의 최솟값을 return 하도록 solution 함수를 작성하세요.

⚠️ 제한사항

  • N은 1 이상 9 이하입니다.
  • number는 1 이상 32,000 이하입니다.
  • 수식에는 괄호와 사칙연산만 가능하며 나누기 연산에서 나머지는 무시합니다.
  • 최솟값이 8보다 크면 -1을 return 합니다.

📋 코드

from collections import defaultdict

def solution(N, number):
    dic = defaultdict(set)

    for i in range(1, 9):
        dic[i].add(int(str(N) * i))

        for j in range(1, i):
            for x in dic[j]:
                for y in dic[i-j]:
                    for value in (x * y, x + y, x - y):
                        dic[i].add(value)

                    if y != 0:
                        dic[i].add(x // y)

        if number in dic[i]:
            return i
    return -1

💡 한마디

N에 따라 만들 수 있는 수를 set 에 저장했다. 만약 N = 5 인 경우,

  • N이 1개 : 5
  • N이 2개 : 55, 5 * 5, 5 + 5, 5 - 5, 5 / 5
  • N이 3개 : 55 * 5, 55 + 5, 55 - 5, 55 / 5, 25 * 5, 25 + 5, 25 - 5, …

이런 식으로 풀었다. N이 4개인 경우 1개로 만든 수들과 3개로 만든 수들을, 2개로 만든 수들끼리 각각 사칙연산 해주면 된다.

 

코딩테스트 연습 - N으로 표현

 

programmers.co.kr