[programmers] 숫자와 숫자 영단어가 섞인 문자열을 정수로 변환하기

[level 1] 숫자 문자열과 영단어 - 81301

문제 링크

구분

코딩테스트 연습 > 2021 카카오 채용연계형 인턴십

문제 설명

네오와 프로도가 숫자놀이를 하고 있습니다. 네오가 프로도에게 숫자를 건넬 때 일부 자릿수를 영단어로 바꾼 카드를 건네주면 프로도는 원래 숫자를 찾는 게임입니다.

다음은 숫자의 일부 자릿수를 영단어로 바꾸는 예시입니다.

• 1478 → "one4seveneight"
• 234567 → "23four5six7"
• 10203 → "1zerotwozero3"

이렇게 숫자의 일부 자릿수가 영단어로 바뀌어졌거나, 혹은 바뀌지 않고 그대로인 문자열 s가 매개변수로 주어집니다. s가 의미하는 원래 숫자를 return 하도록 solution 함수를 완성해주세요.

참고로 각 숫자에 대응되는 영단어는 다음 표와 같습니다.

숫자	영단어
0	zero
1	one
2	two
3	three
4	four
5	five
6	six
7	seven
8	eight
9	nine

---

제한사항

• 1 ≤ s의 길이 ≤ 50
• s가 "zero" 또는 "0"으로 시작하는 경우는 주어지지 않습니다.
• return 값이 1 이상 2,000,000,000 이하의 정수가 되는 올바른 입력만 s로 주어집니다.

 

---

1차 시도

class Solution {
    public int solution(String s) {
        String answer = "";
        
        StringBuilder sb = new StringBuilder(answer);
        
        Map<String, Integer> numberMap = Map.of(
            "zero", 0, "one", 1, "two", 2, "three", 3, "four", 4, "five", 5, "six", 6, "seven", 7, "eight", 8, "nine", 9);
        
        for(int i = 0; i < s.length(); i++){
            // 숫자인지 문자인지 확인
            if(numberMap.containsValue(s.charAt(i) - '0')){ 
                // 숫자이면 바로 append
                sb.append(s.charAt(i) - '0');
            }else{
                // 문자이면 한 단어를 추출 -> map에서 value 찾아서 append
                // keyset과 문자열 비교
            }
        }
        
        answer = Integer.parseInt(answer);
        
        return answer;
    }
}

 

숫자 문자열을 key, 변환할 값을 value로 Map을 만들어서 반복문 내에서 숫자가 오면 바로 결과값에 append하고,

문자가 오면 한 단어를 추출해서 Map을 통해 append를 진행하려고 했다.

이렇게 하려면 else문 안에 반복문을 하나 더 만들고, 내부에서 keyset과 비교한 다음에 값을 집어 넣어야 한다.

그리고 two-three, four-five 등 첫 글자가 일치하는 숫자 문자열이 있기 때문에 상황에 따라 문자열의 2번째 자리까지도 비교하는 로직이 필요했다.

 

가독성 있으면서 깔끔한 코드를 만들고 싶어서 Map을 사용해본 건데 반복문 내부가 너무 지저분해지는 것 같아 이 방법은 폐지하기로 했다.

 

---

개선된 코드

class Solution {
    public int solution(String s) {
        
        String[] strArr = {"zero", "one", "two", "three","four", "five", "six", "seven", "eight", "nine"};
        
        for(int i = 0; i < strArr.length; i++){
            s = s.replaceAll(strArr[i], Integer.toString(i));
        }
        
        return Integer.valueOf(s);
    }
}

 

이 문제로 replaceAll 함수를 처음으로 사용해보았다.

replaceAll은 문자열에서 해당 단어가 몇 글자인지 비교할 필요없이, 전체 문자열에 대해 첫 번째 파라미터로 넣어준 값이 등장하면 두 번째 파라미터로 값을 바꿔준 뒤 새로운 문자열을 반환해준다.

 

바꾸고자 하는 숫자 문자열을 미리 배열에 만들어 두고, 반복문을 사용해서 각 문자열이 등장할 때마다 동일한 값을 정수로 표현하고 있는 인덱스 값으로 변환하도록 코드를 작성했다.

 

---

성능 요약

메모리: 83.1 MB, 시간: 3.13 ms