原文链接: Depth First Search/Traversal in Binary Tree
目标: -给定一个二叉树,进行深度优先搜索/遍历
方法:
- 方法非常简单,使用栈
- 首先把根节点加入栈
- 从栈中弹出一个元素,然后把它的右孩子和左孩子加入栈
- 弹出一个元素并打印出来,然后把它的孩子加入栈
- 重复上面的两个步骤直到栈为空.
例子:
完整代码:
public class BSTDFS {
public void DFS(Node root) {
Stack<Node> s = new Stack<Node>();
s.add(root);
while (s.isEmpty() == false) {
Node x = s.pop();
if(x.right!=null) s.add(x.right);
if(x.left!=null) s.add(x.left);
System.out.print(" " + x.data);
}
}
public static void main(String args[]){
Node root = new Node(1);
root.left = new Node(2);
root.left.left = new Node(4);
root.left.right = new Node(5);
root.right = new Node(3);
root.right.left = new Node(6);
root.right.right = new Node(7);
BSTDFS b = new BSTDFS();
System.out.println("Depth-First-Search : ");
b.DFS(root);
}
}
class Node {
int data;
Node left;
Node right;
public Node(int data) {
this.data = data;
left = null;
right = null;
}
}
输出:
Depth-First-Search :
1 2 4 5 3 6 7