문제 링크

문제 해설

Idea

  • Dynamic Programming
  • S[1] = {N}
  • S[2] = {NN, N+N, N-N, N*N, N/N}
  • S[3] = {NNN, S[2] (+,-,*,/) S[1][y], …}
  • 2부터 8까지의 범위를 가진 i와 1부터 i-1까지의 범위를 가진 j에 대해,
    S[j]와 S[i-j]의 사칙연산 결과를 S[i]에 추가하고 해당 집합이 number를 포함하는지 검증

Time Complexity

  • DP: O(1)

Data Size

  • N: 1 <= int <= 9
  • number: 1 <= int <= 32,000
  • answer: int <= 8

해설 코드

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20

from itertools import product

def solution(N, number):
    S = [set() for _ in range(9)]

    if N == number:
        return 1
    else:
        S[1].add(N)

    for i in range(2,9):
        S[i].add(int(str(N)*i))
        for j in range(1,i):
            for x,y in product(S[j],S[i-j]):
                S[i].update({x+y,x-y,x*y})
                if y != 0:
                    S[i].add(x//y)
        if number in S[i]:
            return i
    return -1