Python中的NZEC错误（Chef and Way）（动态编程）

我是动态规划的初学者。我用动态编程尝试了这个问题。 子任务1问题显示积极结果，但子任务2问题显示错误RE（NZEC）

这是我的密码：

def way(i, k, a, dp):
    if(i<=k):
        return a[i]*a[0]
    res = 0
    if dp[i] == 0:
        res = min(a[i]*way(i-j-1, k, a, dp) for j in range(k))
    else:
        res = dp[i]
    dp[i] = res
    return res

n, k = raw_input().split()
n = int(n)
k = int(k)
a = list(map(int, raw_input().split()))

dp = [0 for i in range(n)]

print way(n-1, k, a, dp)%1000000007

定义方式（i、k、a、dp）： 如果（i）