Java 二叉树遍历
小玄ks 人气:0二叉树的前中后序遍历
核心思想:用栈来实现对节点的存储。一边遍历,一边将节点入栈,在需要时将节点从栈中取出来并遍历该节点的左子树或者右子树,重复上述过程,当栈为空时,遍历完成。
前序遍历
//非递归 //根 左 右 class Solution { public List<Integer> preorderTraversal(TreeNode root) { //用数组来存储前序遍历结果 List<Integer> list = new ArrayList<>(); if(root==null) return list; //创建一个栈 Stack<TreeNode> st = new Stack<>(); //cur指向根节点 TreeNode cur = root; while(!st.isEmpty()||cur!=null){ //一边向左遍历,一边将遍历到的节点入栈,节点值入数组 while(cur!=null){ list.add(cur.val); st.push(cur); //根 cur=cur.left; //左 } //指针指向栈顶节点(即上一个节点),并将栈顶节点出栈 cur = st.pop(); //指针指向该节点的右子树,开始下一轮的遍历 cur = cur.right; //右 } return list; } }
中序遍历
//非递归 //左 根 右 class Solution { public List<Integer> inorderTraversal(TreeNode root) { List<Integer> list = new ArrayList<>(); if(root==null) return list; Stack<TreeNode> st = new Stack<>(); TreeNode cur = root; while(!st.isEmpty()||cur!=null){ //一边向左遍历,一边将遍历到的节点入栈 while(cur!=null){ st.push(cur); cur = cur.left; //左 } cur = st.pop(); //存储遍历结果 list.add(cur.val); //根 cur = cur.right; //右 } return list; } }
前序遍历和中序遍历的代码基本相同,但是后序遍历与它们不太一样
后序遍历
//非递归 //左 右 根 class Solution { public List<Integer> postorderTraversal(TreeNode root) { List<Integer> list = new ArrayList<>(); if(root==null) return list; TreeNode cur = root; //关键在于定义一个cur的前驱节点 TreeNode pre = null; Stack<TreeNode> st = new Stack<>(); while(cur!=null||!st.isEmpty()){ //一边向左遍历,一边将遍历到的节点入栈 while(cur!=null){ st.push(cur); cur = cur.left; } cur = st.pop(); //若该节点的右节点为空,或者右节点被遍历过,数组才能存储该节点的数值(也就是我们最后才遍历的根) if(cur.right==null||cur.right==pre){ list.add(cur.val); pre = cur; cur=null; }else{//如果不满足,说明该节点的右节点还没有被遍历过,那么接着向右子节点遍历 st.push(cur); cur=cur.right; } } return list; } }
加载全部内容