算法提高之LeetCode刷题数据结构和算法分析笔试&&面试经验

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

2019-06-08  本文已影响0人  yang_zcybb
//中序遍历递归
    public List<Integer> inorderTraversal_1(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        LinkedList<Integer> ans = new LinkedList<>();
        subInorderTraversal(root, ans);
        return ans;
    }
    private void subInorderTraversal(TreeNode root, List<Integer> 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<Integer> inorderTraversal_2(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        LinkedList<Integer> 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;
    }
上一篇 下一篇

猜你喜欢

热点阅读