Convert Sorted List to Binary Search Tree 将有序链表转化为平衡二叉排序树

最新推荐文章于 2020-03-17 10:53:41 发布

原创最新推荐文章于 2020-03-17 10:53:41 发布 · 659 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

119 篇文章

订阅专栏

本文介绍一种方法，将给定的有序链表转换为平衡二叉搜索树。通过先将链表转换为数组，再利用数组构建平衡二叉树的方式实现。此方法保证了树的高度尽可能小。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

链接

解答：

将上一篇中的将有序数组转化为平衡二叉树改一下，先把链表存成数组。

代码：

  class Solution {
  public:
	  TreeNode *sortedListToBST(ListNode *head) {
		  if (head == NULL)
			  return NULL;
		  vector<int> num;
		  while (head)
		  {
			  num.push_back(head->val);
			  head = head->next;
		  }
		  TreeNode * res = sortedArrayToBST(num);
		  return res;
	  }
  private:
	  TreeNode *sortedArrayToBST(vector<int> &num) {
		  if (num.size() == 0)
			  return NULL;
		  TreeNode *root = new TreeNode(0);    //新建一个节点作为根节点
		  build(root, 0, num.size() - 1, num);
		  return root;
	  }
	  void build(TreeNode *root, int start, int end, vector<int> &num)
	  {
		  int mid = start + (end - start) / 2;  //将中间值作为根节点的值
		  root->val = num[mid];

		  if (start <= mid - 1)   //如果左边数组不为空
		  {
			  TreeNode *node = new TreeNode(0);   //新建一个节点，为根节点的左子节点，同时作为左子树的根节点传入递归程序
			  root->left = node;
			  build(root->left, start, mid - 1, num);
		  }
		  if (mid + 1 <= end)  //如果右边数组不为空
		  {
			  TreeNode *node = new TreeNode(0);  //新建一个节点，为根节点的右子节点，同时作为右子树的根节点传入递归程序
			  root->right = node;
			  build(root->right, mid + 1, end, num);
		  }
	  }
  };