3.25 阿里笔试题 1 动态规划 python

3.25 阿里笔试题 1

动态规划

作者：海森堡CSQ
链接：https://www.nowcoder.com/discuss/391530?type=1
来源：牛客网

第一题，给定一个数组n，比如
5 10 5 4 4
1  7  8 4 0
3  4  9 0 3
从每一列选择一个数，求出后一列减去前一列的绝对值的和的最小值

N = [[5, 3, 5, 4, 4], 
	 [1, 7, 8, 4, 0],
	 [3, 4, 9, 0, 3],
     [3, 4, 10, 0, 3]]
# 初始化 所有的和 因为第一行没有值为0
dp = [[0 for i in range(len(N[0]))]for j in range(len(N))]

for i in range(1, len(N[0])):
    for j in range(len(N)):
        # 遍历除了第一列的所有数
        min_list = []
        for m in range(len(N)):
            # 遍历前面一行的所有值 
            min_list.append(abs(N[j][i] - N[m][i - 1]) + dp[m][i - 1])
        # 赋值 所有的最小值和 之后的每一步都不需要前面的值
        # 故这里使用动态规划
        dp[j][i] = min(min_list)
result = float('inf')
for i in range(len(N)):
    if dp[i][-1] < result:
        result = dp[i][-1]