acwing 49. 二叉搜索树与双向链表

2021-02-06 22:15

阅读:692

标签:解法   image   content   递归   注意   inf   tin   不能   def   

地址:https://www.acwing.com/problem/content/87/

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。

要求不能创建任何新的结点,只能调整树中结点指针的指向。

注意:

  • 需要返回双向链表最左侧的节点。

例如,输入下图中左边的二叉搜索树,则输出右边的排序双向链表。

技术图片

 

解法

树的处理 一半都是递归  分为 根  树的左子树 和树的右子树

子树也是一棵树 进行递归处理  向上返回一个双链表  返回链表的头尾

最后全部转化链表

技术图片

代码

/**
 * 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* rethead = NULL;

    TreeNode* gleft = NULL;
    TreeNode* gright = NULL;
    
    
    void convertInner(TreeNode* root)
    {
        if (NULL == root) return;
    
        if (root->val val) rethead = root;
    
        if (root->left == NULL && root->right == NULL) {
            gleft = root; gright = root;
            return;
        }
        else if (root->left != NULL && root->right == NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
            gright = root;
        }
        else if (root->right != NULL && root->left == NULL) {
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
            gleft = root;
        }
        else if (root->right != NULL && root->left != NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
            
            TreeNode* leftcopy = gleft;
    
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
            gleft = leftcopy;
        }
    }
    
    
    
    TreeNode* convert(TreeNode* root) {
        if (NULL == root) return NULL;
        rethead = root;
    
        if (root->left == NULL && root->right == NULL) return root;
    
        if (root->left != NULL && root->right == NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
        }
        else if (root->right != NULL && root->left == NULL) {
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
        }
        else if (root->right != NULL && root->left != NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
    
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
        }
    
        return rethead;
    }
    
};
/**
 * 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* rethead = NULL;

    TreeNode* gleft = NULL;
    TreeNode* gright = NULL;
    
    
    void convertInner(TreeNode* root)
    {
        if (NULL == root) return;
    
        if (root->val val) rethead = root;
    
        if (root->left == NULL && root->right == NULL) {
            gleft = root; gright = root;
            return;
        }
        else if (root->left != NULL && root->right == NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
            gright = root;
        }
        else if (root->right != NULL && root->left == NULL) {
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
            gleft = root;
        }
        else if (root->right != NULL && root->left != NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
            
            TreeNode* leftcopy = gleft;
    
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
            gleft = leftcopy;
        }
    }
    
    
    
    TreeNode* convert(TreeNode* root) {
        if (NULL == root) return NULL;
        rethead = root;
    
        if (root->left == NULL && root->right == NULL) return root;
    
        if (root->left != NULL && root->right == NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
        }
        else if (root->right != NULL && root->left == NULL) {
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
        }
        else if (root->right != NULL && root->left != NULL) {
            convertInner(root->left);
            gright->right = root;
            root->left = gright;
    
            convertInner(root->right);
            gleft->left = root;
            root->right = gleft;
        }
    
        return rethead;
    }
    
};

 

acwing 49. 二叉搜索树与双向链表

标签:解法   image   content   递归   注意   inf   tin   不能   def   

原文地址:https://www.cnblogs.com/itdef/p/11407420.html


评论


亲,登录后才可以留言!