Home > Programmers > N으로 표현

N으로 표현
C++ C# Go Java JavaScript Kotlin Python2 Python3 Ruby Scala Swift

출처 : N으로 표현

Solution
Python

# 사칙연산 후 값 저장해주는 함수
def pmmd(visited, cnt, i1, i2):
    for n1 in visited[i1]:
        for n2 in visited[i2]:
            visited[cnt].add(n1+n2)
            visited[cnt].add(n1-n2)
            visited[cnt].add(n1*n2)
            if n2!=0:
                visited[cnt].add(n1//n2)

def solution(N, number):
    # 중복 연산 방지를 위해 set으로 설정
    visited = [set([0])] + [set([int(str(N)*i)]) for i in range(1,9)]
    # N과 number가 같다면, 1개만 사용하면 된다.
    if number==N: return 1
    # 사용하게 될 수는 1~8까지 / 1개 사용하는 경우는 앞서 확인했으니 2부터 체크
    for count in range(2, 9):
        for i in range(1, count):
            pmmd(visited, count, count-i, i)
            if number in visited[count]:
                return count
    return -1

if __name__ == '__main__':
    # testcase 1 : 4
    print(solution(5, 12))
    
    # testcase 2 : 3
    print(solution(2, 11))

문제 설명

아래와 같이 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 합니다.
입출력 예
N number return
5 12 4
2 11 3
입출력 예 설명

예제 #1
문제에 나온 예와 같습니다.

예제 #2
11 = 22 / 2와 같이 2를 3번만 사용하여 표현할 수 있습니다.

※ 공지 - 2020년 9월 3일 테스트케이스가 추가되었습니다.