二叉树展开为链表(前序遍历逆序实现)

题目

给定一个二叉树,原地将它展开为一个单链表。
二叉树展开为链表(前序遍历逆序实现)_第1张图片

注意点

1、我们发现展开后的链表的顺序为二叉树前序遍历的顺序(根节点->左节点->右节点,1->2->3->4->5->6);
2、但我们会发现使用前序遍历的方式进行原地展开时,会丢失右子树(比如遍历到节点2时,需要将节点2作为根节点的右节点,这时会导致右子树丢失);
3、所以我们考虑使用前序遍历逆序的方式实现,这样就不会丢失右子树(右节点->左节点->根节点,6<-5<-4<-3<-2<-1);
4、需要记录前节点,比如遍历到5的时候,需要将5指向6。

实现

/**
 * Definition for a binary tree node.
 * 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;
 *     }
 * }
 */
class Solution {
	// 前节点,初始为空
    private TreeNode pre = null; 
    public void flatten(TreeNode root) {
        if (root == null) {
            return;
        }
		// 前序遍历的逆序
        revPreorderTraversal(root);
    }

    public void revPreorderTraversal(TreeNode root) {
        if (root == null) {
            return;
        }
		
		// 操作右子树
        revPreorderTraversal(root.right);
        // 操作左子树
        revPreorderTraversal(root.left);
        
        // 当前节点的右节点指向前节点
        root.right = pre;
        // 当前节点的左节点置空
        root.left = null;
        // 更新前节点为当前节点
        pre = root;
    }
}

你可能感兴趣的:(leetcode记录,二叉树,leetcode)