[Programmers/Java/Lv.0/문자열 유형] 51.컨트롤 제트

문제 설명

숫자와 "Z"가 공백으로 구분되어 담긴 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 "Z"가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 "Z"로 이루어진 문자열 s가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.

 

제한사항

• 1 ≤ s의 길이 ≤ 200
• -1,000 < s의 원소 중 숫자 < 1,000
• s는 숫자, "Z", 공백으로 이루어져 있습니다.
• s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.
• 연속된 공백은 주어지지 않습니다.
• 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
• s는 "Z"로 시작하지 않습니다.
• s의 시작과 끝에는 공백이 없습니다.
• "Z"가 연속해서 나오는 경우는 없습니다.

 

입출력 예

s	result
"1 2 Z 3"	4
"10 20 30 40"	100
"10 Z 20 Z 1"	1
"10 Z 20 Z"	0
"-1 -2 -3 Z"	3

입출력 예 #1 : 본문과 동일합니다.
입출력 예 #2 : 10 + 20 + 30 + 40 = 100을 return 합니다.
입출력 예 #3 : "10 Z 20 Z 1"에서 10 다음 Z, 20 다음 Z로 10, 20이 지워지고 1만 더하여 1을 return 합니다.
입출력 예 #4, #5 : 설명 생략

 

나의 풀이

class Solution {
    public static int solution(String s) {
        s = s.replace(" ", "");
        char[] ch = s.toCharArray();

        int answer1 = 0;
        int answer2 = 0;

        for (int i = 0; i < ch.length; i++) {
            if (ch[i] == 'Z') {
                answer1 += (ch[i - 1] - '0');
            }
        }

        String filtered = s.replace("Z", ""); 
        char[] ch2 = filtered.toCharArray();

        for (char c : ch2) {
            answer2 += (c - '0');
        }

        return answer2 - answer1;
    }
}

// 테스트 케이스 1번만 통과함

문제점

• 숫자가 두 자리 이상일 때 char 단위로 처리하면 오류 발생
• 예를 들어, "10 20 30 40"을 char[]로 변환하면 { '1', '0', '2', '0', ...} 이렇게 쪼개지므로, "10"이 아닌 "1" + "0"이 되어서 잘못된 값이 나옴. → 공백을 기준으로 split(" ")하여 숫자 단위로 처리해야 함.

 

• 음수를 제대로 처리하지 않음
• '-' 문자도 char로 처리되면, '1' - '0' 같은 방식으로 처리하기 어려움. → Integer.parseInt()를 사용하여 정수 변환이 필요함.

 

다른 풀이1

import java.util.*;
class Solution {
    public int solution(String s) {
        // String[] split(String regex)
        String[] list = s.split(" "); // replace()를 사용하면 숫자가 두 자리 이상 or 음수일 때 문제가 발생함
        int answer = 0;
        for(int i = 0 ; i < list.length ;i++){
            if(list[i].equals("Z")){
                answer -= Integer.valueOf(list[i-1]);  
            } else { // "-1 -2 -3 Z"인 테스트 케이스에도 "Z"가 나타날 때 까지는 계속 else문으로 가니깐 괜찮
                answer += Integer.valueOf(list[i]);  
            }
        }
        return answer;
    }
}

 

다른 풀이2

import java.util.*;

class Solution {
    public int solution(String s) {
        int answer = 0;
        Stack<Integer> stack = new Stack<>();

        for (String w : s.split(" ")) {
            if (w.equals("Z")) {
                stack.pop();
            } else {
                stack.push(Integer.parseInt(w));
            }
        }
        for (int i : stack) {
            answer += i;
        }
        return answer;
    }
}