N으로 표현 (python, javascript)

SeoYng·2020년 8월 24일
3
post-custom-banner

프로그래머스 문제 - N으로 표현 LV3

https://programmers.co.kr/learn/courses/30/lessons/42895

👀 깃헙 소스

문제설명
아래와 같이 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	number	return
5	12	4
2	11	3

솔루션
N은 1 이상 9 이하입니다. =>
pos에 각 i+1번 사용해서 만들수 있는 가능한 숫자들을 set으로 넣어둔다
초기화 : [{5}, {55}, {555}, {5555}, {55555}, {555555}, {5555555}, {55555555}]

i+1번 사용해서 만들수 있는 가능한 숫자
ex) i=3, i+1=4, 1번(0)+3번(2) & 2번(1)+2번(1) & 3번(2)+1번(0)
더해서 ==> i-1

코드

# 파이썬
def solution(N, number):

    pos = [set([int(str(N)*i)]) for i in range(1,9)]
    
    for i in range(8):
        for j in range(i):
            for op1 in pos[j]:
                for op2 in pos[i-j-1]:
                    pos[i].add(op1 * op2)
                    pos[i].add(op1 - op2)
                    pos[i].add(op1 + op2)
                    if op2 != 0:
                        pos[i].add(op1 // op2)
        if number in pos[i] :
            return i+1
    return -1

// 자바스크립트
function solution(N, number) {
    const set = new Array(8).fill().map(() => new Set());
    // [set() for x in range(8)]
    for(let i=0; i<8; i++){
        set[i].add(Number(N.toString().repeat(i+1)));
        console.log(set[-1])
        for (let j=0; j < i; j++) {
            for(const arg1 of set[j]){
                for(const arg2 of set[i-j-1]){
                    set[i].add(arg1+arg2);
                    set[i].add(arg1-arg2);
                    set[i].add(arg1*arg2);
                    set[i].add(arg1/arg2); // 자스는 0 나누는거 상관없음 -> infinity
                }
            }
        }
        if(set[i].has(number)) return i+1;
    }
    return -1;
}
profile
Junior Web FE Developer
post-custom-banner

1개의 댓글

comment-user-thumbnail
2021년 5월 15일

잘보고갑니다~

답글 달기