切分数组

标签: 数组 数学 动态规划 数论

难度: Hard

给定一个整数数组 nums ,小李想将 nums 切割成若干个非空子数组,使得每个子数组最左边的数和最右边的数的最大公约数大于 1 。为了减少他的工作量,请求出最少可以切成多少个子数组。

示例 1:

输入:nums = [2,3,3,2,3,3]

输出:2

解释:最优切割为 [2,3,3,2] 和 [3,3] 。第一个子数组头尾数字的最大公约数为 2 ,第二个子数组头尾数字的最大公约数为 3 。

示例 2:

输入:nums = [2,3,5,7]

输出:4

解释:只有一种可行的切割:[2], [3], [5], [7]

限制:

  • 1 <= nums.length <= 10^5
  • 2 <= nums[i] <= 10^6

Submission

运行时间: 361 ms

内存: 38.0 MB

M = 10 ** 6 + 1
np = [0] * M
pfs=[2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,79,83,89,97,101,103,107,109,113,127,131,137,139,149,151,157,163,167,173,179,181,191,193,197,199,211,223,227,229,233,239,241,251,257,263,269,271,277,281,283,293,307,311,313,317,331,337,347,349,353,359,367,373,379,383,389,397,401,409,419,421,431,433,439,443,449,457,461,463,467,479,487,491,499,503,509,521,523,541,547,557,563,569,571,577,587,593,599,601,607,613,617,619,631,641,643,647,653,659,661,673,677,683,691,701,709,719,727,733,739,743,751,757,761,769,773,787,797,809,811,821,823,827,829,839,853,857,859,863,877,881,883,887,907,911,919,929,937,941,947,953,967,971,977,983,991,997] 
for p in pfs:
    for i in range(p*p,M,p):
        np[i]=p

class Solution:
    def splitArray(self, nums: List[int]) -> int:
        f = {}
        pre = 0
        for i,x in enumerate(nums):
            cur = 100001
            while np[x]:    #如果还存在小于1000的质因素
                # p = np[x]   #当前质因素
                # while x % p == 0:
                    # x //= p
                
                f[np[x]] = min(f.get(np[x],100001),pre + 1)   #跳跃到fac这个因子,需要多少步 f[fac]。
                cur = min(cur,f[np[x]])
                x//=np[x]    
                
            if x>1:  # 如果存在大于1000的质因素
                f[x] = min(f.get(x,100001),pre + 1)
                cur = min(cur,f[x])
            pre = cur
        return cur

Explain

本题解采用动态规划的思想以及预处理的方法来解决问题。首先,通过一个预处理步骤,使用筛法计算每个数字的最小质因子。这个预处理步骤是基于一个修改版的埃拉托斯特尼筛法,只针对较小的质数。对于数组中的每个数字,我们使用预处理的结果来快速找到它的所有质因子,并且通过动态规划的方式计算如何分割数组以满足题目要求。动态规划的状态 f[p] 表示最后一个使用质数 p 作为头尾最大公约数的子数组所形成的最小切割数。对于每个数字 x,我们更新它的所有质因子对应的状态,然后找出最小的切割数作为当前位置的结果,并不断向前推进。

时间复杂度: O(n log x)

空间复杂度: O(M)

M = 10 ** 6 + 1
np = [0] * M
pfs=[2,3,5,7,...,997] # 列出了所有小于1000的质数
for p in pfs:
    for i in range(p*p, M, p):
        np[i]=p  # 使用修改的埃拉托斯特尼筛法计算最小质因子
class Solution:
    def splitArray(self, nums: List[int]) -> int:
        f = {}  # 动态规划状态存储
        pre = 0  # 记录当前的最小切割数
        for i, x in enumerate(nums):
            cur = 100001  # 初始化当前位置的最小切割数
            while np[x]:  # 遍历 x 的所有质因子
                f[np[x]] = min(f.get(np[x], 100001), pre + 1)  # 更新使用当前质因子的最小切割数
                cur = min(cur, f[np[x]])  # 更新当前位置的最小切割数
                x //= np[x]  # 除去当前质因子
            if x > 1:  # 如果 x 有大于1000的质因子
                f[x] = min(f.get(x, 100001), pre + 1)
                cur = min(cur, f[x])
            pre = cur  # 更新前一个位置的最小切割数
        return cur  # 返回最终的最小切割数

Explore

题解中所提到的埃拉托斯特尼筛法的修改主要集中在不是标记所有的合数,而是只标记每个合数的最小质因子。这种修改使得在后续处理中,我们可以快速获取任何一个数的最小质因子,而不需要经过复杂的因式分解。具体来说,对于每个质数p,我们从p*p开始(因为小于p*p的合数已经被更小的质数标记过了),以p为步长遍历,将这些位置上的数标记为p。这样,当我们查询一个数的最小质因子时,可以直接得到,从而加速了解题过程中对质因子的查询。

题解中使用的动态规划的状态定义为f[p],这表示以质数p为头尾的子数组的最小切割数。状态转移方程是基于每个数字x的质因子来更新的。具体地,对于每个数字x,我们遍历其所有质因子,并尝试更新这个质因子对应的状态f[p]。更新规则是:f[p] = min(f.get(p, 100001), pre + 1),其中pre是上一个位置的最小切割数。这样,通过不断更新各个质因子对应的状态,我们可以计算出到当前数字为止的最小切割数。

题解中选择只计算小于1000的质数是因为,在大多数情况下,较小的质数是组成数的主要质因子,而大于1000的质数较少出现作为因子,特别是在考虑到数字范围常常远大于1000的情况下。处理大于1000的质因子时,题解采用的策略是直接检查剩余的数x是否大于1,如果是,那么这个数本身是一个质数,因此直接更新f[x] = min(f.get(x, 100001), pre + 1)。这种处理方式确保了即使是大于1000的质数也能被正确地处理,而不需要额外的复杂筛法。