【转载】Python实现 leetcode 279. Perfect Squares

问题

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.

给定一个正整数n，一定存在若干整数平方和为该正整数，求满足该条件的最小整数个数。平方数为（1，4，9，16…），使其和为n。
例如给定n=12，则返回3，因为12=4+4+4；给定n=13，则返回2，因为13=4+9.

解决思路

1.动态规划

使用动态规划，维护一个长度为n+1的数组，第i位存储和为i的最少整数个数。则f(i)只与i之前的数组有关。

用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]

2.四平方和定理

变态解法（四平方和定理），任何一个正整数都可以由4个整数的平方和组成。即对所有n，都有n=aa+bb+cc+dd。

四平方和定理的用法

也就是说，这个题目返回的答案只有1、2、3、4这四种可能。

推理1. 我们可以将输入的数字除以4来大大减少计算量，并不改变答案。

推理2. 一个数除以8的余数，如果余数为7， 则其必然由四个完全平方数组成。

推理3. 然后检测是否可以将简化后的数拆分为两个完全平方数，否则一定由三个完全平方数组成。

代码：

class Solution:
    def numSquares(n):
        """
        :type n: int
        :rtype: int
        """
        import math
        while n % 4 == 0: 
        	n = n // 4
        if n % 8 == 7: 
        	return 4
        if int(math.sqrt(n)) ** 2 == n: 
        	return 1
        i = 1
        while i*i <= n:
            j = math.sqrt(n - i*i)
            if int(j) == j: 
            	return 2
            i += 1
        return 3

转载自
https://blog.csdn.net/weixin_37251044/article/details/89003696