【随想】每日两题Day.21

发布时间:2023年12月20日

目录

1.二叉树的递归遍历

前序:

后序:

中序:

思路:

2.二叉树的迭代遍历

前序:

思路:

后序:

思路:

中序:

?思路:


1.二叉树的递归遍历

前序:

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new LinkedList<>();
        preorder(root,list);
        return list;
    }
    public void preorder(TreeNode root,List<Integer> list) {
        if(root == null) return;
        list.add(root.val);
        preorder(root.left,list);
        preorder(root.right,list);
    }
}

后序:

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new LinkedList<>();
        postorder(root,list);
        return list;
    }
    public 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> inorderTraversal(TreeNode root) {
        List<Integer> list = new LinkedList<>();
        inorder(root,list);
        return list;
    }
    public void inorder(TreeNode root,List<Integer> list) {
        if(root == null) return;
        inorder(root.left,list);
        list.add(root.val);
        inorder(root.right,list);
    }
}
思路:

递归实现二叉树遍历还是很简单的。

2.二叉树的迭代遍历

前序:

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new LinkedList<>();
        Stack<TreeNode> stack = new Stack<>();
        if(root != null) stack.push(root);
        while(!stack.empty()) {
            TreeNode node = stack.pop();
            list.add(node.val);
            if(node.right != null) stack.push(node.right);
            if(node.left != null) stack.push(node.left);
        }
        return list;
    }
}
思路:

我们利用入栈结点,然后出栈时将数值存入结果数组,然后要入栈此节点的孩子结点。由于前序遍历:根 左 右?

由于栈是先进后出,所以我们要先入栈右结点,后入栈左节点

后序:

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new LinkedList<>();
        Stack<TreeNode> stack = new Stack<>();
        if(root != null) stack.push(root);
        while(!stack.empty()) {
            TreeNode node = stack.pop();
            list.add(node.val);
            if(node.left != null) stack.push(node.left);
            if(node.right != null) stack.push(node.right);
        }
        Collections.reverse(list);
        return list;
    }
}
思路:

根据前序遍历的特点:根 左 右?

? ? ? ?后序遍历的特点: 左 右 根

所以将左 右 入栈的顺序颠倒后,再将结果数组reverse,就完成了后序遍历的迭代

中序:

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null){
            return result;
        }
        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();
                result.add(cur.val);
                cur = cur.right;
            }
        }
        return result;
    }
}
?思路:

中序遍历的迭代与前序后序不同,因为他是左 根 右 ,所以不是遍历到就存储的。

我们用栈存储我们遍历过的结点,用来我们回溯,当cur一直向左直到为null时,我们回溯

将cur指向出栈的结点,然后加入到结果数组,再将cur指向cur.right

直到cur和stack都为空时跳出循环。

文章来源:https://blog.csdn.net/weixin_63895720/article/details/135103213
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。