语言
Java
递归遍历
递归说明
递归有三要素:
1.确定递归函数的参数和返回值
2.确定终止条件
3.确定单层递归的逻辑
按照这三点来讲基本不会错。
来三道题目练练手。
题目
思路
讲一道中序遍历,创建一个ArrayList,使用递归函数,递归函数的参数是树的节点和结果。
在方法中先判断终止条件即节点不为空,因为是中序就左中右,先调用递归函数参数是左节点、ArrayList,再添加节点的值到ArrayList中,最后再调用递归函数参数是右节点、ArrayList。
最后返回结果ArrayList。
代码
// 前序遍历·递归·LC144_二叉树的前序遍历
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<Integer>();
preorder(root, result);
return result;
}
public void preorder(TreeNode root, List<Integer> result) {
if (root == null) {
return;
}
result.add(root.val);
preorder(root.left, result);
preorder(root.right, result);
}
}
// 中序遍历·递归·LC94_二叉树的中序遍历
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
inorder(root, res);
return res;
}
void inorder(TreeNode root, List<Integer> list) {
if (root == null) {
return;
}
inorder(root.left, list);
list.add(root.val); // 注意这一句
inorder(root.right, list);
}
}
// 后序遍历·递归·LC145_二叉树的后序遍历
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
postorder(root, res);
return res;
}
void postorder(TreeNode root, List<Integer> list) {
if (root == null) {
return;
}
postorder(root.left, list);
postorder(root.right, list);
list.add(root.val); // 注意这一句
}
}
易错点
前序和后序遍历是一样的思路,更改递归函数的顺序就好了。
易错的就是考虑终止条件和递归参数。
迭代遍历
题目
主要讲讲前序遍历和中序遍历
前序遍历思路
采用栈的方式来模拟,将结果存在数组里。
首先创建一个数组,判断根节点是否为空,为空返回空数组。
再建立一个栈,将根节点添加到数组中,开始遍历终止条件是栈为空。
在循环里:先将根节点弹出,值放入数组中,判断如果右节点不为空放入右节点,再判断左节点,不为空把左节点放到栈中。这时小伙伴可能疑惑了,难道前序不是中左右吗这里为啥先加右节点?
因为栈是先入后出的。最后返回数组。
后序遍历思路
就是把中间的步骤改为先放左节点后放右节点,再把数组翻转。
前序遍历和后序遍历代码
class Solution {//前序遍历
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pop();
res.add(node.val);
if (node.right != null) {
stack.push(node.right);
}
if (node.left != null) {
stack.push(node.left);
}
}
return res;
}
}
class Solution {//后序遍历
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pop();
res.add(node.val);
if (node.left != null) {
stack.push(node.left);
}
if (node.right != null) {
stack.push(node.right);
}
}
Collections.reverse(res);
return res;
}
}
中序遍历思路
中序遍历顺序 左中右 入栈顺序左右
首先创建一个数组,判断根节点是否为空,为空返回空数组。
再建立一个栈,将根节点添加到数组中,开始遍历终止条件是栈为空或节点为空。
判断节点是否为空,若不为空,将节点加入到栈中,节点向左子树遍历。
若为空,先弹出栈中的元素,再把元素的值加入到数组中,节点向右子树遍历。
最后返回结果数组。
中序遍历代码
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Stack<TreeNode> stack = new Stack<>();
TreeNode cur = root;
while (cur != null || !stack.isEmpty()) {
if (cur != null) {
stack.push(cur);
cur = cur.left;
} else {
cur = stack.pop();
res.add(cur.val);
cur = cur.right;
}
}
return res;
}
}
统一迭代
思路
针对三种遍历,使用迭代法可以写出统一风格的代码!
依然采取栈的数据结构,我们将访问的节点放入栈中,把要处理的节点也放入栈中但是要做标记。
就是处理的节点放入栈后,紧接着放入一个空指针作为标记。
代码
中序遍历
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();//创建数组
Stack<TreeNode> stack = new Stack<>();//创建栈
if (root != null) stack.push(root);//判断根节点是否为空
while (!stack.empty()) {//循环遍历二叉树
TreeNode node = stack.peek();
if (node != null) {
stack.pop();//将节点弹出,避免重复操作
if (node.right != null) stack.push(node.right);//添加右节点
stack.push(node);// 添加中节点
stack.push(null);//访问过但没处理,做一个标记
if (node.left != null) stack.push(node.left);
} else {
stack.pop();
node = stack.peek();
stack.pop();
res.add(node.val);
}
}
return res;
}
}
前序遍历
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();//创建数组
Stack<TreeNode> stack = new Stack<>();//创建栈
if (root != null) stack.push(root);//判断根节点是否为空
while (!stack.empty()) {//循环遍历二叉树
TreeNode node = stack.peek();
if (node != null) {
stack.pop();//将节点弹出,避免重复操作
if (node.right != null) stack.push(node.right);//添加右节点
if (node.left != null) stack.push(node.left);//左节点
stack.push(node);// 添加中节点
stack.push(null);//访问过但没处理,做一个标记
} else {
stack.pop();
node = stack.peek();
stack.pop();
res.add(node.val);
}
}
return res;
}
}
后序遍历
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();//创建数组
Stack<TreeNode> stack = new Stack<>();//创建栈
if (root != null) stack.push(root);//判断根节点是否为空
while (!stack.empty()) {//循环遍历二叉树
TreeNode node = stack.peek();
if (node != null) {
stack.pop();//将节点弹出,避免重复操作
stack.push(node);// 添加中节点
stack.push(null);//访问过但没处理,做一个标记
if (node.right != null) stack.push(node.right);//添加右节点
if (node.left != null) stack.push(node.left);//左节点
} else {
stack.pop();
node = stack.peek();
stack.pop();
res.add(node.val);
}
}
return res;
}
}
层序遍历
思想
想到了两种方式,递归和非递归。
一层一层的遍历,将每一层的个数存起来,放在数组中。
最后都放在二维数组中。
代码
递归方式
class Solution {
public List<List<Integer>> resList = new ArrayList<List<Integer>>();//用来保存层序遍历的结果
public List<List<Integer>> levelOrder(TreeNode root) {
check01(root, 0);//调用方法 参数为节点和0
return resList;//返回二维数组
}
public void check01(TreeNode node, Integer deep) {//传入节点 和 深度
if (node == null) return;//判断节点是否为空
deep++;//深度加一
if (resList.size() < deep) {//如果数组的长度小于二叉树深度
List<Integer> item = new ArrayList<Integer>();//创建一个一维数组
resList.add(item);//将一维数组放在二维数组中
}
resList.get(deep - 1).add(node.val);//添加深度- 1节点的值
check01(node.left, deep);//向左子树递归
check01(node.right, deep);//向右子树递归
}
}
迭代方式-借助队列
class Solution {
public List<List<Integer>> resList = new ArrayList<List<Integer>>();//用来保存层序遍历的结果
public List<List<Integer>> levelOrder(TreeNode root) {
check02(root);//调用方法 参数为节点和0
return resList;//返回二维数组
}
public void check02(TreeNode node) {//传入节点
if (node == null) return;//判断节点是否为空
Queue<TreeNode> que = new LinkedList<TreeNode>();//创建队列
que.offer(node);//把节点添加到队列中
while (!que.isEmpty()) {//判断条件为队列不为空
List<Integer> itemList = new ArrayList<Integer>();//创建一维数组
int len = que.size();//获取队列的长度
while (len > 0) {
TreeNode tmpNode = que.poll();//弹出队列中的节点并赋值给临时节点
itemList.add(tmpNode.val);//在一维数组中添加临时节点的值
if (tmpNode.left != null) que.offer(tmpNode.left);//判断 加入左子节点
if (tmpNode.right != null) que.offer(tmpNode.right);//判断 加入右字节点
len--;//循环终止条件 让他遍历一层的数量大小
}
resList.add(itemList);//将一维数组添加到二维数组中。
}
}
}
总结
二叉树这里我感觉非常有意思,而且完成它花费了很长时间。继续努力距离大厂梦又进一步。