1382.将二叉搜索树变平衡

思路：先用中序遍历一遍，由于二叉搜索树的特殊性，左根右得到的便是一个升序数组，所以不用排序即可直接递归加二分查找来创建树。这个递归记录二叉树结点的方法很nice，用一个记录数组下标的量int* pos，记录到数组int arr[]里，中值arr[mid]作为每一次的根。

题目：

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

如果一棵二叉搜索树中，每个节点的两棵子树高度差不超过 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] 也是一个可行的构造方案。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
 /*如果一棵二叉搜索树中，每个节点的两棵子树高度不超过1，我们就称这棵二叉搜索树时平衡的*/
void inOrderTraverse(struct TreeNode* root,int* pos,int arr[]){
    if(root==NULL)return;
    inOrderTraverse(root->left,pos,arr);
    arr[(*pos)++]=root->val;
    inOrderTraverse(root->right,pos,arr);
}//中序遍历存入数组
struct TreeNode* create(int *nums,int low,int high){
    if(low>high)return NULL;//递归出口
    int mid=(low+high)/2;
    /*创造一个树节点并赋值*/
    struct TreeNode* t=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    t->val=nums[mid];
    t->left=create(nums,low,mid-1);
    t->right=create(nums,mid+1,high);
    return t;
}
struct TreeNode* balanceBST(struct TreeNode* root) {
    int arr[10000];
    int *pos=(int*)malloc(sizeof(int));
    *pos=0;
    inOrderTraverse(root,pos,arr);
    return create(arr,0,*pos-1);
}

 对比我的版本，我的是创建了一个完全二叉平衡搜索树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
struct TreeNode* createAVL(struct TreeNode* T,int n,int stor[]){//递归创建平衡二叉搜索树，靠左，完全二叉树
    if(n==0){
        return NULL;
    }
    int st2=0,i=0;
    while(pow(2,i)-1<n){
        i++;
    }
    if(i<=1) st2=0;
    else if((n-(pow(2,i-1)-1))<=(pow(2,i)-pow(2,i-1))/2) st2=pow(2,i-2)-1;//判断右子树的结点个数
    else st2=pow(2,i-2)-1+(n-(pow(2,i-1)-1)-(pow(2,i)-pow(2,i-1))/2);//右结点的个数=所有的结点数n减去最后一层以上再减去（第i层结点个数/2）
    T=(struct TreeNode*)malloc(sizeof(struct TreeNode));
    T->val=stor[n-st2-1];//根
    printf("   %d %d %d   ",T->val,i,st2);
    T->right=createAVL(T->right,st2,stor+(n-st2));//创建右子树
    T->left=createAVL(T->left,n-st2-1,stor);//创建左子树
    return T;
}
struct TreeNode* balanceBST(struct TreeNode* root) {
    struct TreeNode* stack[99999];
    int top=-1,stor[99999],s=0,i,j,t;
    struct TreeNode* T=root;
    while(T||top!=-1){
        while(T){
            stack[++top]=T;
            T=T->left;
        }
        T=stack[top--];
        stor[s++]=T->val;
        printf("%d ",T->val);
        T=T->right;
    }
    T=NULL;
    T=createAVL(T,s,stor);
    printf(" %d",T->val);
    return T;
}