完全平方数这题我就用BFS去解了,因为我感觉最容易理解的就是BFS(?)至少现在这么认为。。。
广度优先遍历就是利用队列,如图所示,12入队,12出队,现在我要消耗完13,那么只有i<n^2的范围内,也就是说剩下的数字就是11,8,3,接着,这三个数字入队,以此类推衍生除了10,7,2,7,4,2这些,再将其入队,又可得出子子子子节点,何时结束呢,一直找到那个消耗完的时候就结束了
未被优化的算法
public int numSquares(int n) {
Queue<Integer> queue = new ArrayDeque<>();
queue.add(n);
Set<Integer> visited = new HashSet<>();
int level = 0;
while (!queue.isEmpty()) {
level++;
int length = queue.size();
for (int i = 0; i < length; i++) {
int remain = queue.poll();
for (int j = 0; j * j <= remain; j++) {
int next = remain - j * j;
if (next == 0) {
return level;
}
queue.add(next);
}
}
}
return level;
}
在此,我们看到很多重复的值,假如说前面遇到了7,很难凑到底,那么可以将其排除,建个备忘录,
public int numSquares(int n) {
Queue<Integer> queue = new ArrayDeque<>();
queue.add(n);
Set<Integer> visited = new HashSet<>();
int level = 0;
while (!queue.isEmpty()) {
level++;
int length = queue.size();
for (int i = 0; i < length; i++) {
int remain = queue.poll();
for (int j = 0; j * j <= remain; j++) {
int next = remain - j * j;
if (next == 0) {
return level;
}
if (!visited.contains(next)) {
queue.add(next);
visited.add(next);
}
}
}
}
return level;
}
在画图的过程中我还发现了个问题,我们一直把0纳入到了循环的过程中,但是0是可以被忽视的
public int numSquares(int n) {
Queue<Integer> queue = new ArrayDeque<>();
queue.add(n);
Set<Integer> visited = new HashSet<>();
int level = 0;
while (!queue.isEmpty()) {
level++;
int length = queue.size();
for (int i = 0; i < length; i++) {
int remain = queue.poll();
for (int j = 1; j * j <= remain; j++) {
int next = remain - j * j;
if (next == 0) {
return level;
}
if (!visited.contains(next)) {
queue.add(next);
visited.add(next);
}
}
}
}
return level;
}