1382. 将二叉搜索树变平衡-常规方法

1382. 将二叉搜索树变平衡

给你一棵二叉搜索树，请你返回一棵 平衡后 的二叉搜索树，新生成的树应该与原来的树有着相同的节点值。如果有多种构造方法，请你返回任意一种。

如果一棵二叉搜索树中，每个节点的两棵子树高度差不超过 1 ，我们就称这棵二叉搜索树是 平衡的 。

示例 1：

输入：root = [1,null,2,null,3,null,4,null,null]
输出：[2,1,3,null,null,null,4]
解释：这不是唯一的正确答案，[3,1,4,null,2,null,null] 也是一个可行的构造方案。

示例 2：

输入: root = [2,1,3]
输出: [2,1,3]

解题代码如下：

/** * Definition for a binary tree node. * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * }; */
 int len;
 void or_search(struct TreeNode* root){
    
     if(root){
    
          or_search(root->left);
           len++;

         or_search(root->right);
          
    
     }
    
 }
 void or_search2(struct TreeNode* root,int *a){
    
     if(root){
    
          or_search2(root->left,a);
          a[len++]=root->val;

         or_search2(root->right,a);
     }
 }
  void creat_root(struct TreeNode** root,int *a,int low,int high){
    
      if(low<=high){
    
      int mid=(low+high)/2;
      (*root)->val=a[mid];
      (*root)->left=(struct TreeNode*)malloc(sizeof(struct TreeNode));
       (*root)->right=(struct TreeNode*)malloc(sizeof(struct TreeNode));
       creat_root(&((*root)->left),a,low,mid-1);
       creat_root(&((*root)->right),a,mid+1,high);
        }
        else{
    
            (*root)=NULL;
        }


  }


struct TreeNode* balanceBST(struct TreeNode* root){
    
    len=0;
    or_search(root);
   struct TreeNode* re=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    int *a=(int *)malloc(sizeof(int)*len);
    len=0;
     or_search2(root,a);
     printf("len %d ",len);
     creat_root(&re,a,0,len-1);
     return re;


}