队列 --- Leedcode 279 完全平方数 （Medium）

题目

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

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

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

解析

这题本是一道动态规划的题，但这篇文章介绍使用队列来解题。

广度优先遍历：基于队列实现：

• 从n到0,每个数表示一个节点
• 如果两数之间相差一个完全平方数，则在两点之间连接一条边
• 最终找到n与0之间的最短路径

下图以11为例：

• 11入队，取出队首，找到与其相差完全平方数的数入队（10、7、2入队） ，层数加一。
• 再次取出队列中的所有元素（取一层），并且将相差完全平方数的数入队。如果取出的数为零，此时的层数就是所求结果；否则如果该层没有出现0，层数加一，再次操作步骤2
• 取到0，就是最短的路径。而红色区域表示，被重复添加的数字。
  
  上面的操作，小伙伴们可以取一个稍微大的数来画一画。

代码一：

 public int numSquares(int n) {
       Queue<Integer> queue = new LinkedList<>();
		queue.offer(n);
		int count = 0; // 记录层数  层数就是最短路径 
		while(!queue.isEmpty()) {
			// 每次循环将队列中的元素取完（取完一层）
			int len = queue.size();
			for(int i = 0; i<len;i++) {
				int temp = queue.poll();// 取出队首
				// System.out.println(temp);
				if(temp == 0) { // 找到0   就找到了最短路径 
					return count;
				}
				// 将相差平方数的数添加到队列
				  for(int k = 1; temp - k*k >= 0;k++){
					    	queue.offer(temp - k*k); 
	                }
			}
			count++;
		}
		return count;
    }

这版代码会超时，因为当数据比较大时，会有很多重复的元素被加入（这里）：

代码二：

解决上面的重复问题，我们加一个数组，只要该数被添加过我们就不再重复添加。

 public int numSquares(int n) {
       Queue<Integer> queue = new LinkedList<>();
		queue.offer(n);
		int [] visited = new int[n+1];
		int count = 0; // 记录层数  层数就是最短路径 
		while(!queue.isEmpty()) {
			// 每次循环将队列中的元素取完（取完一层）
			int len = queue.size();
			for(int i = 0; i<len;i++) {
				int temp = queue.poll();// 取出队首
				// System.out.println(temp);
				if(temp == 0) { // 找到0   就找到了最短路径 
					return count;
				}
				// 将相差平方数的数添加到队列
				  for(int k = 1; temp - k*k >= 0;k++){
					    if(visited[temp - k*k] == 0) {
					    	queue.offer(temp - k*k);
					    	visited[temp - k*k] = 1;
					    }
	                    
	                }
			}
			count++;
		}
		return count;
    }