JS Practice) 프로그래머스 : 문자열 계산하기

 

 

문자열 계산하기

문제설명 :

my_string은 "3 + 5"처럼 문자열로 된 수식입니다. 문자열 my_string이 매개변수로 주어질 때, 수식을 계산한 값을 return 하는 solution 함수를 완성해주세요.

 

제한 사항

• 연산자는 +, -만 존재합니다.
• 문자열의 시작과 끝에는 공백이 없습니다.
• 0으로 시작하는 숫자는 주어지지 않습니다.
• 잘못된 수식은 주어지지 않습니다.
• 5 ≤ my_string의 길이 ≤ 100
• my_string을 계산한 결과값은 1 이상 100,000 이하입니다.
• my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다.
• 계산에 사용하는 숫자는 1 이상 20,000 이하인 자연수입니다.
• my_string에는 연산자가 적어도 하나 포함되어 있습니다.
• return type 은 정수형입니다.
• my_string의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

 

입출력 예

my_string	result
"3 + 4"	7
"3 + 4 - 2"	5

 

접근

:  숫자와 연산자는 공백으로 구분되어 있으므로 공백을 기준으로 split()하였다. my_string 배열의 요소를 돌며 해당 요소가 숫자일때, 연산자가 있으면 answer에 해당 요소를 더하거나 빼주었고, 연산자가 없으면 해당 요소가 처음 들어온 요소라는 의미이니 answer에 해당 숫자를 더해주었다. 이때 요소를 숫자로 바꿔주지 않으면 문자열이 그냥 더해지니 이 부분을 주의해야 한다. 마지막으로 해당 요소가 연산자인 경우, operators에 추가해주었다. 그리고 해당 연산자가 사용된 경우, 연산자를 초기화 해주었다.

function solution(my_string) {
    my_string = my_string.split(" ")
    let operators = ''
    let answer = 0
    for(const el of my_string){
        if(Number(el)){
            if(operators.length > 0) {
                if(operators[0] === '+') answer += Number(el)
                else answer -= Number(el)
            } else answer += Number(el)
            operators = ''
        } else operators += el
    }
    return answer
}

 

 

solution

function solution(my_string) {
    const arr = my_string.split(' ')
    while(arr.length > 1) arr.unshift(+arr.shift() + (arr.shift() === "+" ? 1 : -1) * arr.shift())
    return arr[0]
}

: my_string을 공백으로 split()한 후 차례대로 3개의 요소를 가져옴과 동시에 배열에서 제거한다. 첫번째 요소는 '+'를 붙여 숫자로 형변환을 시켜주고, 두번째 요소가 '+'인 경우, 세번째 요소에 을 곱해주고 두번째 요소가 '-'인 경우, 세번째 요소에 '-1'을 곱해준다. 그럼 두 숫자의 연산이 완료되며 이를 배열의 맨 앞에 넣어준다.

 

예를 들어, my_string이 "3 + 4 - 2"인 경우, 이를 배열([ '3', '+', '4', '-', '2' ])로 만든 후 3개의 요소 '3', '+', '4'을 가져온다. 먼저 첫번째 요소인 '3'에 더하기를 붙여 숫자로 만든다. 그리고 두번째 요소인 연산자가 덧셈인지 확인하고 덧셈이면 세번째 요소인 4에 +1을 곱해주어 숫자형으로 만든다. 이후 숫자 3과 4를 더한 값을 기존의 배열에 넣어준다. 그럼 배열의 상태는 [7, '-', '2' ]로 업데이트 된다.