二叉树的遍历

面试中经常会考的一道题目,就是二叉树的遍历,很简单,你会说“使用递归,根据要求(前序、中序、后序,以中序为例)依次对节点和其左右子节点调用递归方法。

public void printNode(Node node) {
    if (null != node.left)
        printNode(node.lef);
    System.out.print(node.toString());
    if(null != node.right)
        printNode(node.right);
    }

一般还会继续追问"能否使用非递归的方法?",在美团点评面试的时候就被问到了(同LeetCode 144)。这时候,我们可以使用栈来完成,分为下面三个步骤:

  1. 深度遍历找到这棵树最左边的节点,这个过程中将遍历的节点压入栈中。
  2. 将栈中的节点pop出来,如果该节点有右子节点,对右子节点进行步骤一的操作,如果没有,则可以打印出来,继续pop栈。
  3. 当栈空了或者进行操作的根节点为空时推出循环。
    下面就用代码来实现一下LeetCode 144
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List preorderTraversal(TreeNode root) {
       ArrayList result = new ArrayList<>();
        if (null == root) {
            return null;
        }
        Stack nodeStack = new Stack<>();

        TreeNode temp = root;

        while (null != temp || !nodeStack.empty()) {
            nodeStack.push(temp);
            result.add(temp.val);
            while (null != temp.left) {
                temp = temp.left;
                nodeStack.push(temp);
                result.add(temp.val);
            }
            while (!nodeStack.empty() && null == nodeStack.peek().right) {
                nodeStack.pop();
            }
            if (nodeStack.empty())
                break;
            temp = nodeStack.pop();
            if (null != temp.right) {
                temp = temp.right;
                continue;
            }
        }
        return result; 
    }
}

你可能感兴趣的:(二叉树的遍历)