中序遍历就是先遍历左再遍历根,最后遍历右 左根右
递归实现:
public List inorderTraversal(TreeNode root) {
List list=new ArrayList<>();
inorder(root,list);
return list;
}
public void inorder(TreeNode root,List list){
if(root==null){
return ;
}
inorder(root.left,list);
list.add(root.val);
inorder(root.right,list);
}
迭代实现
利用栈来实现,二叉树的中序遍历,由于中序遍历是左根右,先定义节点找到最左节点入栈,之后出栈,判断该节点是否有右孩子
public List inorderTraversal(TreeNode root) {
//迭代实现
List list =new LinkedList<>();
Stack stack=new Stack<>();
TreeNode cur=root;
while(cur!=null||!stack.isEmpty()){
//先找到左节点
while(cur!=null){
stack.push(cur);
cur=cur.left;
}
TreeNode node=stack.pop();
list.add(node.val);
if(node.right!=null){
cur=node.right;
}
}
return list;
}
================================================================
《一线大厂Java面试题解析+后端开发学习笔记+最新架构讲解视频+实战项目源码讲义》
【docs.qq.com/doc/DSmxTbFJ1cmN1R2dB】 完整内容开源分享
=================
后序遍历就是左右根
递归实现:
public List postorderTraversal(TreeNode root) {
List list=new ArrayList<>();
postorder(root,list);
return list;
}
public void postorder(TreeNode root,List list){
if(root==null){
return ;
}
postorder(root.left,list);
postorder(root.right,list);
list.add(root.val);
}
非递归实现:
用两个栈来实现二叉树的后序遍历
第一个栈先放入根节点,之后弹出放入第二个节点,之后第一个栈放入左右孩子,之后再弹出放入第二个栈,即可实现
public List postorderTraversal(TreeNode root) {
//利用双栈实现后序遍历
Stack s1=new Stack<>();
Stack s2=new Stack<>();
List list=new LinkedList<>();
if(root==null) return list;
s1.push(root);
while(!s1.isEmpty()){
TreeNode node=s1.pop();
s2.push(node);
if(node.left!=null) s1.push(node.left);
if(node.right!=null) s1.push(node.right);
}
while(!s2.isEmpty()){
TreeNode cur=s2.pop();
list.add(cur.val);
}
return list;
}
=========================================================================
用队列实现层序遍历
public List<List> levelOrder(TreeNode root) {
//用队列实现层序遍历
//第一层节点先进队列,出的节点带下一层的节点
List <List> list=new ArrayList<>();
if(root==null) return list;
Queue queue=new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
List list1=new ArrayList<>();
int size=queue.size();
while(size!=0){
TreeNode cur=queue.poll();
list1.add(cur.val);
if(cur.left!=null){
queue.offer(cur.left);
}
if(cur.right!=null){
queue.offer(cur.right);
}
size–;
}
list.add(list1);
}
return list;
}