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