897. 递增顺序搜索树
题目链接
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/increasing-order-search-tree/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
题目描述
给你一棵二叉搜索树，请你 按中序遍历 将其重新排列为一棵递增顺序搜索树，使树中最左边的节点成为树的根节点，并且每个节点没有左子节点，只有一个右子节点。
示例 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
题目分析
- 根据题目描述将一棵二叉搜索树排列为一棵递增顺序搜索树
- 线中序遍历二叉搜索树，使用递归遍历，将二叉搜索书中的结点存储到vector中
- 顺序遍历vector中的结点的左子树设为nullptr，右子树设为vector的下一个
- 将vector中最后一个结点的左右子树都设为nullptr
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* increasingBST(TreeNode* root) {
vector<TreeNode*> treeList;
ergodicTree(treeList, root);
int len = treeList.size() - 1;
for (int i = 0; i < len; ++i) {
treeList[i]->left = nullptr;
treeList[i]->right = treeList[i + 1];
}
treeList[len]->left = nullptr;
treeList[len]->right = nullptr;
return root == nullptr ? nullptr : treeList[0];
}
private:
void ergodicTree(vector<TreeNode*>& treeList, TreeNode* root) {
if (root == nullptr) return;
ergodicTree(treeList, root->left);
treeList.push_back(root);
ergodicTree(treeList, root->right);
}
};