[LeetCode 简单 二叉树]1382. 将二叉搜索树变平衡

tech2022-08-31  120

题目描述

给你一棵二叉搜索树,请你返回一棵 平衡后 的二叉搜索树,新生成的树应该与原来的树有着相同的节点值。

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

提示:

树节点的数目在 1 到 10^4 之间。 树节点的值互不相同,且在 1 到 10^5 之间。

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/balance-a-binary-search-tree 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

代码实现

class Solution { public TreeNode balanceBST(TreeNode root) { List<Integer> nodes = new ArrayList<>(); // 构建节点值的数组 getNodes(root,nodes); return inBST(nodes,0,nodes.size()-1); } public void getNodes(TreeNode root, List<Integer> nodes){ if(root != null){ getNodes(root.left, nodes); nodes.add(root.val); getNodes(root.right, nodes); } } public TreeNode inBST(List<Integer> nodes,int left, int right){ if(left>right) return null; int mid = left + (right - left)/2; // 中间的位置选为根结点 TreeNode root = new TreeNode(nodes.get(mid)); // 递归产生左子树 root.left = inBST(nodes,left, mid - 1); // 进行右子树的递归构建 root.right = inBST(nodes,mid+1,right); return root; } }

问题出现在 最后选二叉树的根结点和108题有点不一样 但应该都对

最新回复(0)