C++Leetcode897:递增顺序查找树

题目
给定一个树,按中序遍历重新排列树,使树中最左边的结点现在是树的根,并且每个结点没有左子结点,只有一个右子结点。
示例 :
在这里插入图片描述
提示:
给定树中的结点数介于 1 和 100 之间。
每个结点都有一个从 0 到 1000 范围内的唯一整数值。

思路
1、暴力解法。中序遍历二叉树,将值逐个存入一个动态数组;然后遍历数组,建一个新的树。

实现方法
一、暴力解法

/**
 * 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:
    TreeNode* increasingBST(TreeNode* root) {
        vector<int> value;
        TreeNode* newNode=new TreeNode(1);
        TreeNode* p=newNode;
        increasingBST(root,value);
        
        for(int i=0;i<value.size();i++){
            newNode->val=value[i];
            newNode->left=0;
            if(i!=value.size()-1){
                TreeNode* q=new TreeNode(1);
                newNode->right=q;
                newNode=q;
            }
        }
        return p;
    }
    
    void increasingBST(TreeNode* root,vector<int> &value){
        if(root){
            increasingBST(root->left,value);
            value.push_back(root->val);
            increasingBST(root->right,value);  
        }
    }
};

二、简化写法

/**
 * 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:
    TreeNode* increasingBST(TreeNode* root) {
        vector<int> value;
        increasingBST(root,value);
        TreeNode* newRoot=new TreeNode(value[0]);
        TreeNode* p=newRoot;
        
        for(int i=1;i<value.size();i++){
            p->left=0;
            p->right=new TreeNode(value[i]);
            p=p->right;
            }
        return newRoot;
        }  
    
    void increasingBST(TreeNode* root,vector<int> &value){
        if(root){
            increasingBST(root->left,value);
            value.push_back(root->val);
            increasingBST(root->right,value);  
        }
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43434305/article/details/87881333