[SWEA] 5208 전기버스2

문제 링크: 5208. [파이썬 S/W 문제해결 구현] 5일차 - 전기버스2

코드

def func(i, val, cnt):
    global result

    for j in range(battery[i], 0, -1):
        if i + j >= n:
            result = min(result, cnt)
            continue
        if cnt < result:
            func(i + j, val + battery[i+j], cnt + 1)


for test_case in range(1, int(input()) + 1):
    battery = list(map(int, input().split()))
    n = battery[0]
    result = 1000
    func(1, battery[1], 0)
    print(f'#{test_case} {result}')