Leetcode 114. 二叉树展开为链表

先整成左子树的形式【右子结点在先序遍历中递归返回后还要遍历,不能覆盖】

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* p=NULL;
    void preorder(TreeNode* root){
        if(p) p->left=root;
        p=root;
        if(root->left) preorder(root->left);
        if(root->right) preorder(root->right);
    }
    void flatten(TreeNode* root) {
        if(!root) return;
        preorder(root);
        while(root->left) root->right=root->left,root->left=NULL,root=root->right;
    }
};

猜你喜欢

转载自blog.csdn.net/bendaai/article/details/80555109