【例1 完美平方】

---

问题描述

给定一个正整数n，找到若干个完全平方数，是它们的和等于n，完全平方数的个数最少。

---

问题示例

n = 12, 12 = 4+4+4, return 3
n = 13, 13 = 4+9, return 2

---

算法思想

由 Lagrange’s four-square theorem(四平方和定理)
得知返回值只可能是1，2，3，4

---

代码实现

class Solution:
    def numSquare(self, n):
        while n % 4 == 0:
            n //= 4  # 代码的优化
        if n % 8 == 7:  # 7 = 1+1+1+4
            return 4
        for i in range(n+1):
            temp = i * i
            if temp <= n:
                if int((n - temp) ** 0.5) ** 2 + temp == n:
                    return 1 + (0 if temp == 0 else 1)
            else:
                break
        return 3
if __name__ == '__main__':
    n = 12
    print("初始值：",n)
    solution = Solution()
    print("结果：",solution.numSquare(n))

---