Leetcode 3669. Balanced K-Factor Decomposition

  • [Leetcode 3669. Balanced K-Factor Decomposition](#Leetcode 3669. Balanced K-Factor Decomposition)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题思路上同样还是比较直观的,就是分两步:

  • 第一步就是找出给定的 n n n的所有质因数;
  • 第二步就是探索有的 n n n的质因数的组合,将其合并为 k k k个因数,并使其最为接近。

其中,关于第一步,我们只需要提前计算出所有的质因数,然后遍历一下看看 n n n的所有质因数的分布即可。

然后关于第二步,我们使用一个动态规划即可处理。

2. 代码实现

给出python代码实现如下:

python 复制代码
def get_primes(n):
    primes = []
    status = [0 for _ in range(n+1)]
    for i in range(2, n+1):
        if status[i] == 1:
            continue
        primes.append(i)
        for j in range(i, n+1, i):
            status[j] = 1
    return primes

PRIMES = get_primes(10**5)

class Solution:
    def minDifference(self, n: int, k: int) -> List[int]:
        
        def split(n):
            factors = []
            for p in PRIMES:
                if n < p:
                    break
                while n % p == 0:
                    factors.append(p)
                    n = n // p
            return factors

        factors = split(n)
        if len(factors) <= k:
            return [1] * (k-len(factors)) + factors
        # print(factors)

        @lru_cache(None)
        def dfs(arr):
            if len(arr) == k:
                return arr
            ans = []
            e = arr[0]
            for i, num in enumerate(arr):
                if i == 0:
                    continue
                nxt = list(arr[1:i] + arr[i+1:])
                bisect.insort(nxt, e * num)
                candi = dfs(tuple(nxt))
                if ans == [] or candi[-1]-candi[0] < ans[-1]-ans[0]:
                    ans = candi
            return ans

        return dfs(tuple(factors))

提交代码评测得到:耗时12ms,占用内存18.55MB。

相关推荐
青云交7 小时前
Java 大视界 -- Java 大数据在智能公交调度优化与准点率提升中的应用实践(416)
java·动态规划·flink cep·spark mllib·智能公交调度·杭州公交案例·准点率提升
贝塔实验室7 小时前
ADMM 算法的基本概念
算法·数学建模·设计模式·矩阵·动态规划·软件构建·傅立叶分析
dragoooon341 天前
[优选算法专题三.二分查找——NO.20搜索插入位置 ]
算法·leetcode·动态规划
睡不醒的kun11 天前
leetcode算法刷题的第三十四天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
散11212 天前
01数据结构-初探动态规划
数据结构·动态规划
cwplh13 天前
MX模拟赛总结
算法·动态规划
睡不醒的kun13 天前
leetcode算法刷题的第三十二天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
共享家952713 天前
经典动态规划题解
算法·leetcode·动态规划
二哈不在线13 天前
代码随想录二刷之“动态规划”~GO
算法·golang·动态规划
楼田莉子14 天前
C++动态规划算法:斐波那契数列模型
c++·学习·算法·动态规划