首先了解创建树以及进行树的连接的代码如下
TreeNode tree1 = new TreeNode(1);
TreeNode tree2 = new TreeNode(2);
TreeNode tree3 = new TreeNode(3);
TreeNode tree4 = new TreeNode(4);
tree1.leftNode = tree2;
tree1.rightNode = tree3;
tree2.leftNode = tree4;
深度优先遍历(Depth First Search)
原理:从一个未访问的节点开始,沿着一条路走到最后一个节点,再从最后一个节点回退到上一个节点,再 从另一条路走到底,不断重复递归此过程,直到遍历完所有的节点。
第一种实现方法:递归实现
public void dfs(TreeNode head){
if(head == null)
{
return ;
}
System.out.println(head.val);
dfs(head.left);
dfs(head.right);
}
缺点是层级过多时容易造成栈溢出;
第二种方法:利用栈来实现(先进后出)
public void show(TreeNode head){
Stack stack = new Stack<>;
if(head == null){return ;}
stack.add(head);
while(!stack.isEmpty()){
TreeNode node = stack.pop();
System.out.println(node.val);
if(node.rightNode!=null){
stack.push(node.rightNode);
}
if(node.leftNode!=null){
stack.push(node.leftNode);
}
}
}
广度优先遍历BFS(Breath First Search)
原理是从一个节点出发,先遍历这个节点的相邻节点,再依次遍历每个相邻节点的相邻节点。
public void show(TreeNode head){
if(head = =null){
return;
}
Queue MyQueue = new LinkedList<>()
MyQueue.add(head);
while(!MyQueue.isEmpty()){
TreeNode node = MyQueue.poll();
System.out.println(node.val);
if(node.leftNode!=null){
MyQueue.add(node.leftNode);
}
if(node.rightNode!=null){
MyQueue.add(node.rightNode);
}
}
}