java——前中后序遍歷

leetcode有相關(guān)的題目

轉(zhuǎn)一個我覺得寫的很清楚的題解:
https://leetcode-cn.com/problems/binary-tree-preorder-traversal/solution/leetcodesuan-fa-xiu-lian-dong-hua-yan-shi-xbian-2/


import java.util.ArrayList;
import java.util.List;
import java.util.Stack;


public class Search {
    private List<Integer> list;   //存儲最后的結(jié)果
//前序 遞歸算法
    public List<Integer> preorderTraversal1(TreeNode root) {
        list = new ArrayList<>();
        preVisit(root);
        return list;
    }
    public void preVisit(TreeNode root){
        if(root == null) return;
        else{
            list.add(root.val);
            preVisit(root.left);
            preVisit(root.right);
        }
    }
//前序  迭代算法
    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.empty()){
            TreeNode pop = stack.pop();
            res.add(pop.val);
            if(pop.right !=null)
                stack.push(pop.right);
            if(pop.left !=null)
                stack.push(pop.left);
        }
        return res;
    }

//中序  遞歸算法

    public List<Integer> inorderTraversal1(TreeNode root) {
        list = new ArrayList<>();
        inorderVisit(root);
        return list;
    }

    public void inorderVisit(TreeNode root){
        if(root == null) return;
        else{
            inorderVisit(root.left);
            list.add(root.val);
            inorderVisit(root.right);

        }
    }

//中序  迭代算法
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;
        while(cur!=null || !stack.empty()){
            while (cur!=null){
                stack.push(cur);
                cur = cur.left;
            }
            TreeNode pop = stack.pop();
            res.add(pop.val);
            if(pop.right!=null){
                cur=pop.right;
            }
        }
        return res;
    }

//后序   遞歸算法
    public List<Integer> postorderTraversal1(TreeNode root) {
        list = new ArrayList<>();
        postVisit(root);
        return list;
    }
    public void postVisit(TreeNode root){
        if(root == null) return;
        else{
            postVisit(root.left);
            postVisit(root.right);
            list.add(root.val);
        }
    }
//后序  迭代算法
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res  = new ArrayList<>();
        if(root == null )return res;
        Stack<TreeNode> stack = new Stack<>();
        TreeNode reNode = root;
        stack.push(root);
        while(!stack.empty()){
            TreeNode cur = stack.peek();
            if(cur.left!=reNode && cur.right!=reNode && cur.left!=null){
                stack.push(cur.left);
            }
            else if(cur.right!=reNode && cur.right!=null){
                stack.push(cur.right);
            }
            else{
                res.add(cur.val);
                stack.pop();
                reNode=cur;
            }
        }
        return res;
    }



}

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容