[leetcode] 279. Perfect Squares @ python

原题

Given a positive integer n, find the least number of perfect square numbers (for example, 1, 4, 9, 16, …) which sum to n.

Example 1:

Input: n = 12
Output: 3
Explanation: 12 = 4 + 4 + 4.
Example 2:

Input: n = 13
Output: 2
Explanation: 13 = 4 + 9.

解法

动态规划, 用dp[i]表示数字组成数字i的最少完美平方的个数, 状态转移方程

dp[i] = min(dp[i], dp[i - j*j] +1)

为方便迭代, 初始化dp[0] = 0, dp[1] = 1, dp[i] = n, 遍历2到n, 不断更新dp[i].

以n = 13为例, 13 - 2*2 = 9, 那么dp[13] = min(dp[13] + dp[9] + 1), 我们求dp[9]的最少完美平方的个数, 加上1就是dp[13]的结果.

Time: O(n)
Space: O(n)

代码

class Solution:
    def numSquares(self, n: 'int') -> 'int':
        dp = [n]*(n+1)
        dp[0] = 0
        dp[1] = 1
        for i in range(2, n+1):
            j = 1
            while j*j <= i:
                dp[i] = min(dp[i], dp[i-j*j] + 1)
                j += 1
        return dp[-1]