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

 

❤️ Problem

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

 

• 입출력 예 & 설명

no	my_string	result
1	"3 + 4"	"7"

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

 

---

 

💛 Solution

풀이

class Solution {
    public int solution(String my_string) {
        String[] arr = my_string.split(" ");
        int calc = Integer.parseInt(arr[0]);

        for(int i=1; i<arr.length; i++) {
            if(arr[i].equals("+")) {
                calc += Integer.parseInt(arr[i+1]);
            } else if (arr[i].equals("-")) {
                calc -= Integer.parseInt(arr[i+1]);
            }
        }
        return calc;
    }
}

 

처리속도 GOOD

 

---

 

💜 Comment

이 문제의 중요한 힌트는 "숫자와 연산자는 공백 하나로 구분되어 있습니다"와 "잘못된 수식은 주어지지 않습니다"이다. 즉, 연산자(+,-) 앞뒤로 공백(띄어쓰기)이 있고 공백을 기준으로 문자열 요소의 맨 앞과 맨 뒤는 필연적으로 숫자로 치환이 가능하다고 보면 된다.

따라서 공백을 기준으로 문자열을 나누어 String[] arr에 넣어둔 뒤, 연산이 된 값을 저장할 변수 calc를 선언한 뒤 arr[0]를 넣어준다. 이후에는 조건문으로 arr[i]가 "+"와 "-" 값을 가지는 경우에만 그 뒤에 나오는 숫자(arr[i+1])를 연산자에 따라 처리해주면 끝.

제한사항 중 "my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다."라는 부분 때문에 처음부터 int[]나 long[]을 사용하지 않고 그 때 그 때 Integer.parseInt()를 사용해서 풀었다. 이렇게 하면 long[] 선언 없이도 풀이가 가능하다.

 

생각보다 금방 풀었는데 8점 GET😍
근데 캡쳐를 깜빡했다,, 헷