[Programmers] Lv.1 숫자 문자열과 영단어 (Go)

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

프로그래머스

 

문제 설명

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

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

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

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

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

 

 

제한사항

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

---

Solution

import (
	"errors"
	"strconv"
	"strings"
)

func solution(s string) int {
   	var sb strings.Builder
	var temp strings.Builder

	for _, ch := range s {

		if ch >= 48 && ch <= 57 {
			sb.WriteRune(ch)
			continue
		}

		temp.WriteRune(ch)
		sNum, err := convertWordToNumber(temp.String())

		if err != nil {
			continue
		}

		sb.WriteString(sNum)
		temp.Reset()
	}

	result, _ := strconv.Atoi(sb.String())

	return result
}


func convertWordToNumber(temp string) (string, error) {

	var n int
	
	switch temp {
	case "zero":
		n = 0
	case "one":
		n = 1
	case "two":
		n = 2
	case "three":
		n = 3
	case "four":
		n = 4
	case "five":
		n = 5
	case "six":
		n = 6
	case "seven":
		n = 7
	case "eight":
		n = 8
	case "nine":
		n = 9
	default:
    	     return "", errors.New("not a number word")
	}

	return strconv.Itoa(n), nil
}