[JS] 문자열 계산하기

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

 

입출력 예 #1

• 3 + 4 = 7을 return 합니다.

---

왜.. 프로그래머스 1단계보다 더 어려운건지.. 입문 문제를 푸는데 참고자료도 없어서 풀 때 꽤나 애를 먹고 있다.

 

[풀이]

function solution(my_string) {
  const arr = my_string.split(" ").filter((el) => el !== "+");
  let result = 0;

  for (let i = 0; i < arr.length; i++) {
    if (arr[i] === "-") {
      result -= parseInt(arr[i + 1] * 2);
    } else {
      result += parseInt(arr[i]);
    }
  }
  return result;
}

"3 + 4" 의 결과값은 7이다.

 

이때 한가지의 경우만 생각하면 아주 오산이다. 여러가지 케이스가 존재한다.

연산자는 + 뿐만 아니라 - 도 존재하기 때문이다.

그래서 문제를 풀 때 케이스를 간단하게 추가하고 시작해도 좋다. 나는 "5 + 3 - 2"이라는 케이스를 추가하고 풀었다.

 

풀이를 하나하나 뜯어보면,

const arr = my_string.split(" ").filter((el) => el !== "+");

배열을 하나 만들어야한다. 배열을 만들 때는 split(" ") 즉, 공백을 기준으로 잘라 넣어야 한다.

 

예시)

my_string = "5 + 3 - 2" 일 때 공백을 기준으로 자르면 arr = ["5", "+", "3", "-", "2"]이다.

 

여기서 filter함수를 통해 "+"를 가지고 있다면 제외하고 반환시켜줘야한다.

결국, arr = ["5", "3", "-", "2"]가 된다.

 

let result = 0;

  for (let i = 0; i < arr.length; i++) {
    if (arr[i] === "-") {
      result -= parseInt(arr[i + 1] * 2);
    } else {
      result += parseInt(arr[i]);
    }
  }

그리고 연산한 값을 담을 result 변수를 만든다.

 

for문 안에서 arr[i]가 "-" 일 때 result에 arr[i+1]의 2배를 빼줘야 한다.

이유는 else에 있는데

 

예시) 

arr = ["5", "3", "-", "2"] 라면

i = 2 일때 "-"와 같으며 다음 숫자인 2를 빼줘야 한다.

근데 i = 3 일때 다시 arr[3]을 더해줘야한다.

그렇게 되면 결국

arr[0] + arr[1] - arr[3] + arr[3] = 5 + 3 - 2 + 2 = 8 이 되어 원하는 값인 6이 나오지 않는다.

 

결과적으로 arr[3]은 빼줘야 하는 값이므로 한 번 더 빼줘야 한다. (그래서 2배를 빼주는 것)

---

[코드 리뷰]

function solution(my_string) {
  const arr = my_string.split(" ");
  let result = Number(arr[0]);

  arr.forEach((el, i) => {
    const next = Number(arr[i + 1]);

    if (el === "+") result += next;
    else if (el === "-") result -= next;
  });

  return result;
}

처음에 arr 만들 때 filter를 안하고 단순하게 생각해도 된다.

 

하지만 처음에 미리 첫 번째 요소를 넣은 상태에서 순회해야 한다!

그래야 순회를 하면서 첫 번째 요소를 놓치지 않고 누적할 수 있다.

 

예시)

arr = ["5", "+", "3", "-", "2"]

result = Number(arr[0]) = 5

 

arr를 forEach를 통해 순회해준다.

arr[1] = "+" 이므로 arr[2]를 result에 누적시켜 더해준다. (5+3)

arr[3] = "-" 이므로 arr[4]를 result에 누적시켜 빼준다. (5+3-2)