Golang Algorithm - 옹알이 (2) (프로그래머스)

옹알이 (2)
문제 설명
머쓱이는 태어난 지 11개월 된 조카를 돌보고 있습니다. 조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음과 네 가지 발음을 조합해서 만들 수 있는 발음밖에 하지 못하고 연속해서 같은 발음을 하는 것을 어려워합니다. 문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해주세요.

제한사항
1 ≤ babbling의 길이 ≤ 100
1 ≤ babbling[i]의 길이 ≤ 30
문자열은 알파벳 소문자로만 이루어져 있습니다.
입출력 예
babbling result
["aya", "yee", "u", "maa"] 1
["ayaye", "uuu", "yeye", "yemawoo", "ayaayaa"] 2
입출력 예 설명
입출력 예 #1

["aya", "yee", "u", "maa"]에서 발음할 수 있는 것은 "aya"뿐입니다. 따라서 1을 return합니다.
입출력 예 #2

["ayaye", "uuuma", "yeye", "yemawoo", "ayaayaa"]에서 발음할 수 있는 것은 "aya" + "ye" = "ayaye", "ye" + "ma" + "woo" = "yemawoo"로 2개입니다. "yeye"는 같은 발음이 연속되므로 발음할 수 없습니다. 따라서 2를 return합니다.
유의사항
네 가지를 붙여 만들 수 있는 발음 이외에는 어떤 발음도 할 수 없는 것으로 규정합니다. 예를 들어 "woowo"는 "woo"는 발음할 수 있지만 "wo"를 발음할 수 없기 때문에 할 수 없는 발음입니다.

 

 

package main

import (
	"strings"
)

func CheckStr(result string, char string) string {
	if strings.Contains(result, char) == true {
		result = strings.Replace(result, char, "1", -1)
	}
	return result
}

func solution(babbling []string) int {
	a := "aya"
	b := "ye"
	c := "woo"
	d := "ma"
	answer := 0

	for _, v := range babbling {

		if strings.Contains(v, "ayaaya") || strings.Contains(v, "yeye") || strings.Contains(v, "woowoo") || strings.Contains(v, "mama") {
			continue
		}

		v = CheckStr(v, a)
		v = CheckStr(v, b)
		v = CheckStr(v, c)
		v = CheckStr(v, d)
		v = strings.Replace(v, "1", "", -1)

		if len(v) == 0 {
			answer += 1
		}

	}

	return answer
}

func main() {
	// babbling := []string{"aya", "yee", "u", "maa"}
	babbling := []string{"ayaye", "uuu", "yeye", "yemawoo", "ayaayaa"}
	solution(babbling)
}