[백준] 9095 : 1, 2, 3 더하기

문제

재귀 개념

수 하나를 1, 2, 3 중에 선택하면 또 1, 2, 3 중에 수를 선택해야 함!
해당 수의 합이 되면 == 수 하나를 선택할 때마다 해당 수에서 그 수를 빼주고 나머지가 0이 되면 count를 하나 올려줘야 함!

풀이

import sys

t = int(sys.stdin.readline())
count = 0
def sol(n):
    count = 0
    if n == 0:
       return 1 # 합이 되면 return 1
    if n < 0:
        return 0 # 합이 되지 않으면 return 0
    for i in range(1, 4):
        count += sol(n-i) # count에 계속 쌓이고 마지막에 count를 return 해줌
    return count

for i in range(t):
    a = int(sys.stdin.readline())
    print(sol(a))

풀이 2.

import sys

t = int(sys.stdin.readline())
count = 0
def sol(n):
    global count
    if n == 0:
       count += 1
    if n < 0:
        return
    for i in range(1, 4):
        sol(n-i) # 1, 2, 3을 선택하고 빼줌

for i in range(t):
    a = int(sys.stdin.readline())
    sol(a)
    print(count)

좋은 웹페이지 즐겨찾기