[프로그래머스(Java)] 문자열 계산하기 / split

 

Level. 0

 

문제

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의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

 

풀이

class Solution {
    public int solution(String my_string) {
        int answer = 0;
        String[] str = my_string.split(" ");
        answer = Integer.parseInt(str[0]);
        for(int i=0; i<str.length-1; i++){
            if(str[i].equals("+")) answer += Integer.parseInt(str[i+1]);
            else if (str[i].equals("-")) answer -= Integer.parseInt(str[i+1]);
        }
        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();
    }
}

1. '+ '는 제거하고, '- '는 '-'로 변경하여 음수처럼 만든 후 split을 사용해 문자열을 배열로 변환하였다.

2. mapToInt를 사용하여 숫자 스트림 변환해 준 후 sum()으로 모든 값을 더해주었다. 

 

trim : 오른쪽 왼쪽 공백을 제거한다. 

 

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/120902

프로그래머스