二叉树遍历方式
- 深度优先:前序、中序、后序
- 广度优先:层序遍历
- 说明:
前序遍历【中、左、右】:先访问中间节点,再访问左孩子,再访问右孩子。左、右孩子分别也是前序
中序遍历【左、中、右】:
后序遍历【左、右、中】
遍历:递归 & 非递归【递归的实现–>栈,所以非递归借助栈实现】
代码实现如下:
1-(1)递归 – 前序
import java.util.ArrayList;
/**
- Definition for a binary tree node.
- public class TreeNode {
- int val;
- TreeNode left;
- TreeNode right;
- TreeNode() {}
- TreeNode(int val) { this.val = val; }
- TreeNode(int val, TreeNode left, TreeNode right) {
- this.val = val;
- this.left = left;
- this.right = right;
- }
- }
*/
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<>();
preOrder(root,result);
return result;
}
void preOrder(TreeNode root,List<Integer> result){
if (root == null) return ;
result.add(root.val);// 中
preOrder(root.left,result);//左
preOrder(root.right,result);// 右
}
}
1-(2)非递归-前序
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
//非递归实现
Stack<TreeNode> st = new Stack<>();
List<Integer> res = new ArrayList<>();
if (root == null) return res;
st.push(root);
while (!st.empty()){
//中
TreeNode node = st.peek();
st.pop();
res.add(node.val);
if (node.right != null) st.push(node.right); // 右
if (node.left != null) st.push(node.left);//左
}
return res;
}
}
说明:
-
递归的实现中,前、中、后有统一的编码方式,只需要将递归函数中中、左、右逻辑调换顺序即可
-
非递归:借用栈实现
1)前序:中、左、右。所以访问到一个节点,将其入栈【先进后出】,后处理节点【加入结果集合中】
访问顺序:由于栈的先进后出,所以是先访问右孩子,入栈、处理;再访问左孩子、入栈、处理;
2)后序:左右中
【前序】中左右 --(调换左、右顺序)–>中右左–>反转操作–>左右中【即后序】
2-(1)递归–后序
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
//1、递归算法
List<Integer> result = new ArrayList<>();
postorder(root,result);
return result;
}
void postorder(TreeNode root,List<Integer> result){
//递归终止条件
if (root == null) return;
postorder(root.left,result);
postorder(root.right,result);
result.add(root.val);
}
}
2-(2)非递归-后序
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
// 非递归- 后序【左、右、中】
//前序:中左右-->中右左-(反转)->左右中
//数组反转的实现:双指针、库函数中的reverse
Stack<TreeNode> st = new Stack<>();
List<Integer> res = new ArrayList<>();
if (root == null) return res;
st.push(root);
while (!st.empty()){
//中
TreeNode node = st.peek();
st.pop();
res.add(node.val);
if (node.left != null) st.push(node.left);//左
if (node.right != null) st.push(node.right); // 右
}
//res进行反转
Collections.reverse(res);
return res;
}
}
3-(1) 递归-中序
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<>();
inorder(root,result);
return result;
}
void inorder(TreeNode root,List<Integer> result){
if(root == null) return;
inorder(root.left,result);//左
result.add(root.val); //中
inorder(root.right,result); //右
}
}
3-(2)非递归-中序
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
// 非递归中序遍历:左中右
//访问顺序,与处理顺序不一致
//栈:记录访问过的节点
// 指针:遍历二叉树
List<Integer> result = new ArrayList<>();
Stack<TreeNode> st = new Stack<>();
TreeNode cur = root;// 指针:遍历
while (cur != null || !st.empty()){ //cur==null&& st.empty()使停止循环,注意在while中是||
if (cur != null){
st.push(cur);
cur = cur.left;
}else {
cur = st.peek();
st.pop();
result.add(cur.val);
cur = cur.right;
}
}
return result;
}
}