力扣0108——将有序数组转换为二叉搜索树

将有序数组转换为二叉搜索树

难度：简单

题目描述

给你一个整数数组 nums ，其中元素已经按 升序 排列，请你将其转换为一棵 高度平衡 二叉搜索树。

高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例1

输入： nums = [-10,-3,0,5,9]
输出：[0,-3,9,-10,null,5]
或者也可以为：

输出：[0,-10,5,null,-3,null,9]

示例2

输入： nums = [1,3]
输出：[3,1]

题解

直接使用回溯法，基本原理就是二分数组，将左边的值按照二分法赋值给左子树，右边的值按照二分法赋值给右子树

想法代码

public class TreeNode
{
    public int val;
    public TreeNode left;
    public TreeNode right;

    public TreeNode(int val = 0, TreeNode left = null, TreeNode right = null)
    {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

class Solution
{
    public static void Main(string[] args)
    {
        int[] nums = { -10, -3, 0, 5, 9 };
        Solution solution = new Solution();
        TreeNode root = solution.SortedArrayToBST(nums);
        IList<int> list = new List<int>();
        solution.AddTree(root, list);
        foreach (int i in list)
        {
            Console.Write(i + " ");
        }
    }

    public TreeNode SortedArrayToBST(int[] nums)
    {
        return BackTrack(nums,0,nums.Length - 1);
    }

    public TreeNode BackTrack(int[] nums, int start, int end)
    {
        if (start > end)
        {
            return null;
        }
        int mid = (end - start) / 2 + start;
        return new TreeNode(nums[mid], BackTrack(nums, start, mid - 1), BackTrack(nums, mid + 1, end));
    }

    public void AddTree(TreeNode root, IList<int> ans)
    {
        if (root == null)
        {
            return;
        }
        AddTree(root.left, ans);
        ans.Add(root.val);
        AddTree(root.right, ans);
    }
}