Dice Combinations

MOD = 10**9 + 7

def count_ways(n):
    # Initialize dp array to hold the number of ways to get each sum
    dp = [0] * (n + 1)
    
    # Base case: 1 way to get a sum of 0 (by not rolling the dice at all)
    dp[0] = 1
    
    # Fill the dp array using the recurrence relation
    for i in range(1, n + 1):
        dp[i] = 0
        for j in range(1, 7):
            if i - j >= 0:
                dp[i] = (dp[i] + dp[i - j]) % MOD
    
    return dp[n]

# Input
n = int(input())

# Output the number of ways to construct the sum n
print(count_ways(n))

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: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: TIME LIMIT EXCEEDED

Test 15

Verdict: TIME LIMIT EXCEEDED

Test 16

Verdict: TIME LIMIT EXCEEDED

Test 17

Verdict: TIME LIMIT EXCEEDED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: TIME LIMIT EXCEEDED

Test 20

Verdict: ACCEPTED