代码随想录算法训练营第十四天|二叉树递归遍历|迭代遍历|统一遍历

递归遍历

递归的三要素：

（1）确定递归函数的参数和返回值： 确定哪些参数是递归的过程中需要处理的，那么就在递归函数里加上这个参数， 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。

（2）确定终止条件： 写完了递归算法, 运行的时候，经常会遇到栈溢出的错误，就是没写终止条件或者终止条件写的不对，操作系统也是用一个栈的结构来保存每一层递归的信息，如果递归没有终止，操作系统的内存栈必然就会溢出。

（3）确定单层递归的逻辑： 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。

LeetCode144.二叉树的前序遍历

（1）递归

       基本思路：  二叉树的前序遍历的输出顺序为 根 - 左 - 右，当递归到的当前节点不为空，那么先将当前节点加入结果集中，然后再分别递归当前节点的左右孩子（顺序不能改变）。

Java代码如下：

    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
    	preorder(root,res);
    	return res;
    }
    public void preorder(TreeNode node,List<Integer> res) 
    {
    	if(node == null) 
    	{
    		return;
    	}
    	res.add(node.val);
    	preorder(node.left,res);
    	preorder(node.right,res);
    }

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

		TreeNode() {
		}

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

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

（2）迭代

        基本思路：递归其实就是系统帮忙维护一个栈，我们可以自行维护栈，模拟递归操作。while当遍历到的节点不为空的时候，当前元素入栈并将指加入结果集，然后当前的节点指向左孩子，继续遍历，如果为空，栈顶元素出栈当前节点等于出栈的元素，然后遍历其右孩子。最后返回结果集。

Java代码如下：

public List<Integer> preorderTraversal_1(TreeNode root) {
    	List<Integer> res = new ArrayList<>();
    	if(root == null) 
    	{
    		return res;
    	}
    	Deque<TreeNode> stack = new LinkedList<TreeNode>();
    	TreeNode node = root;
    	while(!stack.isEmpty() || node != null) 
    	{
    		while(node != null) 
    		{
    			res.add(node.val);
    			stack.push(node);
    			node = node.left;
    		}
    		node = stack.pop();
    		node = node.right;
    	}
    	
    	return res;
    }

 LeetCode145.二叉树的后序遍历

（1）递归

        基本思路：二叉树的后序遍历的输出顺序为： 左 - 右 - 根。如果当前递归到的节点不为空，那么继续递归左右孩子，并把结果传入到结果集中，如果为空，直接返回。

Java代码如下：

    public List<Integer> postorderTraversal(TreeNode root) {
    	List<Integer> res = new ArrayList<>();
    	postorder(root,res);
    	return res;
    }
    
    public void postorder(TreeNode root,List<Integer> res) 
    {
    	if(root == null) 
    	{
    		return;
    	}
    	
    	postorder(root.left,res);
    	postorder(root.right,res);
    	res.add(root.val);
    }

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

		TreeNode() {
		}

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

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

 

 

（2）迭代

         基本思路：于先序迭代思路相似，只是先放入左节点，再放入右节点。

Java代码如下：

    public List<Integer> postorderTraversal(TreeNode root) {
    	List<Integer> res = new ArrayList<>();
    	if(root == null) 
    	{
    		return res;
    	}
    	Deque<TreeNode> stack = new LinkedList<>();
    	TreeNode prev = null;
    	while(!stack.isEmpty() || root != null) 
    	{
    		while(root != null) 
    		{
    			stack.push(root);
    			root = root.left;
    		}
    		root = stack.pop();
    		if(root.right == null || root.right == prev) 
    		{
    			res.add(root.val);
    			prev = root;
    			root = null;
    		}else 
    		{
    			stack.push(root);
    			root = root.right;
    		}
    	}
    	return res;
    }

LeetCode94.二叉树的中序遍历

（1）递归

        基本思路：中序遍历的输出顺序为 左 - 根 - 右，基本思路和上述基本相同，只是结果添加的位置改变。

Java代码如下：

public List<Integer> inorderTraversal(TreeNode root) {
    	List<Integer> res = new ArrayList<>();
    	inorder(root,res);
    	return res;
    }
    
    public void inorder(TreeNode root,List<Integer> res) 
    {
    	if(root == null) 
    	{
    		return;
    	}
    	inorder(root.left,res);
    	res.add(root.val);
    	inorder(root.right,res);
    }

 

 

（2）迭代

        基本思路：与上述两种方法基本一致。

Java代码如下：

    public List<Integer> inorderTraversal_1(TreeNode root) {
    	List<Integer> res = new ArrayList<>();
    	Deque<TreeNode> stack = new LinkedList<>();
    	while(root != null || !stack.isEmpty()) 
    	{
    		while(root != null) 
    		{
    			stack.push(root);
    			root = root.left;
    		}
    		root = stack.pop();
    		res.add(root.val);
    		root = root.right;
    	}
    	return res;
    }