题目如下:
我们采用递归的方法进行解答,当然也可以用动态规划进行自底向上的进行运算
将当前数放在第一行:f[i][j] += f[i - 1][j]
将当前数放在第二行:f[i][j] += f[i][j - 1]
递归代码如下:
import sys
sys.setrecursionlimit(9000000) #这里设置大一些
from functools import lru_cache
@lru_cache(maxsize=2020) #maxsize为缓存的最大次数,为2n时效果最佳,为none则表示不限制
def dp_func(i,j):
if j>i:
return 0
else:
if j==0:
if i==0:
return 0
else:
return 1
else:
if j==i:
return dp_func(i,j-1)
else:
return (dp_func(i-1,j)+dp_func(i,j-1))%2020
print(dp_func(1010,1010))
因为python递归有自己的限制,我们调大一些,
动态规划代码如下:
import numpy
dp = numpy.zeros((1011, 1011))
dp[0][0] = 1
for i in range(0, 1011):
for j in range(0, 1011):
if i > j:
dp[i][j] += dp[i - 1][j]
if j:
dp[i][j] += dp[i][j - 1]
print(dp[1010][1010]%2020)