leetcode 剑指offer 14:剪绳子

给你一根长度为 n 的绳子,请把绳子剪成整数长度的 m 段(m、n都是整数,n>1并且m>1),每段绳子的长度记为 k[0],k[1]…k[m - 1] 。请问 k[0]k[1]…*k[m - 1] 可能的最大乘积是多少?例如,当绳子的长度是8时,我们把它剪成长度分别为2、3、3的三段,此时得到的最大乘积是18。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。

示例 1:
输入: 2
输出: 1
解释: 2 = 1 + 1, 1 × 1 = 1
示例 2:
输入: 10
输出: 36
解释: 10 = 3 + 3 + 4, 3 × 3 × 4 = 36

提示:2 <= n <= 1000

思路: 动态规划
状态定义: dp[i] 将长度为i的绳子拆分后可得到的最大乘积
状态转移:dp[i] = max(dp[i], dp[i-j]*j, (i-j)*j) 和将i拆分成j以i-j和j, 以及将j再拆分两种情况

class Solution:
    def cuttingRope(self, n: int) -> int:
        dp = [1 for i in range(n+1)]
        for i in range(2, n+1):
            for j in range(1, i):
                dp[i] = max(dp[i], dp[i-j]*j, (i-j)*j)
        return dp[-1]

这题也可以通过数学的思想找规律进行求解,具体参考题解链接
主要思想: 经过推理,将绳子以3长度为一段分可以得到较大的乘积,具体通过一系列求导可以得到
算法流程:
当 n≤3 时,按照规则应不切分,但由于题目要求必须剪成 m>1 段,因此必须剪出一段长度为 1 的绳子,即返回 n−1 。
当 n>3 时,求 n 除以 3 的 整数部分 a 和 余数部分 b (即n=3a+b ),并分为以下三种情况(设求余操作符号为 “⊙” ):
当 b=0 时,直接返回 3^a⊙1000000007;
当 b = 1时,要将一个 1+3 转换为 2+2,因此返回 (3^ a−1×4)⊙1000000007;
当 b = 2时,返回 (3 ^ a×2)⊙1000000007。
这里考虑了打算越界的问题,但是在python中不存在这一问题,不过也可以学习一下大数求余数的两种方式:循环求余 、 快速幂求余 ,其中后者的时间复杂度更低,两种方法均基于以下求余运算规则推出:

(xy)⊙p=[(x⊙p)(y⊙p)]⊙p

  1. 循环求余:
    根据求余运算性质推出(∵ 本题中 x<p,∴ x⊙p=x ):
    x^a ⊙p=[(x ^ a−1 ⊙p)(x⊙p)]⊙p=[(x ^a−1 ⊙p)x]⊙p
    解析: 利用此公式,可通过循环操作依次求 x ^ 1 , x^ 2, …, x^ {a-1}, x^ a
    对 p 的余数,保证每轮中间值 rem 都在 int32 取值范围中。封装方法代码如下所示。
# 求 (x^a) % p —— 循环求余法
def remainder(x, a, p):
    rem = 1
    for _ in range(a):
        rem = (rem * x) % p
    return rem

时间复杂度 O(N) : 其中 N=a ,即循环的线性复杂度。

  1. 快速幂求余:
    根据求余运算性质可推出:
    在这里插入图片描述

当 a 为奇数时 a/2 不是整数,因此分为以下两种情况( ‘’ // ‘’ 代表向下取整的除法):
在这里插入图片描述
解析: 利用以上公式,可通过循环操作每次把指数 a 问题降低至指数 a//2 问题,只需循环log 2(N) 次,因此可将复杂度降低至对数级别。封装方法代码如下所示。

# 求 (x^a) % p —— 快速幂求余
def remainder(x, a, p):
    rem = 1
    while a > 0:
        if a % 2: rem = (rem * x) % p
        x = x ** 2 % p
        a //= 2
    return rem

所以本题的代码可以写为:

class Solution:
    def cuttingRope(self, n: int) -> int:
        if n <= 3: return n - 1
        a, b, p, x, rem = n // 3 - 1, n % 3, 1000000007, 3 , 1
        while a > 0:
            if a % 2: rem = (rem * x) % p
            x = x ** 2 % p
            a //= 2
        if b == 0: return (rem * 3) % p # = 3^(a+1) % p
        if b == 1: return (rem * 4) % p # = 3^a * 4 % p
        return (rem * 6) % p # = 3^(a+1) * 2  % p

(但其实python不需要考虑大数越界的问题)

# 由于语言特性,Python 可以不考虑大数越界问题
class Solution:
    def cuttingRope(self, n: int) -> int:
        if n <= 3: return n - 1
        a, b, p = n // 3, n % 3, 1000000007
        if b == 0: return 3 ** a % p
        if b == 1: return 3 ** (a - 1) * 4 % p
        return 3 ** a * 2 % p
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值