문자열 계산하기
문제 #
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) {
String[] strArr = my_string.split(" ");
int answer = 0;
for(int i = 0; i<strArr.length; i++) {
// 연산자가 + 일경우 넘어간다.
if(strArr[i].equals("+")) continue;
// 연산자가 - 일경우 다음에 오는 숫자를 음수로 바꾼다.
// 마지막 인덱스의 값은 항상 숫자이므로 strArr[i+1]은 오류를 발생하지 않음
else if(strArr[i].equals("-")) strArr[i+1] = "-"+strArr[i+1];
// 연산자가 아닐경우 숫자이므로 answer에 더해준다.
else answer += Integer.parseInt(strArr[i]);
}
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();
}
}
/* 해설 */
// myString이 "10 + 30 - 20" 이라고 가정 할 때
Arrays.stream(
// "- "을 포함 한 경우 "-"으로 대체
myString.replaceAll("- ", "-") // "10 + 30 -20"
// "+"을 포함 한 경우 ""으로 대체
.replaceAll("[+] ", "") // "10 30 -20"
// 문자 앞 뒤 공백 제거
.trim()
// String배열로 변경
.split(" ") // ["10", "30", "-20"]
).mapToInt(Integer::parseInt) //[10, 30, -20]
.sum(); //(10)+(30)+(-20)
나의 코드 개선 #
class Solution {
public int solution(String my_string) {
int answer = 0;
String[] strArr = my_string.replaceAll("- ", "-").replaceAll("[+] ", "").split(" ");
for(String s : strArr) {
answer += Integer.parseInt(s);
}
return answer;
}
}
기존 나의 코드 → 개선 후 코드