树的广度优先遍历(BFS)-递归和迭代(Java)
## BFS上面这棵树的广度优先遍历(层序遍历)的顺序是:
A–>B–>C–>D–>E–>F–>G–>H–>I
英文缩写BFS(Breadth First Search
)。
其检验过程就是对每一层的节点的依次访问,访问完一层进入下一层,而且每个节点只能访问一次。
广度优先遍历树,需要用到队列(Queue)来存储节点对象,队列的特点就是先进先出。例如,上面这颗树的访问如下:
- 首先将A节点插入队列中,队列中有元素(A);
- 将A节点弹出,同时将A节点的左、右节点依次插入队列,B在队首,C在队尾,(B,C),此时得到A节点;
- 继续弹出队首元素,即弹出B,并将B的左、右节点插入队列,C在队首,E在队尾(C,D,E),此时得到B节点;
- 继续弹出,即弹出C,并将C节点的左、中、右节点依次插入队列,(D,E,F,G,H),此时得到C节点;
- 将D弹出,此时D没有子节点,队列中元素为(E,F,G,H),得到D节点;
- 。。。以此类推。。
迭代递归
public static List<Integer> bfs(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
List<Integer> list = new LinkedList<>();
if (root == null) return list;
queue.add(root);
while (!queue.isEmpty()) {
//将队列的头部的元素拿出来
TreeNode tree = queue.remove();
if (tree.left != null) queue.add(tree.left);
if (tree.right != null) queue.add(tree.right);
list.add(tree.val);
}
return list;
}
public List<TreeNode> traversal(TreeNode root) {
if (root == null) return null;
return bfs(root);
}
private List<TreeNode> bfs(TreeNode root) {
int curNum = 1;
int nextNum = 0;
Queue<TreeNode> queue = new LinkedList<>();
List<TreeNode> res = new ArrayList<>();
queue.offer(root);
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
res.add(node);
curNum--;
if (node.left != null) {
queue.add(node.left);
nextNum++;
}
if (node.right != null) {
queue.add(node.right);
nextNum++;
}
if (curNum == 0) {
curNum = nextNum;
nextNum = 0;
}
}
return res;
}
public List<TreeNode> Bfs_tree(TreeNode root){
Queue<TreeNode> myq = new LinkedList<>();
List<TreeNode> res = new ArrayList<>();
if(root==null) return null;
myq.add(root);
while(!myq.isEmpty()){
int len = myq.size();
for(int i=0;i<len;i++){
if(myq.peek().left!=null) myq.add(myq.peek().left);
if(myq.peek().right!=null) myq.add(myq.peek().right);
res.add(myq.poll());
}
}
return res;
}