力扣小白刷题之650题只有两个键的键盘

题目描述

最初在一个记事本上只有一个字符‘A’。你每次可以对这个记事本进行两种操作：

1. Copy All：你可以复制这个记事本中的所有字符(部分的复制是不允许的)。
2. Paste：你可以粘贴你上一次复制的字符。

给定一个数字 n。你需要使用最少的操作次数，在记事本中打印出恰好 n 个 ‘A’。输出能够打印出 n 个 ‘A’ 的最少操作次数。

思路

参考自：https://leetcode-cn.com/problems/2-keys-keyboard/solution/dong-tai-gui-hua-fa-javashi-xian-by-zyu/

第一步：状态定义

• dp[i] 表示通过复制粘贴操作，得到 i 个字符，最少需要几步操作。

• 注：

  • 如果一个数是素数，那么最少操作就是一开始复制一个，最后一个一个粘贴
  • 如果一个数不是素数，那么最少操作就可以按它的因数分解一下，简化操作。
  • 举个例子，比如12，可以分解为以下几种情况：
    12 = 2 * 6，需要操作 CPCPPPPP 总共 8 步
    12 = 3 * 4，需要操作 CPPCPPP 总共 7 步
    12 = 4 * 3，需要操作 CPPPCPP 总共 7 步
    12 = 6 * 2，需要操作 CPPPPPCP 总共 8 步
  • 其实可以发现，因子相同的情况下，交换因子相乘的顺序，需要的步骤是一样的。我们可以简化一下分解的步骤，只需要找到小于 sqrt(n) 的因子即可。
  • 假设找到的因子是 j，那么需要的最小步骤就是 dp[j] + dp[i/j]，其中，dp[j] 表示需要多少步生成这个因子，dp[i/j] 表示需要多少步基于这个因子 j 得到 i。

第二步：状态转移方程

• 对于素数：dp[i] = i；
• 对于非素数：dp[i] = dp[j] + dp[i / j]
  • i 是 i 个字符( i 的范围是[2, n])，
  • j 是小于 sqrt(n) 的 i 的因数(j <= sqrt(n)， i % j == 0)。

第三步：初始化

• dp[i] = i;

代码