Leetcode 279. 完全平方数(Python3)

279. 完全平方数

给定正整数 n，找到若干个完全平方数（比如 1, 4, 9, 16, ...）使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

示例 1:

输入: n = 12
输出: 3 
解释: 12 = 4 + 4 + 4.

示例 2:

输入: n = 13
输出: 2
解释: 13 = 4 + 9.

---

class Solution(object):
    def numSquares(self, n):
        #py中通过,py3超时
        maxn = int(n ** 0.5)
        dp = [_ for _ in range(n + 1)]
        for i in range(1,maxn + 1):
            for j in range(i * i,n + 1):
                dp[j] = min(dp[j],dp[j - i * i] + 1)
        return dp[n]

---

class Solution(object):
    def numSquares(self, n):
        #@fibonacciWH 大佬的四平方定理做法
        #四平方定理： 任何一个正整数都可以表示成不超过四个整数的平方之和。 
        #推论：满足四数平方和定理的数n（四个整数的情况）
        #必定满足 n=4^a(8b+7)
        while n % 4 == 0: 
            n /= 4 
        if n % 8 == 7: 
            return 4 
        a = 0 
        while a**2 <= n: 
            b = int((n - a**2)**0.5) 
            if a**2 + b**2 == n: 
                return (not not a) + (not not b) 
            a += 1 
        return 3