279完全平方数
思路:
任何正整数都可以拆分成不超过4个数的平方和 如果一个数最少可以拆成4个数的平方和,则这个数还满足 n = (4^a)*(8b+7)
如果不满足,那只可能拆成1、2、3个数的平方和 如果这个数本来就是某个数的平方,那么答案就是1 如果不满足,那只能拆成2或3个数的平方和
如果答案是2,即n=a2+b2,那么我们可以枚举a 否则就只能是由三个数组成
代码:
import math
class Solution(object):
def numSquares(self, n):
"""
:type n: int
:rtype: int
"""
while n%4==0:
n /= 4
if n%8 == 7:
return 4
a = math.sqrt(n)
if int(a) == a:
return 1
for i in range(1,int(math.sqrt(n))+1):
b = math.sqrt(n - i**2)
if b == int(b):
return 2
return 3
(也可以用动态规划