题目
给你一棵二叉搜索树,请你返回一棵 平衡后 的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。如果有多种构造方法,请你返回任意一种。
如果一棵二叉搜索树中,每个节点的两棵子树高度差不超过 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]
解题
方法一:中序遍历转数组+重建平衡二叉树
class Solution { public: vector<int> arr; void traversal(TreeNode* cur){ if(!cur) return; traversal(cur->left); arr.push_back(cur->val); traversal(cur->right); } TreeNode* getTree(int l,int r){ if(l>r) return nullptr; int mid=(l+r)/2; TreeNode* cur=new TreeNode(arr[mid]); cur->left=getTree(l,mid-1); cur->right=getTree(mid+1,r); return cur; } TreeNode* balanceBST(TreeNode* root) { traversal(root); return getTree(0,arr.size()-1); } };