[프로그래머스] Swift - 옹알이(2) (LV.1)

👆문제풀러 가기👆

문제 설명

머쓱이는 태어난 지 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

 

풀이 [ 메모리: 16.4mb, 최대시간:  0.91ms ] 

func solution(_ babbling:[String]) -> Int {
    let impossible = ["ayaaya", "yeye", "woowoo", "mama"] // 발음할 수 없는 단어
    let possible = ["aya", "ye", "woo", "ma"] // 발음할 수 있는 단어
    let babbling = babbling.filter { !$0.contains(impossible[0])
                                    && !$0.contains(impossible[1])
                                    && !$0.contains(impossible[2])
                                    && !$0.contains(impossible[3]) } // 발음할 수 없는 단어가 있으면 필터링
    var count = 0

    for babble in babbling {
        var stack = ""

        for s in babble {
            stack += String(s)
            
            // 발음할 수 있는 단어인지 확인
            if possible.contains(stack) {
                stack = ""
                continue
            }
            // 스택이 발음할 수 있는 단어의 최대길이를 넘기면 더이상 확인하지 않음.
            if stack.count > 3 {
                break
            }
        }
        // 발음할 수 없는 단어가 섞여있지 않으면 count + 1
        if stack.isEmpty {
            count += 1
        }
    }

    return count
}

문제를 받고 생각한 것.

1. 발음할 수 있는 단어(possible)와 없는 단어(impossible)를 배열로 만들기
2. 발음할 수 없는 단어(impossible)를 babbling(옹알이 예시)에서 필터링하기
3. 단어의 문자를 하나씩 쌓아(stack) 발음할 수 있는 단어인지 아닌지 체크하기
4. 전부 발음할 수 있는 단어이면, stack이 비어있을 테니 stack이 빈 값이면 count+1
5. (최적화) stack이 발음할 수 있는 최대 문자의 갯수(3)를 초과하면 더이상 해당 단어를 확인하지 않음.
6. 3~5 과정 반복하여 발음할 수 있는 단어의 수 세기

 

다른 풀이 [ 메모리: 16.4mb, 최대시간:  3.82ms ] 

func solution(_ babbling:[String]) -> Int {
    let possible = ["aya", "ye", "woo", "ma"]
    var count = 0

    for babble in babbling {
        var word = babble

        for i in 0..<possible.count {
            word = word.replacingOccurrences(of: possible[i], with: "\(i+1)")
        }

        if Int(word) != nil && !(word.contains("11") || word.contains("22")
                                 || word.contains("33") || word.contains("44"))  {
            count += 1
        }
    }

    return count
}

조금 더 심플한 코드.

1. 발음할 수 있는 단어를 숫자로 변경
2. 숫자가 연속적으로 반복되지 않거나 단어를 온전히 숫자로 바꿀 수 있으면 count + 1