二叉树深度优先搜索（DFS）- Java

public class DFSMain {

    public static void main(String[] args) {
		// 创建二叉树
        TreeNode n11 = new TreeNode(11);
        TreeNode n10 = new TreeNode(10);
        TreeNode n9 = new TreeNode(9);
        TreeNode n8 = new TreeNode(8);
        TreeNode n7 = new TreeNode(7, n11, null);
        TreeNode n6 = new TreeNode(6, n9, n10);
        TreeNode n5 = new TreeNode(5);
        TreeNode n4 = new TreeNode(4, null, n8);
        TreeNode n3 = new TreeNode(3, n6, n7);
        TreeNode n2 = new TreeNode(2, n4, n5);
        TreeNode n1 = new TreeNode(1, n2, n3);

        System.out.println(recursive(n1));  // [1, 2, 4, 8, 5, 3, 6, 9, 10, 7, 11]
        System.out.println(unRecursive(n1));    // [1, 2, 4, 8, 5, 3, 6, 9, 10, 7, 11]
    }


    /**
     * 递归
     */
    public static List<Integer> recursive(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        dfs(root, list);
        return list;
    }

    public static void dfs(TreeNode root, List<Integer> list) {
        if (root == null) {
            return;
        }

        list.add(root.val);
        // 左递归
        dfs(root.left, list);
        // 右递归
        dfs(root.right, list);
    }

    /**
     * 非递归
     */
    public static List<Integer> unRecursive(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();

        if (root == null) {
            return list;
        }

        stack.push(root);

        while (!stack.isEmpty()) {
            // 弹栈
            TreeNode node = stack.pop();
            // 右子树先入栈
            if (node.right != null) {
                stack.push(node.right);
            }
            // 左子树后入栈
            if (node.left != null) {
                stack.push(node.left);
            }

            list.add(node.val);
        }

        return list;
    }

}

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }

    TreeNode(int val) {
        this.val = val;
    }

    @Override
    public String toString() {
        return "TreeNode{" +
                "val=" + val +
                ", left=" + left +
                ", right=" + right +
                '}';
    }
}