中序遍历递归和非递归-Java-LeetCode94

//中序遍历递归
    public List inorderTraversal_1(TreeNode root) {
        Stack stack = new Stack<>();
        LinkedList ans = new LinkedList<>();
        subInorderTraversal(root, ans);
        return ans;
    }
    private void subInorderTraversal(TreeNode root, List list){
        if(root == null){
            return;
        }
        subInorderTraversal(root.left, list);
        list.add(root.val);
        subInorderTraversal(root.right, list);
    }

//中序遍历非递归,判断条件为当前节点和栈至少有一个不为空:
//当cur不为空时,其实表明cur还有left节点(如果cur是叶子,left就是null,在下一循环cur就是null了),那么就一直将cur入栈,并再次访问cur的left;
//直到cur为空,表明此时已经访问到最左节点了,此时弹出stack顶部,加入ans队列,并将顶部的right赋值给cur,开始下一次循环(这样下一次就会循环入栈right的所有left);
//直到都为空,树遍历完成。
    public List inorderTraversal_2(TreeNode root) {
        Stack stack = new Stack<>();
        LinkedList ans = new LinkedList<>();
        TreeNode cur = root;
        while(cur != null || !stack.empty()){
            if(cur != null){
                stack.push(cur);
                cur = cur.left;
            }else{
                cur = stack.pop();
                ans.add(cur.val);
                cur = cur.right;
            }
        }
        return ans;
    }

你可能感兴趣的:(中序遍历递归和非递归-Java-LeetCode94)