例1 完美平方
1. 问题描述
给定一个正整数n,找到若干个完全平方数(例如:1,4,9,…),使得它们的和等于n,完全平方数的个数最少。
2.问题示例
给出n=8,返回2,因为8=4+4;给出n=25,返回1,因为25=25。
3.代码实现
可以使用动态规划的思想来解决这个问题,具体步骤如下:
-
定义一个长度为n+1的数组dp,dp[i]表示数字i最少可以由几个完全平方数相加得到。
-
初始化dp[0]=0,因为0本身就是完全平方数。
-
对于每个数字i,从1开始枚举所有小于等于i的完全平方数jj,然后更新dp[i]的值,即dp[i]=min(dp[i], dp[i-jj]+1)。
-
最终dp[n]就是答案。
import math
def numSquares(n: int) -> int:
dp = [float('inf')] * (n+1)
dp[0] = 0
for i in range(1, n+1):
for j in range(1, int(math.sqrt(i))+1):
dp[i] = min(dp[i], dp[i-j*j]+1)
return dp[n]
print(numSquares(8))
print(numSquares(25))
运行结果:
可以使用广度优先搜索(BFS)的方法来解决这个问题。具体步骤如下:
-
定义一个队列,将初始状态 (n, 0) 入队,其中 n 表示当前剩余的数字,0 表示当前已经使用的完全平方数的个数。
-
进入循环,直到队列为空为止。在每一轮循环中,取出队首元素 (num, count)。
-
判断 num 是否为完全平方数,如果是,则返回 count+1,即找到了最少的完全平方数个数。
-
否则,对于从 1 开始的每个完全平方数 i^2,计算新的剩余数字 new_num = num - i^2,并将 (new_num, count+1) 入队。
import math
from collections import deque
def numSquares(n: int) -> int:
queue = deque([(n, 0)])
visited = set([n])
while queue:
num, count = queue.popleft()
if num == 0:
return count
for i in range(1, int(math.sqrt(num)) + 1):
new_num = num - i*i
if new_num not in visited:
queue.append((new_num, count + 1))
visited.add(new_num)
return -1 # 如果无法找到完全平方数相加等于 n 的情况
# 测试样例
print(numSquares(8)) # 输出 2
print(numSquares(25)) # 输出 1
运行结果: