Leetcode343.整数拆分

 

题目

给定一个正整数 n ，将其拆分为 k 个 正整数 的和（ k >= 2 ），并使这些整数的乘积最大化。

返回 你可以获得的最大乘积 。

示例 1:

输入: n = 2
输出: 1
解释: 2 = 1 + 1, 1 × 1 = 1。

示例 2:

输入: n = 10
输出: 36
解释: 10 = 3 + 3 + 4, 3 × 3 × 4 = 36。

提示:

• 2 <= n <= 58

思路

动规五部曲，分析如下：

1 确定dp数组（dp table）以及下标的含义

dp[i]：分拆数字i，可以得到的最大乘积为dp[i]。

2 确定递推公式

可以想 dp[i]最大乘积是怎么得到的呢？

其实可以从1遍历j，然后有两种渠道得到dp[i].

一个是j * (i - j) 直接相乘。

一个是j * dp[i - j]，相当于是拆分(i - j)，对这个拆分不理解的话，可以回想dp数组的定义。

j是从1开始遍历，拆分j的情况，在遍历j的过程中其实都计算过了。那么从1遍历j，比较(i - j) * j和dp[i - j] * j 取最大的。递推公式：dp[i] = max(dp[i], max((i - j) * j, dp[i - j] * j));

也可以这么理解，j * (i - j) 是单纯的把整数拆分为两个数相乘，而j * dp[i - j]是拆分成两个以及两个以上的个数相乘。

如果定义dp[i - j] * dp[j] 也是默认将一个数强制拆成4份以及4份以上了。

所以递推公式：dp[i] = max({dp[i], (i - j) * j, dp[i - j] * j});

3 dp的初始化

dp初始化为n+1的全0列表

dp[2] = 1

4 确定遍历顺序

确定遍历顺序，先来看看递归公式：dp[i] = max(dp[i], max((i - j) * j, dp[i - j] * j));

dp[i] 是依靠 dp[i - j]的状态，所以遍历i一定是从前向后遍历，先有dp[i - j]再有dp[i]。

所以遍历顺序为：

for (int i = 3; i <= n; ++i) {
    for (int j = 1; j<= i - 1; ++j) {
        dp[i] = max(dp[i], max((i - j) * j, dp[i - j] * j));
    }
}

优化上述代码:

for (int i = 3; i <= n; ++i) {
    for (int j = 1; j<= i / 2; ++j) {
        dp[i] = max(dp[i], max((i - j) * j, dp[i - j] * j));
    }
}

C++代码

#include <iostream>
#include <vector>

int integerBreak(int n) {
    std::vector<int> dp(n + 1);
    dp[2] = 1;
    for (int i = 3; i <= n; ++i) {
        for (int j = 1; j<= i / 2; ++j) {
            dp[i] = std::max(dp[i], std::max((i - j) * j, dp[i - j] * j));
        }
    }
    return dp[n];
}

int main() {
    int n = 10;
    std::cout << integerBreak(n);
    return 0;
}

python代码

def integerBreak(n):
    dp = [0] * (n + 1)
    dp[2] = 1
    for i in range(3, n+1):
        for j in range(1, i//2 + 1):
            dp[i] = max(dp[i], j * (i - j), j * dp[i - j])
    return dp[-1]
        

if __name__ == '__main__':
    print(integerBreak(2))
    print(integerBreak(5))
    print(integerBreak(10))