알고리즘/프로그래머스
N으로 표현
Ferrero Rocher
2020. 7. 27. 14:38
문제 설명
아래와 같이 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 합니다.
입출력 예
Nnumberreturn
5 | 12 | 4 |
2 | 11 | 3 |
입출력 예 설명
예제 #1
문제에 나온 예와 같습니다.
예제 #2
11 = 22 / 2와 같이 2를 3번만 사용하여 표현할 수 있습니다.
코드
def solution(N, number):
s = [ set() for x in range(8) ]
for idx, val in enumerate(s, start = 1): # idx = 1부터, val = set()
val.add(int(str(N) * idx))
print(s[1])
for i in range(1, len(s)):
for j in range(i):
for op1 in s[j]:
for op2 in s[i-j-1]:
# print(i,op1,op2)
s[i].add(op1 + op2)
s[i].add(op1 - op2)
s[i].add(op1 * op2)
if op2 != 0:
s[i].add(op1 // op2)
if number in s[i]:
return i+1
return -1
print(solution(5,12))
- N 값이 최대 8개까지 사용가능하므로 N이 한개부터 8개까지 있는 set으로 이루어진 배열을 만듬.
- 각 set()을 사용한 이유는 중복 추가를 막기 위해서
- start가 1인 이유는 N이 최소 한개 이상 써져야하기 때문
- N이 두개 일 때 조합은 N이 한개 일 때에 N을 하나 더 붙여서 사칙연산한 것과 같음.
- 즉 i번째는 i-1번째 값에 N을 사칙연산 해서 붙인 값과 같음.