[프로그래머스]문자열 계산하기

문자열 계산하기

문제 설명

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 합니다.

---

나의 풀이 코드

import java.util.*;

class Solution {
    public int solution(String my_string) {
        int answer = 0;
        
        // " "으로 스플릿해서 리스트에 담기        
        String[] ArraysStr = my_string.split(" ");
        List<String> list = Arrays.asList(ArraysStr);
        
        // list의 요소를 판별하며 만약 "+"나 "-"가 있으면 (i+1)번째 요소를 더하거나 빼기
        int sum = Integer.parseInt(list.get(0));
        for (int i=1; i<list.size(); i++){
            if(list.get(i).equals("+")){
                System.out.println("+발견");
                sum += Integer.parseInt(list.get(i+1));
            }
            else if (list.get(i).equals("-")){
                System.out.println("+발견");
                sum -= Integer.parseInt(list.get(i+1));
            }
        }
        answer = sum;
        return answer;
    }
}

배운점

import java.util.Arrays;

class Solution {
    public int solution(String myString) {
        return Arrays.stream(myString.replaceAll("- ", "-").replaceAll("[+] ", "").trim().split(" "))
                     .mapToInt(Integer::parseInt)
                     .sum();
    }
}

다른사람이 스트림으로 푼 코드이다.
replaceAll과 trim으로 공백을 지우고 split한 문자열 배열 스트림으로 변환하고,
정수로 변환하고 모두 더한다.

이때 replaceAll 과정에서 -를 다음 문자열에 붙여주어 결과적으로 음수로 만든다.