[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)