Dice Combinations

n = int(input())
dice = [1, 2, 3, 4, 5, 6]

def count(n, memo):
    if n < 0:
        return 0
    elif n == 0:
        return 1
    if n in memo:
        return memo[n]
    
    s = 0
    for d in dice:
        s += count(n - d, memo)
    
    memo[n] = s
    return s

memo = {}
print(count(n, memo))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 13

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 14

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 15

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 16

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 17

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 18

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 19

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "/box/input/code.py", line 20, in <module>
    p...

Test 20

Verdict: WRONG ANSWER