矩阵(蓝桥杯)——python

题目如下:
我们采用递归的方法进行解答,当然也可以用动态规划进行自底向上的进行运算
将当前数放在第一行: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)

你可能感兴趣的:(经验分享,python,动态规划)