Dice Combinations

from time import time
mod = int(1E9)+7
def dpWhole(inpt):
    myCahce = [0]*(inpt+1)
    myCahce[0] = 1
    for ii in range(1, inpt+1):
        su = 0
        for jj in range(1,7):
            if (ii - jj < 0): continue
            su += myCahce[ii-jj]
        myCahce[ii] = su % mod
    return(myCahce[-1])
def dpSmall(inpt):
    myCahce = [0]*6
    myCahce[0] = 1
    for ii in range(1, inpt+1):
        # roller += 1
        # roller %= 6
        su = 0
        for jj in range(1,7):
            if (ii - jj < 0): continue
            su += myCahce[(ii-jj)%6]
        myCahce[ii%6] = su % mod
    return(myCahce[inpt%6])

inpt = int(input())
# t0 = time()
# dpWholeCount = dpWhole(inpt)
# t1 = time()
dpSmallCount = dpSmall(inpt)
# t2 = time()
# print(dpWholeCount, dpSmallCount)
# print(t1-t0, t2-t1)
print(dpSmallCount)

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