当前位置:   article > 正文

动态规划经典问题整理

动态规划经典问题

单序列:  

最长连续递增序列  最长递增子序列   最长递增子序列的个数  最大子序和  
最大整除子集

 爬楼梯最长定差子序列  爬楼梯 使用最小花费爬楼梯 比特位计数 旋转数字

把数字翻译成字符串 青蛙过河 最低加油次数  栅栏涂色 寻找数组的错位排列

所有子字符串中的元音  打家劫舍  打家劫舍 II  解决智力问题  带因子的二叉树 解决智力问题   

  分隔数组以得到最大和 可被三整除的最大和 检查数组是否存在有效划分 活字印刷 填充书架

统计特殊子序列的数目

双序列:

正则表达式匹配    最长公共子序列    不相交的线   最长重复子数组  通配符匹配

编辑距离   交错字符串   同源字符串检测  最长等差数列 最长的斐波那契子序列的长度 判断子序列 子序列的数目 卖木头块 填充书架

区间动态规划:

最少回文分割  猜数字大小 II 戳气球 为运算表达式设计优先级 预测赢家 最大平均值和的分组 多边形三角剖分的最低得分

状态转移动态规划:

删除一次得到子数组最大和  经过一次操作后的最大子数组和 最大交替子数组和 多米诺和托米诺平铺 统计只差一个字符的子串数目 学生出勤记录 II

背包问题:

零钱兑换  组合总和 Ⅳ  零钱兑换 II  分割等和子集  划分为k个相等的子集 

一和零(双序列背包) 目标和   单词拆分 分汤  买钢笔和铅笔的方案数

路径动态规划:

统计全为 1 的正方形子矩阵 最大正方形 三角形中最小路径之和 最大加号标志

最小路径和  不同路径 II  地下城游戏  01 矩阵 统计农场中肥沃金字塔的数目 出界的路径数  ​​​​​​矩阵中最长的连续1线段  出界的路径数   K 站中转内最便宜的航班  网格图中递增路径的数目

决策动态规划:

188. 买卖股票的最佳时机 IV  
新 21 点  926. 将字符串翻转到单调递增  

乘积最大子数组 乘积为正数的最长子数组长度 丑数 II  优美的排列 只有两个键的键盘  翻转游戏 II 计算分配糖果的不同方式 石子游戏 II

动态规划与树结合:

打家劫舍 III   二叉树中的最长交错路径 二叉搜索子树的最大键值和  统计可能的树根数目 树中距离之和

数位动态规划

至少有 1 位重复的数字  不含连续1的非负整数

倍增  

Pow(x, n)  在传球游戏中最大化函数值

单序列

1、最长递增子序列  

  1. for i in range(1, n):
  2. for j in range(0, i):
  3. if nums[j] < nums[i]:
  4. dp[i] = max(dp[i], dp[j] + 1)
  5. if dp[i] > maxl:
  6. maxl = dp[i]

注:采用动态规划的方法的时间复杂度为O(n^2), 最优的方法是采用二分插入的方法如下所示:

  1. LIS = [nums[0]]
  2. for num in nums[1:]:
  3. if num > LIS[-1]:
  4. LIS.append(num)
  5. else:
  6. index = bisect_left(LIS, num)
  7. LIS[index] = num
  8. return len(LIS)

 如果求单调不递增的子序列还需要num >= LIS[-1] 并且使用bisect_right,见使数组 K 递增的最少操作次数

2、最长递增子序列的个数

  1. nums = [-float('inf')] + nums + [float('inf')]
  2. n = len(nums)
  3. dp = [0] * n
  4. for i in range(0, n - 1):
  5. for j in range(i + 1, n):
  6. if nums[j] > nums[i]:
  7. dp[j] = max(dp[i] + 1, dp[j])
  8. path = [1] + [0] * (n - 1)
  9. for i in range(1, n):
  10. for j in range(0, i):
  11. if dp[i] == dp[j] + 1 and nums[i] > nums[j]:
  12. path[i] += path[j]
  13. return path[-1]

 同理,根据1的最优方法,我们可以对2的方法进行优化

3、最长连续递增序列

  1. for i in range(1, n):
  2. if nums[i - 1] < nums[i]:
  3. dp[i] = max(dp[i], dp[i - 1] + 1)
  4. if dp[i] > maxl:
  5. maxl = dp[i]

4、最大子序和

  1. for i in range(1, n):
  2. dp[i] = max(dp[i], dp[i - 1] + nums[i])
  3. if dp[i] > maxl:
  4. maxl = dp[i]

5、最大整除子集

  1. for i in range(n):
  2. for j in range(i + 1, n):
  3. if nums[j] % nums[i] == 0:
  4. dp[j] = max(dp[j], dp[i] + 1)

6、最长定差子序列

  1. d = {}
  2. for a in arr:
  3. d[a] = d.get(a - difference, 0) + 1
  4. return max(d.values())

7、爬楼梯

  1. for i in range(2, n + 1):
  2. dp[i] = dp[i - 1] + dp[i - 2]

8、寻找数组的错位排列

  1. for i in range(2, n):
  2. dp[i] = (i - 1) * (dp[i - 1] + dp[i - 2])

9、 使用最小花费爬楼梯

  1. for i in range(2, len(cost) + 1):
  2. dp[i] = min(dp[i - 2] + cost[i - 2], dp[i - 1] + cost[i - 1])

10、比特位计数

  1. for i in range(1, n + 1):
  2. if i & 1 == 0:
  3. dp[i] = dp[i >> 1]
  4. else:
  5. dp[i] = dp[i - 1] + 1

11、旋转数字

  1. class Solution:
  2. def rotatedDigits(self, N: int) -> int:
  3. ans, dp = 0, [0, 0, 1, -1, -1, 1, 1, -1, 0, 1] + [0] * (N - 9)
  4. for i in range(N + 1):
  5. dp[i] = -1 in (dp[i // 10], dp[i % 10]) and -1 or dp[i // 10] | dp[i % 10]
  6. ans += dp[i] == 1
  7. return ans

12、打家劫舍 II

  1. a, b = nums[0], max(nums[0], nums[1])
  2. for i in range(2, len(nums) - 1):
  3. a = max(a + nums[i], b)
  4. b, a = a, b
  5. res1 = b
  6. a, b = nums[1], max(nums[1], nums[2])
  7. for i in range(3, len(nums)):
  8. a = max(a + nums[i], b)
  9. b, a = a, b
  10. return max(b, res1)

 13、解决智力问题

  1. n = len(questions)
  2. dp = [0] * (n + 1)
  3. for i, q in enumerate(questions):
  4. dp[i + 1] = max(dp[i + 1], dp[i])
  5. j = min(i + q[1] + 1, n)
  6. dp[j] = max(dp[j], dp[i] + q[0])
  7. return dp[n]

14、栅栏涂色

  1. if n == 1: return k
  2. dp = [0] * (n + 1)
  3. dp[1], dp[2] = k, k * k
  4. for i in range(3, n + 1):
  5. dp[i] = dp[i - 1] * (k - 1) + dp[i - 2] * (k - 1)
  6. return dp[-1]

15、带因子的二叉树

  1. arr.sort()
  2. for i, a in enumerate(arr):
  3. dp[a] = 1
  4. for j in range(i):
  5. if a % arr[j] == 0 and a // arr[j] in set(arr):
  6. dp[a] += dp[arr[j]] * dp[a // arr[j]]
  7. return sum(count for num, count in dp.items())

16、分隔数组以得到最大和

  1. n = len(arr)
  2. dp = [0] * (n + 1)
  3. for i in range(1, n + 1):
  4. maxl = 0
  5. for j in range(i - 1, max(i - k, 0) - 1, - 1):
  6. maxl = max(maxl, arr[j])
  7. dp[i] = max(dp[i], dp[j] + maxl * (i - j))
  8. return dp[-1]

17、可被三整除的最大和

  1. dp = [0] * div
  2. for n in nums:
  3. nw = [0] * div
  4. for d in range(div):
  5. nw[d] = dp[d] + n
  6. for t in nw:
  7. dp[t % div] = max(dp[t % div], t)
  8. return dp[0]

18、检查数组是否存在有效划分

  1. dp = [True] + [False] * len(nums)
  2. for i in range(1, len(nums)):
  3. dp[i + 1] |= dp[i - 1] and nums[i] == nums[i - 1]
  4. dp[i + 1] |= dp[i - 2] and nums[i] == nums[i - 1] and nums[i] == nums[i - 2]
  5. dp[i + 1] |= dp[i - 2] and nums[i] - nums[i - 1] == 1 and nums[i - 1] - nums[i - 2] == 1
  6. return dp[-1]

该题也可以用贪心去做,但是贪心的过程较为繁琐,而且时间复杂度也为O(n) 

19、活字印刷 (动态规划 + 组合数学)

  1. dp, cv, alllen = [1] + [0] * len(tiles), Counter(tiles).values(), 0
  2. for v in cv:
  3. alllen += v
  4. for i in range(alllen, 0, -1):
  5. dp[i] += sum(dp[i - j] * math.comb(i, j) for j in range(1, min(i,v) + 1))
  6. return sum(dp) - 1

20、填充书架 

  1. for i in range(1, len(books) + 1):
  2. tmp_width, j, h = 0, i, 0
  3. while j > 0:
  4. tmp_width += books[j - 1][0]
  5. if tmp_width > shelf_width:
  6. break
  7. h = max(h, books[j - 1][1])
  8. dp[i] = min(dp[i], dp[j - 1] + h)
  9. j -= 1
  10. return dp[-1]

21、统计特殊子序列的数目 

  1. class Solution:
  2. def countSpecialSubsequences(self, nums: List[int]) -> int:
  3. mod = 10**9 + 7
  4. f0 = f1 = f2 = 0
  5. for num in nums:
  6. if num == 0:
  7. f0 = (f0 * 2 + 1) % mod
  8. elif num == 1:
  9. f1 = (f1 * 2 + f0) % mod
  10. else:
  11. f2 = (f2 * 2 + f1) % mod
  12. return f2

双序列

1、最长公共子序列

  1. for i in range(1, len(text1) + 1):
  2. for j in range(1, len(text2) + 1):
  3. if text1[i - 1] == text2[j - 1]:
  4. dp[i][j] = dp[i - 1][j - 1] + 1
  5. else:
  6. dp[i][j] = max(dp[i - 1][j], dp[i][j - 1])

2、 不相交的线

  1. for i in range(1, len(nums1) + 1):
  2. for j in range(1, len(nums2) + 1):
  3. if nums1[i - 1] == nums2[j - 1]:
  4. dp[i][j] = dp[i - 1][j - 1] + 1
  5. else:
  6. dp[i][j] = max(dp[i][j - 1], dp[i - 1][j])

3、最长重复子数组

  1. m, n = len(nums1), len(nums2)
  2. dp = [[0] * (n + 1) for _ in range(m + 1)]
  3. maxl = 0
  4. for i in range(1, m + 1):
  5. for j in range(1, n + 1):
  6. if nums1[i - 1] == nums2[j - 1]:
  7. dp[i][j] = dp[i - 1][j - 1] + 1
  8. if dp[i][j] > maxl:
  9. maxl = dp[i][j]
  10. return maxl

注:使用滑动窗口的方法可以使时间复杂度降到O((m + n) *min(m, n)), 空间复杂度降为O(1)

  1. m, n = len(nums1), len(nums2)
  2. maxl = 0
  3. for d in range(1 - m, n):
  4. l = 0
  5. for i in range(max(0, -d), min(m, n - d)):
  6. if nums1[i] == nums2[i + d]:
  7. l += 1
  8. maxl = max(maxl, l)
  9. else: l = 0
  10. return maxl

 还可以使用二分查找+哈希的方法使得时间复杂度降为O((m + n) *log(min(m, n)))

4、 通配符匹配

  1. for i in range(1, len(p) + 1):
  2. for j in range(1, len(s) + 1):
  3. if p[i - 1] == '?':
  4. dp[i][j] = dp[i - 1][j - 1]
  5. elif p[i - 1] == '*':
  6. dp[i][j] = dp[i - 1][j] or dp[i - 1][j - 1] or dp[i][j - 1]
  7. elif p[i - 1] == s[j - 1]:
  8. dp[i][j] = dp[i - 1][j - 1]
  9. else:
  10. dp[i][j] = False

5、正则表达式匹配

  1. for i in range(1, len(p) + 1):
  2. for j in range(1, len(s) + 1):
  3. if p[i - 1] == s[j - 1] or p[i - 1] == '.':
  4. dp[i][j] = dp[i - 1][j - 1]
  5. elif p[i - 1] == '*':
  6. if p[i - 2] == s[j - 1] or p[i - 2] == '.':
  7. dp[i][j] = dp[i - 2][j] or dp[i][j - 1]
  8. else:
  9. dp[i][j] = dp[i - 2][j]
  10. else:
  11. dp[i][j] = False

6、最长等差数列

  1. dp = dict()
  2. for cur in range(1, len(nums)):
  3. for prev in range(cur):
  4. diff = nums[cur] - nums[prev]
  5. dp[(cur, diff)] = dp.get((prev, diff), 1) + 1
  6. return max(dp.values())

7、最长的斐波那契子序列的长度

  1. index = {x: i for i, x in enumerate(A)}
  2. longest = collections.defaultdict(lambda: 2)
  3. ans = 0
  4. for k, z in enumerate(A):
  5. for j in range(k):
  6. i = index.get(z - A[j], None)
  7. if i is not None and i < j:
  8. cand = longest[j, k] = longest[i, j] + 1
  9. ans = max(ans, cand)
  10. return ans if ans >= 3 else 0

8、判断子序列

  1. for i in range(1, len(s) + 1):
  2. for j in range(1, len(t) + 1):
  3. if s[i - 1] == t[j - 1]:
  4. dp[i][j] = dp[i - 1][j - 1]
  5. else:
  6. dp[i][j] = dp[i][j - 1]

该方法可以使用双指针进行求解 

9、子序列的数目

  1. for i in range(1, len(t) + 1):
  2. for j in range(i, len(s) + 1):
  3. if t[i - 1] == s[j - 1]:
  4. dp[j] = dp[i - 1][j - 1] + dp[i][j - 1]
  5. else:
  6. dp[j] = dp[i][j - 1]

10、 通过删除字母匹配到字典里最长单词

 类似于判断子序列,可以采用双指针进行求解,本题是一个长串匹配多个短串,可以采用动态的方法构造状态转移矩阵,减少长串指针的空跳

构造过程:

  1. m, set_s = len(s), set(s)
  2. f = [defaultdict(lambda: m) for _ in range(m + 1)]
  3. for i in reversed(range(m)):
  4. for c in set_s:
  5. if s[i] == c:
  6. f[i][ord(c) - ord('a')] = i
  7. else:
  8. f[i][ord(c) - ord('a')] = f[i + 1][ord(c) - ord('a')]

匹配过程:

  1. res = ""
  2. for t in dictionary:
  3. match = True
  4. j = 0
  5. for i in range(len(t)):
  6. if f[j][ord(t[i]) - ord('a')] == m:
  7. match = False
  8. break
  9. j = f[j][ord(t[i]) - ord('a')] + 1
  10. if match:
  11. if len(t) > len(res) or (len(t) == len(res) and t < res):
  12. res = t
  13. return res

 11、交错字符串

  1. for i in range(len(s1) + 1):
  2. for j in range(len(s2) + 1):
  3. if i > 0 and s1[i - 1] == s3[i + j - 1]:
  4. dp[i][j] = dp[i][j] or dp[i - 1][j]
  5. if j > 0 and s2[j - 1] == s3[i + j - 1]:
  6. dp[i][j] = dp[i][j] or dp[i][j - 1]

 12、编辑距离

  1. for i in range(1, len(word1)+1):
  2. for j in range(1, len(word2)+1):
  3. if word1[i - 1] == word2[j - 1]:
  4. dp[i][j] = dp[i - 1][j -1]
  5. else:
  6. dp[i][j] = min(dp[i-1][j] + 1, dp[i][j-1] + 1, dp[i-1][j-1] + 1)

13、卖木头块

  1. for i in range(1, m + 1):
  2. for j in range(1, n + 1):
  3. dp[i][j] = mp[i, j]
  4. for k in range(1, j + 1):
  5. dp[i][j] = max(dp[i][j], dp[i][j - k] + dp[i][k])
  6. for l in range(1, i + 1):
  7. dp[i][j] = max(dp[i][j], dp[i - l][j] + dp[l][j])

区间动态规划

1、猜数字大小 II

  1. for j in range(2, n + 1):
  2. for i in reversed(range(1, j)):
  3. f[i][j] = min(max(f[i][x - 1], f[x + 1][j]) + x for x in range(i, j))
  4. return f[1][n]

2、最少回文分割

  1. #求取区间是否是回文串
  2. for l in range(n-1, -1, -1):
  3. for r in range(l + 1, n):
  4. if s[l] == s[r]:
  5. if l + 1 == r:
  6. f[l][r] = True
  7. else:
  8. f[l][r] = f[l + 1][r - 1]
  1. #求取最小分割
  2. for i in range(1, len(s)):
  3. dp[i] = 0 if f[0][i] else i
  4. for j in range(1, i + 1):
  5. if f[j][i]:
  6. dp[i] = min(dp[i], dp[j - 1] + 1)

3、戳气球 

  1. for i in reversed(range(n)):
  2. for j in range(i + 2, n + 2):
  3. for k in range(i + 1, j):
  4. val = nums[i] * nums[k] * nums[j]
  5. dp[i][j] = max(dp[i][j], dp[i][k] + dp[k][j] + val)

4、为运算表达式设计优先级 

  1. nums = [int(num) for num in re.split("[+\-*]", input)]
  2. opts = re.sub('[0-9]', "", input)
  3. opt_m = {'+':(lambda x, y:x+ y), '-':(lambda x, y:x - y), '*':(lambda x, y:x * y)}
  4. dp = [[[] for _ in range(len(nums))] for _ in range(len(nums))]
  5. for j in range(len(nums)):
  6. for i in reversed(range(j + 1)):
  7. if i == j:
  8. dp[i][j].append(nums[i])
  9. else:
  10. for k in range(i, j):
  11. dp[i][j].extend([opt_m[opts[k]](x,y) for x, y in product(dp[i][k], dp[k + 1][j])])

5、预测赢家

  1. length = len(nums)
  2. dp = [[0] * length for _ in range(length)]
  3. for i, num in enumerate(nums):
  4. dp[i][i] = num
  5. for i in range(length - 2, -1, -1):
  6. for j in range(i + 1, length):
  7. dp[i][j] = max(nums[i] - dp[i + 1][j], nums[j] - dp[i][j - 1])
  8. return dp[0][length - 1] >= 0

6、 最大平均值和的分组

  1. P, N = [0], len(nums)
  2. for x in nums: P.append(P[-1] + x)
  3. def average(i, j):
  4. return (P[j] - P[i]) / float(j - i)
  5. dp = [average(i, N) for i in range(N)]
  6. for k in range(K-1):
  7. for i in range(N):
  8. for j in range(i+1, N):
  9. dp[i] = max(dp[i], average(i, j) + dp[j])
  10. return dp[0]

7、多边形三角剖分的最低得分

  1. n = len(A)
  2. dp = [[inf] * n for _ in range(n)]
  3. for i in range(n - 1):
  4. dp[i][i + 1] = 0
  5. for i in reversed(range(n)):
  6. for j in range(i + 2, n):
  7. for k in range(i + 1, j):
  8. dp[i][j] = min(dp[i][j], dp[i][k] + A[i] * A[k] * A[j] + dp[k][j])
  9. return dp[0][-1]

状态转移动态规划

1、删除一次得到子数组最大和

  1. dp0, dp1, ans = arr[0], 0, arr[0]
  2. for i in arr[1:]:
  3. dp1 = max(dp1 + i, dp0)
  4. dp0 = max(dp0 + i, i)
  5. ans = max(ans, dp0, dp1)
  6. return ans

2、删除一次得到子数组最大和

  1. dp0, dp1 = nums[0], max(nums[0], nums[0] ** 2)
  2. maxl = max(dp0, dp1)
  3. for i in range(1, len(nums)):
  4. dp1 = max(dp1 + nums[i], max(dp0, 0) + nums[i] ** 2)
  5. dp0 = max(dp0 + nums[i], nums[i])
  6. maxl = max(maxl, dp0, dp1)
  7. return maxl

3、最大交替子数组和

  1. ans, dp0, dp1 = nums[0], nums[0], -inf
  2. for num in nums[1:]:
  3. dp0, dp1 = max(num, dp1 + num), dp0 - num
  4. ans = max(ans, dp0, dp1)
  5. return ans

 4、多米诺和托米诺平铺 

  1. #dp1[k] 总计k列,最后一列满的种数
  2. #dp2[k] 总计k列,最后一列差一个满
  3. dp1, dp2 = [0] * (max(3, n + 1)), [0] * (max(3, n + 1))
  4. dp2[1], dp2[2] = 0, 2
  5. dp1[1], dp1[2] = 1, 2
  6. for i in range(3, n + 1):
  7. dp2[i] = dp2[i - 1] + 2 *dp1[i - 2]
  8. dp1[i] = dp1[i - 1] + dp1[i - 2] + dp2[i - 1]
  9. return dp1[n] % (10 ** 9 + 7)

5、统计只差一个字符的子串数目

        cnt[i-1][j-1]为以s[i-1]t[j-1]结尾的恰好只有一个字符不同的子串对数目,same[i][j]记录以s[i]t[j]结尾的连续相同的字符数量。

  1. class Solution:
  2. def countSubstrings(self, s: str, t: str) -> int:
  3. m, n = len(s), len(t)
  4. dp = [(0, 0)]*(n+1) # 记录 (以s[i]和t[j]结尾的 满足条件的子字符串对数目, 连续相同的位数)
  5. tot = 0 # 总的满足条件的子字符串对数目
  6. for i in range(m):
  7. last = dp[0]
  8. for j in range(n):
  9. if s[i] == t[j]:
  10. cur = (dp[j][0], dp[j][1] + 1)
  11. tot += dp[j][0]
  12. else:
  13. cur = (dp[j][1] + 1, 0)
  14. tot += dp[j][1] + 1
  15. dp[j], last = last, cur
  16. dp[n] = last
  17. return tot

6、 学生出勤记录 II

  1. class Solution:
  2. def checkRecord(self, n: int) -> int:
  3. mod = 10 ** 9 + 7
  4. dp = [[0] * 2 for _ in range(3)]
  5. dp[0][0] = 1
  6. for i in range(n):
  7. tp = [[0] * 2 for _ in range(3)]
  8. #以A为结尾
  9. for j in range(3):
  10. tp[0][1] = (tp[0][1] + dp[j][0]) % mod
  11. #以P为结尾
  12. for j in range(3):
  13. for k in range(2):
  14. tp[0][k] = (tp[0][k] + dp[j][k]) % mod
  15. #以L为结尾
  16. for j in range(1, 3):
  17. for k in range(2):
  18. tp[j][k] = (tp[j][k] + dp[j - 1][k]) % mod
  19. dp = tp
  20. return sum([sum(p) for p in dp]) % mod

本题还可以使用快速幂的方法将时间复杂度降为O(logn)

 代码如下:

  1. class Solution:
  2. def checkRecord(self, n: int) -> int:
  3. MOD = 10**9 + 7
  4. mat = [
  5. [1, 1, 0, 1, 0, 0],
  6. [1, 0, 1, 1, 0, 0],
  7. [1, 0, 0, 1, 0, 0],
  8. [0, 0, 0, 1, 1, 0],
  9. [0, 0, 0, 1, 0, 1],
  10. [0, 0, 0, 1, 0, 0],
  11. ]
  12. def multiply(a: List[List[int]], b: List[List[int]]) -> List[List[int]]:
  13. rows, columns, temp = len(a), len(b[0]), len(b)
  14. c = [[0] * columns for _ in range(rows)]
  15. for i in range(rows):
  16. for j in range(columns):
  17. for k in range(temp):
  18. c[i][j] += a[i][k] * b[k][j]
  19. c[i][j] %= MOD
  20. return c
  21. def matrixPow(mat: List[List[int]], n: int) -> List[List[int]]:
  22. ret = [[1, 0, 0, 0, 0, 0]]
  23. while n > 0:
  24. if (n & 1) == 1:
  25. ret = multiply(ret, mat)
  26. n >>= 1
  27. mat = multiply(mat, mat)
  28. return ret
  29. res = matrixPow(mat, n)
  30. ans = sum(res[0])
  31. return ans % MOD

背包问题

1、零钱兑换

  1. for i in range(amount + 1):
  2. for coin in coins:
  3. if i - coin >= 0:
  4. dp[i] = min(dp[i], 1 + dp[i - coin])

2、零钱兑换II  (单序列完全背包,正向遍历, 无放入顺序)

  1. for coin in coins:
  2. for i in range(amount + 1):
  3. if i - coin >= 0:
  4. dp[i] = dp[i] + dp[i - coin]

3、买钢笔和铅笔的方案数

  1. dp = [1] + (total) * [0]
  2. for cost in (cost1, cost2):
  3. for i in range(1, total + 1):
  4. if i - cost >= 0:
  5. dp[i] += dp[i - cost]
  6. return sum(dp)

该题最优的方法是使用乘法原理:

  1. res = 0
  2. for i in range(total // cost1 + 1):
  3. res += math.ceil((total-cost1 * i) // cost2 + 1)

3、组合总和Ⅳ (单序列完全背包,正向遍历, 有放入顺序)

  1. for i in range(target + 1):
  2. for num in nums:
  3. if i - num >= 0:
  4. dp[i] = dp[i] + dp[i - num]

4、 一和零(双序列01背包,逆向遍历)

  1. for s in strs:
  2. c0, c1 = s.count('0'), s.count('1')
  3. for i in reversed(range(c0, m + 1)):
  4. for j in reversed(range(c1, n + 1)):
  5. dp[i][j] = max(dp[i][j], dp[i - c0][j - c1] + 1)

5、分割等和子集 (单序列01背包)

  1. for i in range(1, len(nums)):
  2. for j in range(1, target + 1):
  3. dp[i][j] = dp[i - 1][j] | dp[i - 1][j - nums[i]]

6、划分为k个相等的子集 

  1. size, n = 1 << len(nums), len(nums)
  2. dp, s = [False] * size, [0] * size
  3. dp[0] = True
  4. for i in range(size):
  5. if not dp[i]: continue
  6. for j in range(n):
  7. if i & (1 << j) != 0:
  8. continue
  9. next_i = i | (1 << j)
  10. if dp[next_i]: continue
  11. if (s[i] % side) + nums[j] <= side:
  12. s[next_i] = s[i] + nums[j]
  13. dp[next_i] = True
  14. else: break
  15. return dp[-1]

7、完全平方数

  1. for i in range(1, n + 1):
  2. j = 1
  3. while j * j <= n:
  4. if i - j * j >= 0:
  5. dp[i] = min(dp[i - j * j] + 1, dp[i])
  6. j += 1
  7. else:
  8. break

8、目标和

  1. for num in nums:
  2. for j in range((s + t) // 2, num - 1, -1):
  3. dp[j] += dp[j - num]

 9、单词拆分

  1. for i in range(len(s)):
  2. for word in wordDict:
  3. if s[i + 1 - len(word): i + 1] == word:
  4. dp[i + 1] = dp[i + 1] or dp[i + 1 - len(word)]

10、分汤(双序列完全背包,正向遍历)

  1. for i in range(1, n + 1):
  2. for j in range(1, n + 1):
  3. dp[i][j] = 0.25 * (dp[max(i - 4, 0)][j] + dp[max(i - 3, 0)][j - 1] +
  4. dp[max(i - 2, 0)][max(j - 2, 0)] + dp[i - 1][max(j - 3, 0)])

当维数过大,或者维数不确定时,使用自底向上动态规划容易造成维数灾难,例如:638. 大礼包,所以应该使用自顶向下的记忆化搜索的方法。

对于所给数据如果有一定特征,可以使用贪心的算法来优化,例如:和为 K 的最少斐波那契数字数目

路径动态规划

1、统计全为 1 的正方形子矩阵

  1. for i in range(0, m):
  2. for j in range(0, n):
  3. if i == 0 or j == 0:
  4. dp[i][j] = matrix[i][j]
  5. elif matrix[i][j] == 1:
  6. dp[i][j] = min(dp[i-1][j-1], dp[i][j-1], dp[i-1][j]) + 1
  7. c += dp[i][j]

2、最大正方形

  1. for i in range(0, m):
  2. for j in range(0, n):
  3. if i == 0 or j == 0:
  4. dp[i][j] = int(matrix[i][j])
  5. elif matrix[i][j] == '1':
  6. dp[i][j] = min(dp[i-1][j-1], dp[i][j-1], dp[i-1][j]) + 1
  7. max_c = max(max_c, dp[i][j])

3、 三角形中最小路径之和

  1. for i in range(1, len(triangle)):
  2. for j in range(len(triangle[i])):
  3. if j == 0:
  4. triangle[i][j] = triangle[i -1][0] + triangle[i][j]
  5. elif j == len(triangle[i]) - 1:
  6. triangle[i][j] = triangle[i -1][j - 1] + triangle[i][j]
  7. else:
  8. triangle[i][j] = min(triangle[i -1][j - 1], triangle[i -1][j]) + triangle[i][j]

 4、最小路径和

  1. for i in range(len(grid)):
  2. for j in range(len(grid[i])):
  3. if i == 0 and j > 0:
  4. dp[j] = dp[j - 1] + grid[i][j]
  5. elif j == 0 and i > 0:
  6. dp[j] = dp[j] + grid[i][j]
  7. elif j > 0 and i > 0:
  8. dp[j] = min(dp[j - 1] + grid[i][j], dp[j] + grid[i][j])

5、不同路径 II

  1. for i in range(0, m):
  2. for j in range(0, n):
  3. if obstacleGrid[i][j] == 1:
  4. dp[j] = 0
  5. elif i > 0 and j > 0:
  6. dp[j] = dp[j] + dp[j - 1]
  7. elif j > 0 and i == 0:
  8. dp[j] = dp[j - 1]

6、地下城游戏(不同路径的逆向遍历)

  1. m, n = len(dungeon), len(dungeon[0])
  2. dp = [[[0, 0] for _ in range(n)] for _ in range(m)]
  3. for i, j in product(reversed(range(m)), reversed(range(n))):
  4. if i == m - 1 and j == n - 1:
  5. dp[i][j] = min(0, dungeon[i][j])
  6. elif i == m - 1:
  7. dp[i][j] = min(0, dp[i][j + 1] + dungeon[i][j])
  8. elif j == n - 1:
  9. dp[i][j] = min(0, dp[i + 1][j] + dungeon[i][j])
  10. else:
  11. dp[i][j] = min(0, max(dp[i][j + 1], dp[i + 1][j]) + dungeon[i][j])
  12. return -dp[0][0] + 1

7、01 矩阵

  1. for i in range(m):
  2. for j in range(n):
  3. if mat[i][j] == 0:
  4. dist[i][j] = 0
  5. if i > 0:
  6. dist[i][j] = min(dist[i][j], dist[i - 1][j] + 1)
  7. if j > 0:
  8. dist[i][j] = min(dist[i][j], dist[i][j - 1] + 1)
  9. for i in range(m - 1, -1, -1):
  10. for j in range(n - 1, -1, -1):
  11. if i < m - 1:
  12. dist[i][j] = min(dist[i][j], dist[i + 1][j] + 1)
  13. if j < n - 1:
  14. dist[i][j] = min(dist[i][j], dist[i][j + 1] + 1)

8、统计农场中肥沃金字塔的数目

  1. for i in range(m):
  2. for j in range(n):
  3. if grid[i][j] == 1:
  4. if j == 0 or j == n - 1:
  5. dp[i][j] = 1
  6. else:
  7. dp[i][j] = min(dp[i - 1][j - 1], dp[i - 1][j], dp[i - 1][j + 1]) + 1
  8. else:
  9. dp[i][j] = 0

9、出界的路径数 

  1. for k in range(maxMove):
  2. for i in range(m):
  3. for j in range(n):
  4. if k == 0:
  5. dp[i][j][k] += 1 if i - 1 < 0 else 0
  6. dp[i][j][k]+= 1 if i + 1 >= m else 0
  7. dp[i][j][k]+= 1 if j - 1 < 0 else 0
  8. dp[i][j][k] += 1 if j + 1 >= n else 0
  9. else:
  10. dp[i][j][k] += dp[i - 1][j][k - 1] if i > 0 else 0
  11. dp[i][j][k] += dp[i + 1][j][k - 1] if i + 1 < m else 0
  12. dp[i][j][k] += dp[i][j - 1][k - 1] if j > 0 else 0
  13. dp[i][j][k] += dp[i][j + 1][k - 1] if j + 1 < n else 0
  14. return sum(dp[startRow][startColumn])

10、矩阵中最长的连续1线段

  1. m, n = len(mat), len(mat[0])
  2. #用dp[i][j][k]记录,k分别表示水平、垂直、对角、反对角四个状态
  3. dp = [[[0]*4 for _ in range(n+2)] for _ in range(m+2)]
  4. ans = 0
  5. for i in range(1, m+1):
  6. for j in range(1, n+1):
  7. if mat[i-1][j-1] == 1:
  8. dp[i][j][0] = dp[i-1][j][0] + 1
  9. dp[i][j][1] = dp[i][j-1][1] + 1
  10. dp[i][j][2] = dp[i-1][j-1][2] + 1
  11. dp[i][j][3] = dp[i-1][j+1][3] + 1
  12. ans = max(ans, max(dp[i][j]))
  13. return ans

 11、K 站中转内最便宜的航班

  1. dp = [float('inf') for _ in range(n)]
  2. dp[src] = 0
  3. for i in range(K+1):
  4. tmp = dp[:]
  5. for u, v, w in flights:
  6. dp[v] = min(dp[v],tmp[u] + w)
  7. return dp[dst] if dp[dst] != float('inf') else -1

 也可以使用堆的方法进行遍历

  1. graph = collections.defaultdict(dict)
  2. for start,end,cost in flights:
  3. graph[start][end] = cost
  4. queue, v = [(0,0,src)], set()
  5. while queue:
  6. cost, k, pos = heapq.heappop(queue)
  7. if pos == dst: return cost
  8. if (pos, k) in v: continue
  9. v.add((pos, k))
  10. if k > K: continue
  11. for nxpos, nxcost in graph[pos].items():
  12. if (nxpos, k + 1) not in v:
  13. heapq.heappush(queue,(cost + nxcost, k + 1, nxpos))
  14. return -1

12、网格图中递增路径的数目

  1. class Solution:
  2. def countPaths(self, grid: List[List[int]]) -> int:
  3. m, n = len(grid), len(grid[0])
  4. dp = [[1 for i in range(n)] for j in range(m)]
  5. s = []
  6. mod = 10 ** 9 + 7
  7. for i in range(m):
  8. for j in range(n):
  9. s.append([grid[i][j], i, j])
  10. s.sort()
  11. while s:
  12. _, i, j = s.pop()#从尾部弹出则是从大到小遍历
  13. for x, y in [[i + 1, j], [i - 1, j], [i, j + 1], [i, j - 1]]:
  14. if 0 <= x < m and 0 <= y < n and grid[x][y] > _:
  15. dp[i][j] += dp[x][y]
  16. dp[i][j] %= mod
  17. return sum(sum(i) for i in dp) % mod

使用记忆化深搜会更简单:

  1. class Solution:
  2. def countPaths(self, grid: List[List[int]]) -> int:
  3. m, n = len(grid), len(grid[0])
  4. @cache
  5. def dfs(i, j):
  6. res = 1
  7. for dx, dy in [(1, 0), (-1, 0), (0, 1), (0, -1)]:
  8. ni, nj = i + dx, j + dy
  9. if 0 <= ni < m and 0 <= nj < n and grid[ni][nj] > grid[i][j]:
  10. res = (res + dfs(ni, nj)) % (10 ** 9 + 7)
  11. return res
  12. res = 0
  13. for i, j in product(range(m), range(n)):
  14. res = (res + dfs(i, j)) % (10 ** 9 + 7)
  15. return res

        注:有些问题只适合于记忆化深搜,例如:将整数按权重排序,如果利用dp的方法,如果要通过所有测试用例,dp长度至少为300 * 最高值 + 3, 而且运行时间远大于记忆化深搜。

决策动态规划

1、买卖股票的最佳时机 IV

  1. # dp[i][kk][0]表示第i天第k笔交易不持有股票, dp[i][kk][1]表示第i天第k笔交易持有股票
  2. for i in range(1, n):
  3. for kk in range(1, k + 1):
  4. dp[i][kk][0] = max(dp[i-1][kk][0], dp[i-1][kk][1] + prices[i])
  5. dp[i][kk][1] = max(dp[i-1][kk][1], dp[i-1][kk-1][0] - prices[i])
  6. return dp[n-1][k][0]

2、 将字符串翻转到单调递增

  1. #a0最后一位为0最少需要翻转次数,a1最后一位为1最少需要翻转次数
  2. for c in s:
  3. if c == '0':
  4. a1 = min(a0, a1) + 1
  5. else:
  6. a1 = min(a0, a1)
  7. a0 = a0 + 1

3、乘积最大子数组 

  1. maxF, minF, ans = nums[0], nums[0], nums[0]
  2. length = len(nums)
  3. for i in range(1, length):
  4. mx, mn = maxF, minF # 只用两个变量来维护i−1时刻的状态,优化空间
  5. maxF = max(mx * nums[i], nums[i], mn * nums[i])
  6. minF = min(mn * nums[i], nums[i], mx * nums[i])
  7. ans = max(maxF, ans)
  8. return ans

 4、乘积为正数的最长子数组长度

  1. class Solution:
  2. def getMaxLen(self, nums: List[int]) -> int:
  3. length = len(nums)
  4. positive, negative = [0] * length, [0] * length
  5. if nums[0] > 0:
  6. positive[0] = 1
  7. elif nums[0] < 0:
  8. negative[0] = 1
  9. maxLength = positive[0]
  10. for i in range(1, length):
  11. if nums[i] > 0:
  12. positive[i] = positive[i - 1] + 1
  13. negative[i] = (negative[i - 1] + 1 if negative[i - 1] > 0 else 0)
  14. elif nums[i] < 0:
  15. positive[i] = (negative[i - 1] + 1 if negative[i - 1] > 0 else 0)
  16. negative[i] = positive[i - 1] + 1
  17. else:
  18. positive[i] = negative[i] = 0
  19. maxLength = max(maxLength, positive[i])
  20. return maxLength

该题还可以使用贪心的方法进行处理,按零分组,找第一个负数和最后一个负数的位置 

 5、丑数 II

  1. dp = (n + 1) * [0]
  2. n2, n3, n5 = 1, 1, 1
  3. dp[1] = 1
  4. for i in range(2, n + 1):
  5. dp[i] = min(2 * dp[n2], 3 * dp[n3], 5 * dp[n5])
  6. if dp[i] == 2 * dp[n2]: n2 += 1
  7. if dp[i] == 3 * dp[n3]: n3 += 1
  8. if dp[i] == 5 * dp[n5]: n5 += 1
  9. return dp[n]

6、 优美的排列

  1. f = [0] * (1 << n)
  2. f[0] = 1
  3. for mask in range(1, 1 << n):
  4. num = bin(mask).count("1")
  5. for i in range(n):
  6. if mask & (1 << i) and (num % (i + 1) == 0 or (i + 1) % num == 0):
  7. f[mask] += f[mask ^ (1 << i)]
  8. return f[(1 << n) - 1]

7、只有两个键的键盘 

  1. if n == 1: return 0
  2. dp = [[inf] * (n // 2 + 1) for _ in range(n + 1)]
  3. dp[1][0], dp[1][1] = 0, 1
  4. for i in range(2, n + 1):
  5. for k in range(1, i // 2 + 1):
  6. if i - k >= 0:
  7. dp[i][k] = min(dp[i][k], dp[i - k][k] + 1)
  8. if k == i // 2 and i % 2 == 0:
  9. dp[i][k] = min(dp[i][k], min(dp[k]) + 2)
  10. return min(dp[-1])

由于一个素数的字母的最小操作次数是其本身,所以可以转化为如下动态规划算法:

  1. f = [0] * (n + 1)
  2. for i in range(2, n + 1):
  3. f[i] = float("inf")
  4. j = 1
  5. while j * j <= i:
  6. if i % j == 0:
  7. f[i] = min(f[i], f[j] + i // j)
  8. f[i] = min(f[i], f[i // j] + j)
  9. j += 1
  10. return f[n]

 由于两个素数的和小于其乘积,所以可优化成采用分解质因数的方法

8、新 21 点  (前缀和 + 动态规划)

  1. dp = [1] + [0] * n
  2. for i in range(1, n + 1):
  3. if i <= n - k:
  4. dp[i] = dp[i - 1] + 1
  5. continue
  6. dp[i] = dp[i - 1] - (dp[i - maxPts - 1] if i - maxPts > 0 else 0)
  7. dp[i] = dp[i - 1] + dp[i] / maxPts
  8. return dp[-1] - (dp[-2] if n >= 1 else 0)

9、计算分配糖果的不同方式 (斯特林数)

  1. for i in range(1, k + 1): #盒子
  2. for j in range(i + 1, n + 1): #糖果数
  3. #新的糖果,单独一个盒子
  4. dp[i][j] = dp[i-1][j-1]
  5. #新的糖果,加入其他的盒子
  6. dp[i][j] += dp[i][j-1] * i

10、石子游戏 II(前缀和 + 动态规划)

  1. class Solution:
  2. def stoneGameII(self, piles: List[int]) -> int:
  3. pre = [0]
  4. for p in piles:
  5. pre.append(pre[-1] + p)
  6. n = len(piles)
  7. @cache
  8. def dfs(i, M):
  9. if i + 2 * M >= n:
  10. return pre[-1] - pre[i]
  11. maxl = 0
  12. for j in range(1, 2 * M + 1):
  13. maxl = max(maxl, pre[-1] - pre[i] - dfs(j + i, max(M, j)))
  14. return maxl
  15. return dfs(0, 1)

 动态规划与树结合

1、打家劫舍 III

  1. class Solution:
  2. def rob(self, root: TreeNode) -> int:
  3. def dfs(p):
  4. if not p: return 0, 0
  5. la, lb = dfs(p.left)
  6. ra, rb = dfs(p.right)
  7. return max(lb + rb + p.val, ra + la), ra + la
  8. return max(dfs(root))

2、二叉树中的最长交错路径

  1. class Solution:
  2. def longestZigZag(self, root: TreeNode) -> int:
  3. maxl = 0
  4. def dfs(p):
  5. nonlocal maxl
  6. if not p: return -1, -1
  7. l1, r1 = dfs(p.left)
  8. l2, r2 = dfs(p.right)
  9. maxl = max(maxl, r1 + 1, l2 + 1)
  10. return r1 + 1, l2 + 1
  11. dfs(root)
  12. return maxl

3、二叉搜索子树的最大键值和

  1. class Solution:
  2. def maxSumBST(self, root: Optional[TreeNode]) -> int:
  3. maxl = 0
  4. def dfs(p):
  5. min1, max1, s1 = dfs(p.left) if p.left else (inf, -inf, 0)
  6. min2, max2, s2 = dfs(p.right) if p.right else (inf, -inf, 0)
  7. if max1 < p.val < min2:
  8. nonlocal maxl
  9. maxl = max(maxl, s1 + s2 + p.val)
  10. return min(min1, p.val), max(max2, p.val), s1 + s2 + p.val
  11. return -inf, inf, 0
  12. dfs(root)
  13. return maxl

4、 统计可能的树根数目 (换根dp)

(1)先从任意一个根出发,求解所有点

(2)再换根,计算变化的点

  1. class Solution:
  2. def rootCount(self, edges: List[List[int]], guesses: List[List[int]], k: int) -> int:
  3. m = defaultdict(list)
  4. for f, t in edges:
  5. m[f].append(t)
  6. m[t].append(f)
  7. s = defaultdict(int)
  8. for f, t in guesses:
  9. s[f, t] += 1
  10. t = 0
  11. def dfs(i, last):
  12. nonlocal t
  13. for j in m[i]:
  14. if j == last: continue
  15. t += s[i, j]
  16. dfs(j, i)
  17. dfs(0, -1)
  18. res = 0
  19. def dfs2(i, last, t):
  20. nonlocal res
  21. if t >= k: res += 1
  22. for j in m[i]:
  23. if j == last: continue
  24. dfs2(j, i, t - s[i, j] + s[j, i])
  25. dfs2(0, -1, t)
  26. return res

5、 树中距离之和  (换根dp)

  1. class Solution:
  2. def sumOfDistancesInTree(self, n: int, edges: List[List[int]]) -> List[int]:
  3. m = defaultdict(list)
  4. for f, t in edges:
  5. m[f].append(t)
  6. m[t].append(f)
  7. dises, sizes = [0] * n, [0] * n
  8. #节点到所有孩子节点的距离,以及节点与所有孩子节点的个数
  9. def dfs(i, last):
  10. dis, size = 0, 1
  11. for j in m[i]:
  12. if j == last: continue
  13. child_dis, child_size = dfs(j, i)
  14. dis += child_dis + child_size
  15. size += child_size
  16. dises[i], sizes[i] = dis, size
  17. return dis, size
  18. dfs(0, -1)
  19. res = [0] * n
  20. res[0] = dises[0]
  21. def dfs2(i, last, dis, size):
  22. sum_child_size, sum_child_dis = 0, 0
  23. for j in m[i]:
  24. if j == last: continue
  25. sum_child_size += sizes[j]
  26. sum_child_dis += dises[j]
  27. for j in m[i]:
  28. if j == last: continue
  29. parent_dis = dis + 2 * (sum_child_size - sizes[j]) + sum_child_dis - dises[j] + size
  30. parent_size = size + sum_child_size - sizes[j] + 1
  31. res[j] = parent_dis + dises[j]
  32. dfs2(j, i, parent_dis, parent_size)
  33. dfs2(0, -1, 0, 1)
  34. return res

数位动态规划

1、至少有 1 位重复的数字

  1. class Solution:
  2. def numDupDigitsAtMostN(self, n: int) -> int:
  3. s = str(n)
  4. @cache
  5. def f(i: int, mask: int, is_limit: bool, is_num: bool) -> int:
  6. if i == len(s):
  7. return int(is_num)
  8. res = 0
  9. if not is_num: # 可以跳过当前数位
  10. res = f(i + 1, mask, False, False)
  11. up = int(s[i]) if is_limit else 9
  12. for d in range(0 if is_num else 1, up + 1): # 枚举要填入的数字 d
  13. if mask >> d & 1 == 0: # d 不在 mask 中
  14. res += f(i + 1, mask | (1 << d), is_limit and d == up, True)
  15. return res
  16. return n - f(0, 0, True, False)

2、不含连续1的非负整数

  1. class Solution:
  2. def findIntegers(self, n: int) -> int:
  3. s = str(bin(n))[2:]
  4. @cache
  5. def f(i: int, pre1: bool, is_limit: bool) -> int:
  6. if i == len(s):
  7. return 1
  8. up = int(s[i]) if is_limit else 1
  9. res = f(i + 1, False, is_limit and up == 0) # 填 0
  10. if not pre1 and up == 1: # 可以填 1
  11. res += f(i + 1, True, is_limit) # 填 1
  12. return res
  13. return f(0, False, True)

倍增

1、Pow(x, n)  -- 快速幂算法

  1. class Solution:
  2. def myPow(self, x: float, n: int) -> float:
  3. if n == 0: return 1
  4. n, f = abs(n), (n < 0)
  5. p = [x] * n.bit_length()
  6. for i in range(n.bit_length() - 1):
  7. p[i + 1] = p[i] * p[i]
  8. res = 1
  9. for i in range(n.bit_length()):
  10. if (n >> i) & 1:
  11. res *= p[i]
  12. if f: return 1 / res
  13. return res

2、在传球游戏中最大化函数值  -- (求解第k个祖先)

  1. class Solution:
  2. def getMaxFunctionValue(self, receiver: List[int], k: int) -> int:
  3. n = len(receiver)
  4. m = k.bit_length() - 1
  5. pa = [[(p, p)] + [None] * m for p in receiver]
  6. for i in range(m):
  7. for x in range(n):
  8. p, s = pa[x][i]
  9. pp, ss = pa[p][i]
  10. pa[x][i + 1] = (pp, s + ss) # 合并节点值之和
  11. ans = 0
  12. for i in range(n):
  13. x = sum = i
  14. for j in range(m + 1):
  15. if (k >> j) & 1: # k 的二进制从低到高第 j 位是 1
  16. x, s = pa[x][j]
  17. sum += s
  18. ans = max(ans, sum)
  19. return ans

 状态压缩与优化

1、取余法:(70. 爬楼梯

  1. dp = [1, 2]
  2. for i in range(2, n):
  3. dp[i % 2] = dp [(i -1) % 2] + dp[(i - 2) % 2]
  4. return dp[(n - 1) % 2]

2、常数级别压缩: (70. 爬楼梯

  1. a, b = 0, 1
  2. for i in range(n):
  3. a, b = b, a + b
  4. return b

3、二维压缩成一维:(72. 编辑距离

  1. s, t = (word1, word2) if len(word1) > len(word2) else (word2, word1)
  2. dp = [j for j in range(len(t) + 1)]
  3. for i in range(1, len(s)+1):
  4. dpij, dp[0] = dp[0], i
  5. for j in range(1, len(t) + 1):
  6. temp = dp[j]
  7. if s[i - 1] == t[j - 1]:
  8. dp[j] = dpij
  9. else:
  10. dp[j] = min(dp[j] + 1, dp[j - 1] + 1, dpij + 1)
  11. dpij = temp
  12. return dp[-1]

4、使用位运算进行压缩:(2510. Check if There is a Path With Equal Number of 0's And 1's) 

常规方法,使用集合存储结果,运行时间为376ms,内存消耗146.8mb,代码如下:

  1. class Solution:
  2. def isThereAPath(self, grid: List[List[int]]) -> bool:
  3. m, n = len(grid), len(grid[0])
  4. @cache
  5. def dfs(i, j):
  6. if i <= 0 and j < 0 or i < 0 and j <= 0: return {0}
  7. if i < 0 or j < 0: return set()
  8. res = set()
  9. for e in dfs(i - 1, j) | dfs(i, j - 1):
  10. if grid[i][j]:
  11. res.add(e + 1)
  12. else:
  13. res.add(e - 1)
  14. return res
  15. return 0 in dfs(m - 1, n - 1)

如果使用位运算存储结果,运行时间为60ms,内存消耗22.2mb,代码如下:

  1. class Solution:
  2. def isThereAPath(self, grid: List[List[int]]) -> bool:
  3. m, n = len(grid), len(grid[0])
  4. @cache
  5. def dfs(i, j):
  6. if i <= 0 and j < 0 or i < 0 and j <= 0: return 1 << (m + n)
  7. if i < 0 or j < 0: return 0
  8. if grid[i][j]:
  9. return (dfs(i - 1, j) << 1) | (dfs(i, j - 1) << 1)
  10. else:
  11. return (dfs(i - 1, j) >> 1) | (dfs(i, j - 1) >> 1)
  12. return ((dfs(m - 1, n - 1) >> (m + n)) & 1) == 1

 进阶方法:具有哈希功能位运算压缩 -- 得到新鲜甜甜圈的最多组数

常规方法,使用最多10维状态空间,代码如下,运行时间为4662ms,内存占用236.5mb

  1. class Solution:
  2. def maxHappyGroups(self, batchSize: int, groups: List[int]) -> int:
  3. mp = defaultdict(int)
  4. for g in groups:
  5. mp[g % batchSize] += 1
  6. @cache
  7. def dfs(s, *b):
  8. if sum(b) == len(groups):
  9. return 0
  10. res, t = 0, (1 if s % batchSize == 0 else 0)
  11. for j in range(len(b)):
  12. if b[j] >= mp[j]: continue
  13. bb = list(b)
  14. bb[j] += 1
  15. res = max(res, dfs((s + j) % batchSize, *bb) + t)
  16. return res
  17. return dfs(0, *([0] * batchSize))

使用具有哈希功能的位运算压缩,每个数字分配5bit,值最高为31,代码如下,运行时间为1665ms,占用内存43.8mb

  1. class Solution:
  2. def maxHappyGroups(self, batchSize: int, groups: List[int]) -> int:
  3. kWidth = 5
  4. kWidthMask = (1 << kWidth) - 1
  5. cnt = Counter(x % batchSize for x in groups)
  6. start = 0
  7. for i in range(batchSize - 1, 0, -1):
  8. start = (start << kWidth) | cnt[i]
  9. @cache
  10. def dfs(mask: int) -> int:
  11. if mask == 0:
  12. return 0
  13. total = 0
  14. for i in range(1, batchSize):
  15. amount = ((mask >> ((i - 1) * kWidth)) & kWidthMask)
  16. total += i * amount
  17. best = 0
  18. for i in range(1, batchSize):
  19. amount = ((mask >> ((i - 1) * kWidth)) & kWidthMask)
  20. if amount > 0:
  21. result = dfs(mask - (1 << ((i - 1) * kWidth)))
  22. if (total - i) % batchSize == 0:
  23. result += 1
  24. best = max(best, result)
  25. return best
  26. ans = dfs(start) + cnt[0]
  27. dfs.cache_clear()
  28. return ans

01背包问题bitset优化 -- 执行操作可获得的最大总奖励 II 

代码如下:

  1. class Solution:
  2. def maxTotalReward(self, rewardValues: List[int]) -> int:
  3. f = 1
  4. for v in sorted(set(rewardValues)):
  5. f |= (f & ((1 << v) - 1)) << v
  6. return f.bit_length() - 1

5、使用26个字母空间

(1)2370. 最长理想子序列

  1. f = [0] * 26
  2. for c in s:
  3. c = ord(c) - ord('a')
  4. f[c] = 1 + max(f[max(c - k, 0): c + k + 1])
  5. return max(f)

(2)不同的子序列 II

 对于相同的子序列,只会考虑其最后一次出现的位置(下标序列的字典序最大)

  1. class Solution:
  2. def distinctSubseqII(self, s: str) -> int:
  3. mod = 10 ** 9 + 7
  4. dp = [0] * 26
  5. for c in s:
  6. i = ord(c) - ord('a')
  7. for j in range(26):
  8. if i == j: continue
  9. dp[i] = (dp[j] + dp[i]) % mod
  10. dp[i] = (dp[i] + 1) % mod
  11. return sum(dp) % mod

 6、使用线段树 (2407. 最长递增子序列 II

  1. class Solution:
  2. def lengthOfLIS(self, nums: List[int], k: int) -> int:
  3. tmp = SegmentTree([0] * (max(nums) + 1))
  4. for x in nums:
  5. note = tmp.query(max(0, x-k), x-1)
  6. tmp.update(x, note + 1)
  7. return tmp.query(0, max(nums))
  8. class SegmentTree:
  9. def __init__(self, data, merge=max):
  10. self.data = data
  11. self.n = len(data)
  12. self.tree = [None] * (4 * self.n)
  13. self._merge = merge
  14. if self.n:
  15. self._build(0, 0, self.n-1)
  16. def query(self, ql, qr):
  17. return self._query(0, 0, self.n-1, ql, qr)
  18. def update(self, index, value):
  19. self.data[index] = value
  20. self._update(0, 0, self.n-1, index)
  21. def _build(self, tree_index, l, r):
  22. if l == r:
  23. self.tree[tree_index] = self.data[l]
  24. return
  25. mid = (l+r) // 2
  26. left, right = 2 * tree_index + 1, 2 * tree_index + 2
  27. self._build(left, l, mid)
  28. self._build(right, mid+1, r)
  29. self.tree[tree_index] = self._merge(self.tree[left], self.tree[right])
  30. def _query(self, tree_index, l, r, ql, qr):
  31. if l == ql and r == qr:
  32. return self.tree[tree_index]
  33. mid = (l+r) // 2
  34. left, right = tree_index * 2 + 1, tree_index * 2 + 2
  35. if qr <= mid:
  36. return self._query(left, l, mid, ql, qr)
  37. elif ql > mid:
  38. return self._query(right, mid+1, r, ql, qr)
  39. return self._merge(self._query(left, l, mid, ql, mid),
  40. self._query(right, mid+1, r, mid+1, qr))
  41. def _update(self, tree_index, l, r, index):
  42. if l == r == index:
  43. self.tree[tree_index] = self.data[index]
  44. return
  45. mid = (l+r)//2
  46. left, right = 2 * tree_index + 1, 2 * tree_index + 2
  47. if index > mid:
  48. self._update(right, mid+1, r, index)
  49. else:
  50. self._update(left, l, mid, index)
  51. self.tree[tree_index] = self._merge(self.tree[left], self.tree[right])

进阶:利用线段树的分治思想改造递推关系 -- 不包含相邻元素的子序列的最大和

代码如下:

  1. class Solution:
  2. def maximumSumSubsequence(self, nums: List[int], queries: List[List[int]]) -> int:
  3. n = len(nums)
  4. # 4 个数分别保存 f00, f01, f10, f11
  5. t = [[0] * 4 for _ in range(2 << n.bit_length())]
  6. def maintain(o: int):
  7. a, b = t[o * 2], t[o * 2 + 1]
  8. t[o][0] = max(a[0] + b[2], a[1] + b[0])
  9. t[o][1] = max(a[0] + b[3], a[1] + b[1])
  10. t[o][2] = max(a[2] + b[2], a[3] + b[0])
  11. t[o][3] = max(a[2] + b[3], a[3] + b[1])
  12. # 用 nums 初始化线段树
  13. def build(o: int, l: int, r: int) -> None:
  14. if l == r:
  15. t[o][3] = max(nums[l], 0)
  16. return
  17. m = (l + r) // 2
  18. build(o * 2, l, m)
  19. build(o * 2 + 1, m + 1, r)
  20. maintain(o)
  21. # 把 nums[i] 改成 val
  22. def update(o: int, l: int, r: int, i: int, val: int) -> None:
  23. if l == r:
  24. t[o][3] = max(val, 0)
  25. return
  26. m = (l + r) // 2
  27. if i <= m:
  28. update(o * 2, l, m, i, val)
  29. else:
  30. update(o * 2 + 1, m + 1, r, i, val)
  31. maintain(o)
  32. build(1, 0, n - 1)
  33. ans = 0
  34. for i, x in queries:
  35. update(1, 0, n - 1, i, x)
  36. ans += t[1][3] # 注意 f11 没有任何限制,也就是整个数组的打家劫舍
  37. return ans % 1_000_000_007

7、提高状态空间的重用性

(1)2143.在两个数组的区间中选取数字中,如下方法设置状态空间运行结果为超时

  1. class Solution:
  2. def countSubranges(self, nums1: List[int], nums2: List[int]) -> int:
  3. n = len(nums1)
  4. @cache
  5. def dfs(i, l, is_start, r, is_end):
  6. if i == n:
  7. return int(l == r) if is_start else 0
  8. res = 0
  9. if not is_start:
  10. res += dfs(i + 1, l, False, r, False)
  11. res += dfs(i + 1, l + nums1[i], True, r , False)
  12. res += dfs(i + 1, l, True, r + nums2[i], False)
  13. elif is_start and not is_end:
  14. res += dfs(i + 1, l + nums1[i], True, r , False)
  15. res += dfs(i + 1, l, True, r + nums2[i], False)
  16. res += dfs(i + 1, l, True, r, True)
  17. else:
  18. res += dfs(i + 1, l, True, r, True)
  19. return res
  20. return dfs(0, 0, False, 0, False)

       这是因为(l, r) == (1, 1) 与 (l, r) == (6, 6)的状态是等价的,但是状态空间没有复用,导致重复计算,所以可以按如下方法改正,状态空间优化成一个,结果通过:

  1. class Solution:
  2. def countSubranges(self, nums1: List[int], nums2: List[int]) -> int:
  3. n = len(nums1)
  4. @cache
  5. def dfs(i, d, is_start, is_end):
  6. if i == n: return int(d == 0) if is_start else 0
  7. if is_end: return int(d == 0)
  8. res = 0
  9. if not is_start:
  10. res += dfs(i + 1, d, False, False)
  11. res += dfs(i + 1, d + nums1[i], True, False)
  12. res += dfs(i + 1, d - nums2[i], True, False)
  13. elif is_start and not is_end:
  14. res += dfs(i + 1, d + nums1[i], True, False)
  15. res += dfs(i + 1, d - nums2[i], True, False)
  16. res += dfs(i + 1, d, True, True)
  17. else:
  18. res += dfs(i + 1, d, True, True)
  19. return res
  20. return dfs(0, 0, False, False) % (10 ** 9 + 7)

类似的还有 956. 最高的广告牌,题解可参考:. - 力扣(LeetCode) 

(2)在盈利计划中,如下方法超时:

  1. class Solution:
  2. def profitableSchemes(self, n: int, minProfit: int, group: List[int], profit: List[int]) -> int:
  3. mod = 10 ** 9 + 7
  4. @cache
  5. def dfs(i, n, p):
  6. if i == len(group):
  7. return p >= minProfit
  8. res = 0
  9. if group[i] <= n:
  10. res = (res + dfs(i + 1, n - group[i], p + profit[i])) % mod
  11. return (res + dfs(i + 1, n, p)) % mod
  12. return dfs(0, n, 0) % mod

 这是由于当前盈利值达到最低要求后,后续结果只与剩余人数有关,所以按照最小盈利值对状态进行重用,如下所示。

  1. class Solution:
  2. def profitableSchemes(self, n: int, minProfit: int, group: List[int], profit: List[int]) -> int:
  3. mod = 10 ** 9 + 7
  4. @cache
  5. def dfs(i, n, p):
  6. if i == len(group):
  7. return int(p == minProfit)
  8. res = 0
  9. if group[i] <= n:
  10. res = (res + dfs(i + 1, n - group[i], min(minProfit, p + profit[i]))) % mod
  11. return (res + dfs(i + 1, n, p)) % mod
  12. return dfs(0, n, 0) % mod

 8、缩短计算步数 

(1) Minimum Time to Kill All Monsters

        如下解法运行超时,因为当怪物需要的能量比较大时,计算次数变多。

  1. class Solution:
  2. def minimumTime(self, power: List[int]) -> int:
  3. n, mp = len(power), max(power)
  4. @cache
  5. def dfs(g, v, inc):
  6. if v == (1 << n) - 1: return 0
  7. res, gg, step = inf, g - inc, 0
  8. while gg <= mp:
  9. gg += inc
  10. step += 1
  11. for i in range(n):
  12. if (v >> i) & 1 == 0 and gg >= power[i]:
  13. res = min(res, dfs(inc + 1, v | (1 << i), inc + 1) + step)
  14. return res
  15. return dfs(1, 0, 1)

          根据题意,结果只与打败怪物的顺序有关,与积攒能量的过程无关,所以可以优化成下面的方法,减少重复计算,同时优化状态空间。

  1. class Solution:
  2. def minimumTime(self, power: List[int]) -> int:
  3. n, mp = len(power), max(power)
  4. @cache
  5. def dfs(v, inc):
  6. if v == (1 << n) - 1: return 0
  7. res = inf
  8. for i in range(n):
  9. if (v >> i) & 1 == 0:
  10. res = min(res, dfs(v | (1 << i), inc + 1) + ceil(power[i] / inc))
  11. return res
  12. return dfs(0, 1)

(2)好二进制字符串的数量

根据题意设计如下代码,会超时:

  1. class Solution:
  2. def goodBinaryStrings(self, minLength: int, maxLength: int, oneGroup: int, zeroGroup: int) -> int:
  3. @cache
  4. def dfs(i, p):
  5. res = 0
  6. if minLength <= i <= maxLength and p == 0:
  7. res = 1
  8. if i == maxLength: return res
  9. if p <= 0:
  10. res += dfs(i + 1, -((-p + 1) % zeroGroup))
  11. if p >= 0:
  12. res += dfs(i + 1, (p + 1) % oneGroup)
  13. return res % (10 ** 9 + 7)
  14. return dfs(0, 0) % (10 ** 9 + 7)

        由于要求具有连续相同字符,所以不需要计算使用每个字符的状态,如下所示,通过减少状态空间和缩短步数来达到优化的目的。

  1. class Solution:
  2. def goodBinaryStrings(self, minLength: int, maxLength: int, oneGroup: int, zeroGroup: int) -> int:
  3. @cache
  4. def dfs(i):
  5. res = 0
  6. if minLength <= i <= maxLength:
  7. res = 1
  8. if i == maxLength: return res
  9. if i + oneGroup <= maxLength:
  10. res += dfs(i + oneGroup)
  11. if i + zeroGroup <= maxLength:
  12. res += dfs(i + zeroGroup)
  13. return res % (10 ** 9 + 7)
  14. return dfs(0) % (10 ** 9 + 7)

(3) 吃掉 N 个橘子的最少天数

       直接使用暴力dp超时,因为n最大超过10^9, 每次减一需要的时间比较长,所以可以对减一部分进行优化,由于n 变为 二分之一或者三分之一步数总是比连续减一的步数小,所以可以写成如下代码:

  1. class Solution:
  2. @lru_cache(None)
  3. def minDays(self, n: int) -> int:
  4. if n == 0: return 0
  5. if n == 1: return 1
  6. return 1 + min(self.minDays(n//2) + n % 2, self.minDays(n//3) + n % 3)

9、剪枝法

(1) 出租车的最大盈利

  如下使用动态规划的方法,会超时

  1. class Solution:
  2. def maxTaxiEarnings(self, n: int, rides: List[List[int]]) -> int:
  3. rides.sort()
  4. @cache
  5. def dfs(j):
  6. idx = bisect_left(rides, [j, 0, 0])
  7. if idx >= len(rides): return 0
  8. maxl = 0
  9. for i in range(idx, len(rides)):
  10. start, end, tip = rides[i]
  11. maxl = max(maxl, dfs(end) + end - start + tip)
  12. return maxl
  13. return dfs(0)

       这是因为当两个区间不相交时,两个区间都取到时收益最大,所以在新的起始点大于最小终止点时,结束计算,如下所示:

  1. class Solution:
  2. def maxTaxiEarnings(self, n: int, rides: List[List[int]]) -> int:
  3. rides.sort()
  4. @cache
  5. def dfs(j):
  6. idx = bisect_left(rides, [j, 0, 0])
  7. if idx >= len(rides): return 0
  8. maxl = 0
  9. s0, e0, t0 = rides[idx]
  10. for i in range(idx, len(rides)):
  11. start, end, tip = rides[i]
  12. e0 = min(e0, end)
  13. if start >= e0: break
  14. maxl = max(maxl, dfs(end) + end - start + tip)
  15. return maxl
  16. return dfs(0)

 也可以利用区间结束点的顺序和二分查找法对其进行优化:

  1. class Solution:
  2. def maxTaxiEarnings(self, n: int, rides: List[List[int]]) -> int:
  3. rides = sorted(rides, key=lambda x: x[1])
  4. dp = [[0, 0]]
  5. for s, e, p in rides:
  6. i = bisect.bisect(dp, [s + 1]) - 1
  7. if dp[i][1] + p + e - s > dp[-1][1]:
  8. dp.append([e, dp[i][1] + p + e - s])
  9. return dp[-1][1]

(2)石子游戏 V

常规的dp导致超时,根据等比数列求和公式可知,能预估后续得分之和,当小于当前最大值时可以剪枝,代码如下:

  1. class Solution:
  2. def stoneGameV(self, stoneValue: List[int]) -> int:
  3. pre = [0]
  4. for v in stoneValue:
  5. pre.append(pre[-1] + v)
  6. @cache
  7. def dfs(i, j):
  8. if i == j: return 0
  9. maxl = 0
  10. for k in range(i, j):
  11. left = pre[k + 1] - pre[i]
  12. right = pre[j + 1] - pre[k + 1]
  13. if left < right:
  14. if maxl >= 2 * left: continue
  15. maxl = max(maxl, dfs(i, k) + left)
  16. elif right < left:
  17. if maxl >= 2 * right: continue
  18. maxl = max(maxl, dfs(k + 1, j) + right)
  19. else:
  20. if maxl >= 2 * right: continue
  21. maxl = max(maxl, dfs(i, k) + left, dfs(k + 1, j) + right)
  22. return maxl
  23. return dfs(0, len(stoneValue) - 1)

(3)收集所有金币可获得的最大积分

由于每个节点的金币数量最高为10^4,所以当除数大于2^13时金币数为0,可以不用继续往下进行。

  1. class Solution:
  2. def maximumPoints(self, edges: List[List[int]], coins: List[int], k: int) -> int:
  3. m = defaultdict(set)
  4. for f, t in edges:
  5. m[f].add(t)
  6. m[t].add(f)
  7. def dfs(p, last):
  8. for np in list(m[p]):
  9. if np == last:
  10. m[p].remove(np)
  11. continue
  12. dfs(np, p)
  13. dfs(0, -1)
  14. @cache
  15. def dp(p, mm):
  16. r1 = coins[p] // mm - k
  17. r2 = coins[p] // mm // 2
  18. for np in m[p]:
  19. if r1 < r2 and mm < 2 ** 13:
  20. r2 += dp(np, 2 * mm)
  21. r1 += dp(np, mm)
  22. return max(r1, r2)
  23. return dp(0, 1)

(4) 相似度为 K 的字符串

构造环+位掩码的算法,代码如下,运行时间为1381ms:

  1. class Solution:
  2. def kSimilarity(self, s1: str, s2: str) -> int:
  3. ns1, ns2 = '', ''
  4. for i, c in enumerate(s1):
  5. if s1[i] == s2[i]: continue
  6. ns1 += s1[i]
  7. ns2 += s2[i]
  8. s1, s2 = ns1, ns2
  9. m = defaultdict(list)
  10. for i, c in enumerate(s1):
  11. m[c].append(i)
  12. n = len(s1)
  13. @cache
  14. def dfs(mask, head=-1, last=-1):
  15. if mask.bit_count() == n:
  16. return 0
  17. res = []
  18. if head == -1:
  19. for i in range(n):
  20. if (mask >> i) & 1 == 1: continue
  21. res.append(dfs(mask | (1 << i), s1[i], s2[i]))
  22. else:
  23. for j in m[last]:
  24. if (mask >> j) & 1 == 1: continue
  25. if s2[j] == head:
  26. res.append(dfs(mask | (1 << j), -1, -1) - 1)
  27. else:
  28. res.append(dfs(mask | (1 << j), head, s2[j]))
  29. return min(res)
  30. return n + dfs(0, -1, -1)

环的每一点都是等价的,所以从未遍历一点进入即可。同时切分的环越多越好,当到达能够形成环的位置时终止,代码如下,运行时间为37ms

  1. class Solution:
  2. def kSimilarity(self, s1: str, s2: str) -> int:
  3. m, t = defaultdict(list), 0
  4. mask = 0
  5. for i, c in enumerate(s1):
  6. if s1[i] == s2[i]:
  7. mask = mask | (1 << i)
  8. continue
  9. m[c].append(i)
  10. t += 1
  11. n = len(s1)
  12. @cache
  13. def dfs(mask, head=-1, last=-1):
  14. if mask.bit_count() == n:
  15. return 0
  16. res = []
  17. if head == -1:
  18. for i in range(n):
  19. if (mask >> i) & 1 == 1: continue
  20. if s1[i] != s2[i]:
  21. return dfs(mask | (1 << i), s1[i], s2[i])
  22. else:
  23. for j in m[last]:
  24. if (mask >> j) & 1 == 1: continue
  25. if s2[j] == head:
  26. return dfs(mask | (1 << j), -1, -1) - 1
  27. else:
  28. res.append(dfs(mask | (1 << j), head, s2[j]))
  29. return min(res)
  30. return t + dfs(mask, -1, -1)

10、逆向思维求解  -- 好分区的数目

 如果直接求解好分区数量,但是背包容量太大,会超时,代码如下:

  1. class Solution:
  2. def countPartitions(self, nums: List[int], k: int) -> int:
  3. sm = sum(nums)
  4. @cache
  5. def dfs(i, s):
  6. if i == len(nums):
  7. return 1 if sm - s >= k and s >= k else 0
  8. return dfs(i + 1, s + nums[i]) + dfs(i + 1, s)
  9. return dfs(0, 0) % (10 ** 9 + 7)

可以通过求不好分区数量,再用总数相减来提高求解效率,即第一个组或第二个组的元素和小于 k的方案数。根据对称性,我们只需要计算第一个组的元素和小于 k 的方案数,然后乘 2即可:

  1. class Solution:
  2. def countPartitions(self, nums: List[int], k: int) -> int:
  3. sm = sum(nums)
  4. if sm < k * 2: return 0
  5. mod = (10 ** 9 + 7)
  6. @cache
  7. def dfs(i, s):
  8. if i == len(nums): return 1
  9. res = dfs(i + 1, s)
  10. if s >= nums[i]:
  11. res += dfs(i + 1, s - nums[i])
  12. return res % mod
  13. return (pow(2, len(nums), mod) - dfs(0, k - 1) * 2) % mod

再转化成01背包的形式,进一步提高效率:

  1. class Solution:
  2. def countPartitions(self, nums: List[int], k: int) -> int:
  3. if sum(nums) < k * 2: return 0
  4. MOD = 10 ** 9 + 7
  5. f = [0] * k
  6. f[0] = 1
  7. for x in nums:
  8. for j in range(k - 1, x - 1, -1):
  9. f[j] = (f[j] + f[j - x]) % MOD
  10. return (pow(2, len(nums), MOD) - sum(f) * 2) % MOD

11、问题转化

(1) K 次调整数组大小浪费的最小总空间 

直接使用动态规划进行求解,会超时:

  1. class Solution:
  2. def minSpaceWastedKResizing(self, nums: List[int], k: int) -> int:
  3. n = len(nums)
  4. @cache
  5. def dfs(i, j, k):
  6. if i == len(nums): return 0
  7. if nums[i] <= nums[j]:
  8. minl = dfs(i + 1, j, k) + nums[j] - nums[i]
  9. else: minl = inf
  10. if k > 0:
  11. for jj in range(j + 1, n):
  12. if nums[jj] >= nums[i]:
  13. minl = min(minl, dfs(i + 1, jj, k - 1) + nums[jj] - nums[i])
  14. return minl
  15. return min(dfs(0, j, k) for j in range(n))

 可以理解成将数组分成k + 1组,求每组最大值乘以这一段的长度再减去这一段的元素和,找到总体最小值,代码如下:

  1. class Solution:
  2. def minSpaceWastedKResizing(self, nums: List[int], k: int) -> int:
  3. n = len(nums)
  4. pre = [0]
  5. for num in nums:
  6. pre.append(pre[-1] + num)
  7. @cache
  8. def diff(i, j):
  9. return (j - i + 1) * max(nums[i: j + 1]) - pre[j + 1] + pre[i]
  10. @cache
  11. def dfs(i, j, k):
  12. minl = diff(i, j)
  13. if k > 0:
  14. for t in range(i, j):
  15. minl = min(minl, dfs(i, t, k - 1) + diff(t + 1, j))
  16. return minl
  17. return dfs(0, n - 1, k)

 (2)准时抵达会议现场的最小跳过休息次数

直接根据题意,使用距离为空间维度的动态规划超出内容,代码如下:

  1. class Solution:
  2. def minSkips(self, dist: List[int], speed: int, hoursBefore: int) -> int:
  3. n = len(dist)
  4. @cache
  5. def dfs(i, d):
  6. if d > speed * hoursBefore: return inf
  7. if i == n: return 0
  8. if (d + dist[i]) % speed == 0 or i == n - 1:
  9. return dfs(i + 1, d + dist[i])
  10. res = dfs(i + 1, d + dist[i]) + 1
  11. res = min(res, dfs(i + 1, (d + dist[i]) // speed * speed + speed ))
  12. return res
  13. return dfs(0, 0) if dfs(0, 0) < inf else -1

可以转化成求最多跳转次数(j)的最短时间,找出时间符合要求的最小j,代码如下:

  1. class Solution:
  2. def minSkips(self, dist: List[int], speed: int, hoursBefore: int) -> int:
  3. @cache
  4. def dfs(i, j):
  5. if i == 0: return 0
  6. d = dfs(i - 1, j) + dist[i - 1]
  7. if d % speed == 0 or i == len(dist): return d
  8. res = d + speed - d % speed
  9. if j > 0:
  10. res = min(res, dfs(i - 1, j - 1) + dist[i - 1])
  11. return res
  12. for j in range(len(dist)):
  13. if dfs(len(dist), j) <= speed * hoursBefore: return j
  14. return -1

12、降低参数维度

(1)减少不必要的参数 -- 两个子序列的最大点积

如下所示,如果用f来判别是否为空数组,运行时间为1148ms

  1. class Solution:
  2. def maxDotProduct(self, nums1: List[int], nums2: List[int]) -> int:
  3. @cache
  4. def dfs(i, j, f):
  5. if i == len(nums1) or j == len(nums2): return 0 if f else -inf
  6. return max(dfs(i + 1, j, f), dfs(i, j + 1, f), dfs(i + 1, j + 1, True) + nums1[i] * nums2[j])
  7. return dfs(0, 0, False)

 可以改变max的判断来减少维度,运行时间为628ms

  1. class Solution:
  2. def maxDotProduct(self, nums1: List[int], nums2: List[int]) -> int:
  3. @cache
  4. def dfs(i, j):
  5. if i == len(nums1) or j == len(nums2): return -inf
  6. return max(dfs(i + 1, j), dfs(i, j + 1), dfs(i + 1, j + 1) + nums1[i] * nums2[j], nums1[i] * nums2[j])
  7. return dfs(0, 0)

(2)将参数移至返回值   -- 完成任务的最少工作时间段 

当任务已使用时间在函数参数时,运行时间为7092ms

  1. class Solution:
  2. def minSessions(self, tasks: List[int], sessionTime: int) -> int:
  3. n = len(tasks)
  4. @cache
  5. def dfs(v, s):
  6. if v.bit_count() == n:
  7. return 0
  8. minl = inf
  9. for i in range(n):
  10. if (v >> i) & 1 == 1: continue
  11. if tasks[i] <= s:
  12. minl = min(minl, dfs(v | (1 << i), s - tasks[i]))
  13. else:
  14. minl = min(minl, dfs(v | (1 << i), sessionTime - tasks[i]) + 1)
  15. return minl
  16. return dfs(0, sessionTime) + 1

          当任务已使用时间放到返回值时,运行时间为1196ms

  1. class Solution:
  2. def minSessions(self, tasks: List[int], sessionTime: int) -> int:
  3. n = len(tasks)
  4. @cache
  5. def dfs(v):
  6. if v.bit_count() == n:
  7. return 1, 0
  8. minl, mins = inf, inf
  9. for i in range(n):
  10. if (v >> i) & 1 == 1: continue
  11. l, s = dfs(v | (1 << i))
  12. if tasks[i] + s <= sessionTime:
  13. minl, mins = min((minl, mins), (l, s + tasks[i]))
  14. else:
  15. minl, mins = min((minl, mins), (l + 1, tasks[i]))
  16. return minl, mins
  17. return dfs(0)[0]

(3) 执行操作使两个字符串相等

 根据题意,使用动态规划进行求解,代码如下,用时368ms

  1. class Solution:
  2. def minOperations(self, s1: str, s2: str, x: int) -> int:
  3. if abs(s1.count('1') - s2.count('1')) & 1 == 1:
  4. return -1
  5. @cache
  6. def dfs(i, t, is_prev):
  7. if i == len(s1):
  8. return inf if t > 0 or is_prev else 0
  9. if (s1[i] == s2[i]) == (not is_prev):
  10. return dfs(i + 1, t, False)
  11. minl = min(dfs(i + 1, t + 1, False) + x, dfs(i + 1, t, True) + 1)
  12. if t > 0:
  13. minl = min(minl, dfs(i + 1, t - 1, False))
  14. return minl
  15. return dfs(0, 0, False)

 可以证明不同字符之间只能有一次操作,所以可以在不同字符下标处进行转移,缩短计算步数。

代码如下,运行时间为43ms:

  1. class Solution:
  2. def minOperations(self, s1: str, s2: str, x: int) -> int:
  3. if s1 == s2:
  4. return 0
  5. p = [i for i, (x, y) in enumerate(zip(s1, s2)) if x != y]
  6. if len(p) % 2:
  7. return -1
  8. f0, f1 = 0, x
  9. for i, j in pairwise(p):
  10. f0, f1 = f1, min(f1 + x, f0 + (j - i) * 2)
  11. return f1 // 2

(4)从底到上递推,巧用返回值 -- 可处理的最大删除操作数 I 

          该题直接使用三维空间遍历超出内存,使用自底向上的递推方法能根据返回值得知查询顺序,从而节省一维空间,代码如下:

  1. class Solution:
  2. def maximumProcessableQueries(self, nums: List[int], queries: List[int]) -> int:
  3. n = len(nums)
  4. dp = [[0] * (n + 1) for _ in range(n + 1)]
  5. for delta in range(n - 1, -1, -1):
  6. for j in range(delta, n + 1):
  7. i = j - delta
  8. if i > 0:
  9. if dp[i - 1][j] < len(queries) and nums[i - 1] >= queries[dp[i - 1][j]]:
  10. dp[i][j] = dp[i - 1][j] + 1
  11. else:
  12. dp[i][j] = dp[i - 1][j]
  13. if j < len(nums):
  14. if dp[i][j + 1] < len(queries) and nums[j] >= queries[dp[i][j + 1]]:
  15. dp[i][j] = max(dp[i][j], dp[i][j + 1] + 1)
  16. else:
  17. dp[i][j] = max(dp[i][j], dp[i][j + 1])
  18. return max(dp[i][i] for i in range(n))

注:要保证每个状态的结果具有唯一性,例如最短超级串,如果只有mask一维空间则无法返回正确结果,因为相同mask不同拼接顺序的结果可能是不同的,正确代码如下:

  1. class Solution:
  2. def shortestSuperstring(self, words: List[str]) -> str:
  3. n = len(words)
  4. @cache
  5. def dp(mask, k):
  6. if mask.bit_count() == n: return (words[k] if k >= 0 else "")
  7. res = "0" * n * 12
  8. for i in range(n):
  9. if (mask >> i) & 1 == 1: continue
  10. pre = words[k] if k >= 0 else ""
  11. s = dp(mask | (1 << i), i)
  12. if pre in s and len(s) < len(res):
  13. res = s
  14. for j in range(len(pre)):
  15. if pre[:len(pre) - j] == s[len(s) - len(pre) + j:]:
  16. if len(s + pre[len(pre) - j:]) < len(res):
  17. res = s + pre[len(pre) - j:]
  18. break
  19. else:
  20. if len(s + pre) < len(res):
  21. res = s + pre
  22. return res
  23. return dp(0, -1)

13、转化转移方程,减少循环 -- 扣分后的最大得分

        该题直接使用暴力法超时,代码如下:

  1. class Solution:
  2. def maxPoints(self, points: List[List[int]]) -> int:
  3. m, n = len(points), len(points[0])
  4. dp = points[0]
  5. for i in range(1, m):
  6. temp = [0] * n
  7. for j in range(n):
  8. for jj in range(n):
  9. temp[j] = max(temp[j], points[i][j] + dp[jj] - abs(j - jj))
  10. dp = temp
  11. return max(dp)

 最终代码如下:

  1. class Solution:
  2. def maxPoints(self, points: List[List[int]]) -> int:
  3. m, n = len(points), len(points[0])
  4. dp = points[0]
  5. for i in range(1, m):
  6. temp = [0] * n
  7. maxl2 = [-inf]
  8. for j in reversed(range(n)):
  9. maxl2.append(max(maxl2[-1], dp[j] - j))
  10. maxl2 = maxl2[::-1]
  11. maxl = -inf
  12. for j in range(n):
  13. maxl = max(maxl, dp[j] + j)
  14. temp[j] = max(temp[j], points[i][j] - j + maxl, points[i][j] + j + maxl2[j])
  15. dp = temp
  16. return max(dp)

14、 使用二分法进行优化 -- 最多可以参加的会议数目 II 

  1. class Solution:
  2. def maxValue(self, events: List[List[int]], k: int) -> int:
  3. events.sort(key=lambda e: e[1])
  4. n = len(events)
  5. f = [[0] * (k + 1) for _ in range(n + 1)]
  6. for i, (start, end, val) in enumerate(events):
  7. p = bisect_left(events, start, hi=i, key=lambda e: e[1]) # hi=i 表示二分上界为 i(默认为 n)
  8. for j in range(1, k + 1):
  9. # 为什么是 p 不是 p+1:上面算的是 >= start,-1 后得到 < start,但由于还要 +1,抵消了
  10. f[i + 1][j] = max(f[i][j], f[p][j - 1] + val)
  11. return f[n][k]

15、使用单调栈进行优化 -- 工作计划的最低难度

  代码如下:

  1. class Solution:
  2. def minDifficulty(self, a: List[int], d: int) -> int:
  3. n = len(a)
  4. if n < d:
  5. return -1
  6. f = [[inf] * n for _ in range(d)]
  7. f[0] = list(accumulate(a, max))
  8. for i in range(1, d):
  9. st = [] # (下标 j,从 f[i-1][left[j]] 到 f[i-1][j-1] 的最小值)
  10. for j in range(i, n):
  11. mn = f[i - 1][j - 1] # 只有 a[j] 一项工作
  12. while st and a[st[-1][0]] <= a[j]: # 向左一直计算到 left[j]
  13. mn = min(mn, st.pop()[1])
  14. f[i][j] = mn + a[j] # 从 a[left[j]+1] 到 a[j] 的最大值是 a[j]
  15. if st: # 如果这一段包含 <=left[j] 的工作,那么这一段的最大值必然不是 a[j]
  16. f[i][j] = min(f[i][j], f[i][st[-1][0]]) # 答案和 f[i][left[j]] 是一样的
  17. st.append((j, mn)) # 注意这里保存的不是 f[i][j]
  18. return f[-1][-1]

16、改变递推顺序 --  通过给定词典构造目标字符串的方案数

如下代码的运行时间为3532ms

  1. class Solution:
  2. def numWays(self, words: List[str], target: str) -> int:
  3. m, n = len(words[0]), len(target)
  4. mp = defaultdict(lambda: defaultdict(int))
  5. for word in words:
  6. for i, c in enumerate(word):
  7. mp[c][i] += 1
  8. mod = 10 ** 9 + 7
  9. @cache
  10. def dfs(i, j):
  11. if i == n: return 1
  12. res = 0
  13. for k in range(j, m - (n - i - 1)):
  14. if mp[target[i]][k] > 0:
  15. res = (res + mp[target[i]][k] % mod * dfs(i + 1, k + 1) % mod) % mod
  16. return res % mod
  17. return dfs(0, 0)

改成对于word每个下标是否取用作为递推顺序,代码如下,运行时间为632ms

  1. class Solution:
  2. def numWays(self, words: List[str], target: str) -> int:
  3. m, n = len(words[0]), len(target)
  4. mp = defaultdict(lambda: defaultdict(int))
  5. for word in words:
  6. for i, c in enumerate(word):
  7. mp[c][i] += 1
  8. @cache
  9. def dfs(i, j):
  10. if i == n: return 1
  11. if m - j < n - i: return 0
  12. return (mp[target[i]][j] * dfs(i + 1, j + 1) + dfs(i, j + 1)) % (10 ** 9 + 7)
  13. return dfs(0, 0)

17、改变状态空间  -- 每个人戴不同帽子的方案数

如果以帽子使用作为状态空间时则超时,代码如下:

  1. class Solution:
  2. def numberWays(self, hats: List[List[int]]) -> int:
  3. @cache
  4. def dfs(mask):
  5. if mask.bit_count() == len(hats): return 1
  6. res = 0
  7. for h in hats[mask.bit_count()]:
  8. if (mask >> h) & 1 == 0:
  9. res = (res + dfs(mask | (1 << h))) % (10 ** 9 + 7)
  10. return res
  11. return dfs(0)

如果以人的使用作为状态空间时则通过,这是因为人数不超过10,帽子数为40,帽子的状态空间远远比人的要稀疏,代码如下:

  1. class Solution:
  2. def numberWays(self, hats: List[List[int]]) -> int:
  3. mp = defaultdict(list)
  4. s = set()
  5. for i, hat in enumerate(hats):
  6. for a_hat in hat:
  7. mp[a_hat].append(i)
  8. s.add(a_hat)
  9. s = list(s)
  10. @cache
  11. def dfs(i, mask):
  12. if mask.bit_count() == len(hats): return 1
  13. if i == len(s): return 0
  14. res = 0
  15. for j in mp[s[i]]:
  16. if (mask >> j) & 1 == 0:
  17. res = (res + dfs(i + 1, mask | (1 << j))) % (10 ** 9 + 7)
  18. res = (res + dfs(i + 1, mask)) % (10 ** 9 + 7)
  19. return res
  20. return dfs(0, 0)

18、使用前缀和进行优化 -- 生成数组

  1. class Solution:
  2. def numOfArrays(self, N: int, M: int, K: int) -> int:
  3. dp = [[[0 for _ in range(M + 1)] for _ in range(K + 1)] for _ in range(N + 1)]
  4. for k in range(1, M + 1):
  5. dp[1][1][k] = 1
  6. for i, j, k in itertools.product(range(1, N + 1), range(1, K + 1), range(M + 1)):
  7. dp[i][j][k] += dp[i - 1][j][k] * k
  8. dp[i][j][k] += sum(dp[i - 1][j - 1][1:k])
  9. return sum(dp[N][K][1:]) % (10 ** 9 + 7)

使用前缀和进行优化,以减少重复计算,代码如下:

  1. class Solution:
  2. def numOfArrays(self, n: int, m: int, k: int) -> int:
  3. # 不存在搜索代价为 0 的数组
  4. if k == 0:
  5. return 0
  6. f = [[[0] * (m + 1) for _ in range(k + 1)] for __ in range(n + 1)]
  7. mod = 10**9 + 7
  8. # 边界条件,所有长度为 1 的数组的搜索代价都为 1
  9. for j in range(1, m + 1):
  10. f[1][1][j] = 1
  11. for i in range(2, n + 1):
  12. # 搜索代价不会超过数组长度
  13. for s in range(1, min(k, i) + 1):
  14. # 前缀和
  15. presum_j = 0
  16. for j in range(1, m + 1):
  17. f[i][s][j] = (f[i - 1][s][j] * j + presum_j) % mod
  18. presum_j += f[i - 1][s - 1][j]
  19. # 最终的答案是所有 f[n][k][..] 的和
  20. # 即数组长度为 n,搜索代价为 k,最大值任意
  21. ans = sum(f[n][k][j] for j in range(1, m + 1)) % mod
  22. return ans

19、改变枚举方式 -- 最小不兼容性

(1)相比于传统回溯方法,使用combinations函数枚举新状态提升枚举速度

(2)对于一个子集没有相同元素的情况,使用哈希表过滤相同元素的方式来减少枚举范围。

(相比于直接枚举,再过滤相同元素有速度提升)

  1. class Solution:
  2. def minimumIncompatibility(self, nums: List[int], k: int) -> int:
  3. l = len(nums)
  4. k = l // k #k转换为每个子集的元素数量
  5. @lru_cache(None)
  6. def dfs(state):
  7. if state == 2 ** l - 1:#已遍历完,返回0
  8. return 0
  9. d = {}#哈希表,用于记录每个元素的最后下标
  10. for i in range(l):
  11. if state >> i & 1 == 0:#如果未被选,加入哈希表
  12. d[nums[i]] = i
  13. if len(d) < k:#如果非重复数量小于子集元素数量,则无法组合出子集
  14. return float("inf")
  15. res = float("inf")
  16. for it in itertools.combinations(d.keys(), k):#遍历所有组合
  17. nstate = state#新的状态位
  18. for j in it:
  19. nstate |= 1 << d[j]#根据哈希表更新
  20. res = min(res, max(it) - min(it) + dfs(nstate))#继续深搜
  21. return res
  22. res = dfs(0)
  23. return res if res < float("inf") else -1

20、使用贪心的方法 -- 连通两组点的最小成本

使用暴力的方法,代码如下,用时3620ms

  1. class Solution:
  2. def connectTwoGroups(self, cost: List[List[int]]) -> int:
  3. m, n = len(cost), len(cost[0])
  4. @cache
  5. def dfs(i, j, mask, f):
  6. if i == m: return 0 if mask.bit_count() == n else inf
  7. if j == n: return dfs(i + 1, 0, mask, 0) if f == 1 else inf
  8. res = dfs(i, j + 1, mask, f)
  9. res = min(res, dfs(i, j + 1, mask | (1 << j), 1) + cost[i][j])
  10. return res
  11. return dfs(0, 0, 0, 0)

左边只需要和右边的一个点匹配,当匹配完后,右边未匹配的点再贪心匹配左边的点,用时362ms,代码如下: 

  1. class Solution:
  2. def connectTwoGroups(self, s: List[List[int]]) -> int:
  3. n, m = len(s), len(s[0])
  4. h = [] # 预处理 h[j] 表示第 j 列的最小值
  5. for j in range(m):
  6. h.append(min([s[i][j] for i in range(n)]))
  7. @cache
  8. def dfs(p, a):
  9. # p 表示当前需要匹配的左边组的序号
  10. if p == n: # 左边的全部都匹配完
  11. # 将右边组没有匹配的全部与各自能匹配左边组的最小值进行贪心匹配
  12. return sum([h[j] if ((a >> j & 1) == 0) else 0 for j in range(m)])
  13. # 与右边的尝试进行一一匹配
  14. return min([dfs(p + 1, a | (1 << j)) + s[p][j] for j in range(m)])
  15. return dfs(0, 0)

求解最优路径

1、直接记录路径:(最大整除子集

  1. dp= [[x] for x in nums]
  2. for i in range(len(nums)):
  3. for j in range(i + 1, len(nums)):
  4. if nums[j] % nums[i] == 0 and len(dp[i]) + 1 > len(dp[j]):
  5. dp[j] = dp[i] + [nums[j]]
  6. return max(dp, key=len)

2、回溯: (最大整除子集

  1. maxl, temp = max(dp), None
  2. res, l = [], maxl
  3. for i in reversed(range(n)):
  4. if l == dp[i]:
  5. if l == maxl or temp is not None and temp % nums[i] == 0:
  6. temp = nums[i]
  7. res.insert(0, nums[i])
  8. l = l - 1

3、直接返回结果有可能无法保证返回顺序,例如找出分数最低的排列,可以先获取最小值,再递归回溯得到结果集,代码如下:

  1. class Solution:
  2. def findPermutation(self, nums: List[int]) -> List[int]:
  3. n = len(nums)
  4. @cache
  5. def dp(i, m, last):
  6. if i == n:
  7. return 0, []
  8. minl, res = inf, []
  9. for j in range(n):
  10. if (m >> j) & 1 == 1:
  11. continue
  12. if i > 0:
  13. tminl, tt = dp(i + 1, m | (1 << j), j)
  14. if tminl + abs(nums[j] - last) < minl:
  15. minl = tminl + abs(nums[j] - last)
  16. res = [j] + tt
  17. elif tminl + abs(nums[j] - last) == minl:
  18. if [j] + tt < res:
  19. res = [j] + tt
  20. else:
  21. tminl, tt = dp(i + 1, m | (1 << j), j)
  22. if tminl + abs(nums[j] - tt[-1]) < minl:
  23. minl = tminl + abs(nums[j] - tt[-1])
  24. res = [j] + tt
  25. elif tminl + abs(nums[j] - tt[-1]) == minl:
  26. if [j] + tt < res:
  27. res = [j] + tt
  28. return minl, res
  29. minl = dp(0, 0, -1)[0]
  30. res = []
  31. def dfs(m, s, r, last, zero):
  32. nonlocal res
  33. if res != []: return
  34. if m.bit_count() == n:
  35. if s + abs(nums[zero] - last) == minl:
  36. res = r
  37. return
  38. if s > minl: return
  39. for j in range(n):
  40. if (m >> j) & 1 == 1:
  41. continue
  42. if m.bit_count() > 0:
  43. dfs(m | (1 << j), s + abs(nums[j] - last), r + [j], j, zero)
  44. else:
  45. dfs(m | (1 << j), s, r + [j], j, j)
  46. dfs(0, 0, [], -1, -1)
  47. return res

注:使用记忆化搜索有可能会导致超时,例如1770. 执行乘法运算的最大分数,所以可以使用del func 或者func.cache_clear()清除缓存

参考资料

史上最全最丰富的“最长公共子序列”、“最长公共子串”问题的解法与思路_王小东大将军的博客-CSDN博客_最长公共子序列问题

动态规划-----两个字符串交叉组成第三个字符

​​​​​​动态规划解决01背包问题 - Christal_R - 博客园

最大连续子序列之和练习最大m子段和_QiaoRuoZhuo的专栏-CSDN博客

动态规划总结及题目推荐_IMLYZ的博客-CSDN博客

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/883009
推荐阅读
相关标签
  

闽ICP备14008679号