其他分享
首页 > 其他分享> > LeetCode-897-递增顺序搜索树

LeetCode-897-递增顺序搜索树

作者:互联网

LeetCode-897-递增顺序搜索树

题目

给你一棵二叉搜索树,请你 按中序遍历 将其重新排列为一棵递增顺序搜索树,使树中最左边的节点成为树的根节点,并且每个节点没有左子节点,只有一个右子节点。

示例 1:

输入:root = [5,3,6,2,4,null,8,1,null,null,null,7,9]
输出:[1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]
示例 2:

输入:root = [5,1,7]
输出:[1,null,5,null,7]

提示:

树中节点数的取值范围是 [1, 100]
0 <= Node.val <= 1000

思路

如果按照题意,直接中序遍历构造一个新的二叉树,思路上相对简单,但是时间和空间复杂度较高

因此可以在中序遍历二叉树时,改变此二叉树的形状,使其符合题意。

根据观察,需求的二叉树左子树为空,右子树大于根节点,而BST的特点是左子树小于根节点,右子树大于根节点

因此可以将BST中每个节点的左子树变为父节点,然后左子树置为空即可,代码如下所示。

最后的结果时间0ms,空间6M。感觉只用了两个指针变量,不知道为啥会消耗这么多空间,可能算上了二叉树的大小。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
struct TreeNode* increasingBST(struct TreeNode* root){
    if (!root) return NULL;
    struct TreeNode* leftRootNode = root;
    if (root->left) {
        // 将左子树变成父节点
        leftRootNode = increasingBST(root->left);
        struct TreeNode* leftRightNode = root->left;
        while (leftRightNode->right) leftRightNode = leftRightNode->right;
        //printf("%d %d\n", leftRightNode->val, node->val);
        leftRightNode->right = root;
        root->left = NULL;
    }
    if (root->right) {
        // 改变右子树为右子树的返回值
        root->right = increasingBST(root->right);
    }
    // 若节点没有左子树,返回值为当前节点,若有,返回值为左子树的根节点
    return leftRootNode;
}

标签:左子,null,struct,897,节点,right,顺序搜索,root,LeetCode
来源: https://www.cnblogs.com/sakurapiggy/p/14702099.html