Algorithm/Programmers

[Programmers] 연습문제 > 옹알이(1)

메린지 2023. 3. 28. 15:40

문제 설명

머쓱이는 태어난 지 6개월 된 조카를 돌보고 있습니다. 조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음을 최대 한 번씩 사용해 조합한(이어 붙인) 발음밖에 하지 못합니다. 문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해주세요.


제한사항
  • 1 ≤ babbling의 길이 ≤ 100
  • 1 ≤ babbling[i]의 길이 ≤ 15
  • babbling의 각 문자열에서 "aya", "ye", "woo", "ma"는 각각 최대 한 번씩만 등장합니다.
    • 즉, 각 문자열의 가능한 모든 부분 문자열 중에서 "aya", "ye", "woo", "ma"가 한 번씩만 등장합니다.
  • 문자열은 알파벳 소문자로만 이루어져 있습니다.

입출력 예babblingresult
["aya", "yee", "u", "maa", "wyeoo"] 1
["ayaye", "uuuma", "ye", "yemawoo", "ayaa"] 3

입출력 예 설명

입출력 예 #1

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

입출력 예 #2

  • ["ayaye", "uuuma", "ye", "yemawoo", "ayaa"]에서 발음할 수 있는 것은 "aya" + "ye" = "ayaye", "ye", "ye" + "ma" + "woo" = "yemawoo"로 3개입니다. 따라서 3을 return합니다.

유의사항
  • 네 가지를 붙여 만들 수 있는 발음 이외에는 어떤 발음도 할 수 없는 것으로 규정합니다. 예를 들어 "woowo"는 "woo"는 발음할 수 있지만 "wo"를 발음할 수 없기 때문에 할 수 없는 발음입니다.

[풀이]

일단 옹알이 리스트를 만들고, 옹알이 단어는 최대 1번씩만 등장하기 때문에

bab_list() 함수를 만들어 단어에서 옹알이 단어를 차례로 제거해가면서 남는 단어가 있는지 없는지 찾기.

대신 지우는 과정에서 ay woo a 같이 중간 단어가 제거되면서 아닌 단어도 걸릴 수 있기 때문에 이 부분을 유의하면서 코드를 짜는게 중요할 것 같음

대신 근데 이건 bab_list 특화이기 때문에 일반화하기는 힘든 코드 ㅎ 헤헷

 

정리하자면 

  • bab_list : 옹알리 리스트
  • find_bab : 주어진 단어 내 옹알이 탐색
def solution(babbling):
    answer = 0
    bab_list = ["aya", "ye", "woo", "ma"]

    def find_bab(bab_list, bab):
        for ba in bab_list:
            if ba in bab[:len(ba)]:
                bab = bab[len(ba):]
            elif ba in bab[-len(ba):]:
                bab = bab[:len(bab)-len(ba)]
        return bab

    for bab in babbling:
        while(1):
            tmp = find_bab(bab_list, bab)
            if (bab == tmp):
                break
            else:
                if tmp == '':
                    answer += 1
                    break
                elif len(tmp) < 2:
                    break
            bab = tmp
    return answer