刷题-力扣-897. 递增顺序搜索树

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

题目分析

1. 根据题目描述将一棵二叉搜索树排列为一棵递增顺序搜索树
2. 线中序遍历二叉搜索树，使用递归遍历，将二叉搜索书中的结点存储到vector中
3. 顺序遍历vector中的结点的左子树设为nullptr，右子树设为vector的下一个
4. 将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);
    }
};