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

知识点&算法

中序遍历得到数据有序的结点，存入顺序的数据结构，然后遍历数据结构，改变接结点指向。
或者以一全局变量记录前驱结点，直接在中序遍历的过程中进行指针修改。将前驱作为参数传递是不可行的，因为作为全局变量的情况下，前驱结点的修改是在遍历到某一结点时进行的，因此前驱节点的更新一定是按照中序遍历的顺序的；而作为参数传入的情况下，
前驱节点一定是当前节点的双亲节点，但不一定是中序遍历的前驱节点。

**注意：**估计是判题机在检查答案的时候会检查节点的左子孩子是否为空，所以在修改指针的过程中一定要将结点的左子孩子置空，不然有可能会出现互为孩子的情况。右子节点会在后续修改所以无所谓。

题解

顺序数据结构：

class Solution {
public:
    vector<TreeNode* > nodes;
    void traverse(TreeNode* root){
        if(root == nullptr) return;
        traverse(root->left);
        nodes.push_back(root);
        traverse(root->right);
        return;
    }

    TreeNode* increasingBST(TreeNode* root) {
        traverse(root);
        for(int i = 1 ; i < nodes.size() ; ++i){
			//左孩子置空
            nodes[i]->left = nullptr;
            nodes[i-1]->right = nodes[i];
        }
        nodes[0]->left = nullptr;
        return nodes[0];
    }
};

中序遍历直接修改

class Solution {
public:
    TreeNode* pre;
    void traverse(TreeNode* root){
        if(root == nullptr) return;
        traverse(root->left);
		//左孩子置空
        pre->left = nullptr;
        root->left = nullptr;
        pre->right = root;
        pre = root;
        traverse(root->right);
        return;
    }

    TreeNode* increasingBST(TreeNode* root) {
        TreeNode* res = new TreeNode(-1);
        pre = res;
        traverse(root);
        return res->right;
    }
};