力扣每日一题：897. 递增顺序搜索树

题目：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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/increasing-order-search-tree/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

  该题将BST转化为只有右结点且顺序排序的二叉树，实际解题很简单，只需要进行一次中序遍历，在中序遍历的过程中将根结点的指向修改即可。

解题代码

（1）递归

/**
 * 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 {
private:
    TreeNode* p;
public:
    TreeNode* increasingBST(TreeNode* root) {
        TreeNode* newHead = new TreeNode(-1, nullptr, root);
        p = newHead;
        inOrder(root);

        return newHead->right;
    }

    void inOrder(TreeNode* T)
    {
        if(T)
        {
            inOrder(T->left);
            visit(T);
            inOrder(T->right);
        }
    }
    void visit(TreeNode* T)
    {
        T->left = nullptr;
        p->right = T;
        p = T;
    }
};

（2）非递归

/**
 * 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 {
private:
    TreeNode* p;
public:
    TreeNode* increasingBST(TreeNode* root) {
        TreeNode* newHead = new TreeNode(-1, nullptr, root);
        p = newHead;
        InOrder(root);

        return newHead->right;
    }

    void visit(TreeNode* T)
    {
        T->left = nullptr;
        p->right = T;
        p = T;
    }

    void InOrder(TreeNode* T)
    {
        if(!T) return;

        stack<TreeNode*> stk;
        TreeNode* temp = T;
        while(!stk.empty() || temp)
        {
            if (temp)
            {
                stk.push(temp);
                temp = temp->left;
            }
            else
            {
                temp = stk.top();
                stk.pop();
                visit(temp);
                temp = temp->right;
            }
        }
    }
};

解题感悟

  解决二叉树中序遍历的问题，实则还是将遍历过程模拟一遍，就可轻松解题。