现在假设有一颗这样树,(是不是二叉树都没关系,原理都是一样的)
宽度优先搜索
英文缩写为BFS即( Breadth FirstSearch)。其过程检验来说是对每一层节点依次访问,访问完一层进入下一层,而且每个节点只能访问一次。对于上面的例子来说,广度优先遍历的 结果是:A,B,C,D,E,F,G,H,I(假设每层节点从左到右访问)。
宽度优先遍历各个节点,需要使用到队列(Queue)这种数据结构,queue的特点是先进先出,其实也可以使用双端队列,区别就是双端队列首尾都可以插入和弹出节点。整个遍历过程如下:
首先将A节点插入队列中,queue(A);
将A节点弹出,同时将A的子节点B,C插入队列中,此时B在队列首,C在队列尾部,queue(B,C);
将B节点弹出,同时将B的子节点D,E插入队列中,此时C在队列首,E在队列尾部,queue(C,D,E);
将C节点弹出,同时将C的子节点F,G,H插入队列中,此时D在队列首,H在队列尾部,queue(D,E,F,G,H);
将D节点弹出,D没有子节点,此时E在队列首,H在队列尾部,queue(E,F,G,H);
...依次往下,最终遍历完成,Java代码大概如下:
//二叉树的定义
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
import java.util.Queue;
import java.util.LinkedList;
public void breadthFirstSearch(TreeNode root){
if(root==null){
System.out.print("empty tree");
return;
}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(root);
while(queue.size() != 0){
TreeNode tempNode= queue.remove();
System.out.print(tempNode.val+" ");
if(tempNode.left!=null){
queue.add(tempNode.left);
}
if(tempNode.right!=null){
queue.add(tempNode.right);
}
}
}
深度优先
英文缩写为DFS即(Depth First Search)。其过程简要来说是对每一个可能的分支路径深入到不能再深入为止,而且每个节点只能访问一次。对于上面的例子来说深度优先遍历的结果就是:A,B,D,E,I,C,F,G,H.(假设先走子节点的的左侧)。
(1)深度优先的递归算法很简单,以先序遍历为例:
public void preOrderTraverse1(TreeNode root) {
if (root != null) {
System.out.print(root.val+" ");
preOrderTraverse1(root.left);
preOrderTraverse1(root.right);
}
}
(2)深度优先的非递归算法
深度优先遍历各个节点,需要使用到栈(Stack)这种数据结构。stack的特点是是后进先出。整个遍历过程如下:
首先将A节点压入栈中,stack(A);
将A节点弹出,同时将A的子节点C,B压入栈中,(注意先将右孩子结点入栈)此时B在栈的顶部,stack(B,C);
将B节点弹出,同时将B的子节点E,D压入栈中,此时D在栈的顶部,stack(D,E,C);
将D节点弹出,没有子节点压入,此时E在栈的顶部,stack(E,C);
将E节点弹出,同时将E的子节点I压入,stack(I,C);
...依次往下,最终遍历完成,Java代码大概如下:
//二叉树的定义
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
import java.util.Stack;
import java.util.LinkedList;
public void depthFirstSearch(TreeNode root){
if(root==null){
System.out.print("empty tree");
return;
}
Stack<TreeNode> stack= new LinkedList<TreeNode>();
stack.push(root);
while(stack.empty != 0){
TreeNode tempNode= stack.pop();
System.out.print(tempNode.val+" ");
if(tempNode.right!=null){//由于栈的后进先出特性,先将右子树压入栈中
stack.push(tempNode.right);
}
if(tempNode.left!=null){
stack.push(tempNode.left);
}
}
}
图片来源:树图来源