코딩테스트 입문_옹알이(1)_permutations

 

Lv. 0, 29%

 

문제

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

---

제한사항

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

---

입출력 예 #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"를 발음할 수 없기 때문에 할 수 없는 발음입니다.

 

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

프로그래머스

 

---

 

계획 :

 

baby 리스트로 만들 수 있는 모든 조합을 baby2에 몽땅 집어넣어

 

같은 문자열을 찾아 count를 1씩 증가시켜 출력시킬 생각이었고

 

마침 오늘 배운 itertools의 product 함수를 사용해 풀어보려 하였다 (조합형 이터레이터)

 

from itertools import product

baby = ['aya', 'ye', 'woo', 'ma']
baby2 = []


for i, v in product(baby, repeat=2):
    baby2.append(i+v)


def solution(babbling):
    count = 0
    for a in range(len(babbling)):
        if babbling[a] in baby or babbling[a] in baby2:
            count += 1
    return count


babbling = ["ayaye", "uuuma", "ye", "yemawoo", "ayaa"]
result = solution(babbling)
print(result)

 

일단 repeat=2로 짰더니 입출력1은 되는데 입출력2가 안 맞아서 repeat=4로 올렸더니..

 

Traceback (most recent call last): File "babbling.py", line 7, in <module> for i, v in product(baby, repeat=4): ValueError: too many values to unpack (expected 2)

 

product 함수에서 repeat 값이 너무 크다는 오류 발생.........

 

조합형 이터레이터 중 반복되는 요소가 출력되지 않는 permutations()를 사용해보기로 했다

 

from itertools import permutations

baby = ['aya', 'ye', 'woo', 'ma']
babybaby = []

for a, b in permutations(baby, 2):
    babybaby.append(a+b)
    # ['ayaye', 'ayawoo', ..., 'maye', 'mawoo']

for a, b, c in permutations(baby, 3):
    babybaby.append(a+b+c)
    # ['ayayewoo', 'ayayema', ..., 'mawooaya', 'mawooye']

for a, b, c, d in permutations(baby, 4):
    babybaby.append(a+b+c+d)
    # ['ayayewooma', ..., 'mawooyema', 'mawooyeaya']


def solution(babbling):
    count = 0
    for a in range(len(babbling)):
        if babbling[a] in baby or babbling[a] in babybaby:
            count += 1
    return count


babbling = ["ayaye", "uuuma", "ye", "yemawoo", "ayaa"]
result = solution(babbling)
print(result) # 3

 

for문을 따로 쓴 게 조금 아쉽지만..

 

정답률이 29%인 문제여서 그런지 처음으로 9점을 획득했다!

 

문득 예전에 뭣 모르고 풀이 보기로 점수 쓴 거 아깝네..