[Python] 프로그래머스 모음사전 & 백준 9095

• 프로그래머스 모음사전

def find(data, p, step):
    if step == 6:
        return
    if p != '':
        data.append(p)
    for c in ['A','E','I','O','U']:
        find(data, "".join([p,c]), step+1)

def solution(word):
    answer = 0
    data = []
    find(data, "", 0)
    for i in range(len(data)):
        if data[i] == word:
            answer = i+1
            break
    return answer

 

 

• 백준 9095번 1,2,3 더하기

# 9095 1,2,3 더하기
def add_num(now, goal):
    if now > goal:
        return 0
    if now == goal:
        return 1
    cnt = 0
    for i in range(1, 4):
        cnt += add_num(now+i, goal)

    return cnt

t = int(input())
for _ in range(t):
    goal = int(input())
    print(add_num(0, goal))