欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

leetcode 897. Increasing Order Search Tree(增序搜索树)

程序员文章站 2022-03-16 08:39:49
...

题目要求

给定一棵树,按顺序重新排列树,以便树中最左边的节点现在是树的根,并且每个节点都没有左子,只有一个右子。如下图:
leetcode 897. Increasing Order Search Tree(增序搜索树)
leetcode 897. Increasing Order Search Tree(增序搜索树)

解题思路

因为最后的输出要按照从低到高的顺序,而给出的BST按照先序遍历正好是增序排列,所以我们解题的思路就是按照先序遍历得到节点的值,然后根据节点的值建立新树,最后输出即可。

注意:左节点是NULL!

主要代码c++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int>result;
    TreeNode* increasingBST(TreeNode* root) {
        inorder(root);
        TreeNode *head = new TreeNode(0);
        TreeNode *p = head;
        for(int i = 0; i < result.size(); ++i, p=p->right) // 建立新树
            p->right = new TreeNode(result[i]);
        return head->right;
    }
    void inorder(TreeNode* root){ // 先序遍历
        if(root==NULL) return;
        inorder(root->left);
        result.push_back(root->val);
        inorder(root->right);
    }
};
相关标签: 前序遍历