leetcode第一刷_ Flatten Binary Tree to Linked List

Posted 2020-09-10 gccbuaa

提示中说明了，改动后的链表相当于原树的前序遍历结果。前序遍历是根左右，因为要把转换后的左子树链接到根节点的右子树上，因此进入递归之后要先把节点的右子树保存下来，然后进入左子树，左子树转换后应该返回最后一个訪问的节点。这个节点的后继是根节点的转换后右子树。说起来很绕，可能看代码反而好一些。

注意一个问题是，不管如何。转换后的根节点一定要把左子树置空。要么会报错的。

TreeNode* preOrder(TreeNode *root){
    if(!root||(!root->left&&!root->right))
        return root;
    TreeNode *pre=NULL, *next;
    next = root->right;
    if(root->left){
        root->right = root->left;
        pre = preOrder(root->left);
    }
    root->left = NULL;
    if(!next)
        return pre;
    if(pre)
        pre->right = next;
    return preOrder(next);
    
}
class Solution {
public:
    void flatten(TreeNode *root) {
        preOrder(root);
        return;
    }
};