《leetcode-go》有序链表变成二叉搜索树

less_cold

于 2021-10-10 10:23:07 发布

阅读量103

点赞数

分类专栏： leetCode-go 文章标签： leetcode golang 链表

本文链接：https://blog.csdn.net/less_cold/article/details/120683302

版权

leetCode-go 专栏收录该内容

81 篇文章 0 订阅

订阅专栏

该博客介绍了如何将一个已排序的单链表转换为一棵平衡的二叉搜索树（BST）。通过迭代方法找到链表的中间节点作为BST的根节点，然后递归地对左右子链表进行同样的操作，确保了树的平衡和搜索效率。代码中定义了链表和树节点的结构，并提供了关键的`sortedListToBST`函数来完成转换过程。

摘要由CSDN通过智能技术生成

给定一个单链表，其中的元素按升序排序，请将它转化成平衡二叉搜索树（BST）

package main
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */
/*
 * type ListNode struct{
 *   Val int
 *   Next *ListNode
 * }
 */

/**
  * 
  * @param head ListNode类 
  * @return TreeNode类
*/
func sortedListToBST( head *ListNode ) *TreeNode {
    // write code here
    if head == nil {
		return nil
	}
	if head.Next == nil {
		treeNode := new(TreeNode)
		treeNode.Val = head.Val
		return treeNode
	}
	if head.Next != nil && head.Next.Next == nil {
		treeNode := new(TreeNode)
		treeNode.Val = head.Next.Val
		treeNode1 := new(TreeNode)
		treeNode1.Val = head.Val
		treeNode.Left = treeNode1
		return treeNode
	}
	node := head
	fastNode := head.Next
	//找到中间的位置
	var pre = node
	var mid = fastNode
	for fastNode != nil && fastNode.Next != nil {
		pre = node
		node = node.Next
		mid = node
		fastNode = fastNode.Next.Next
	}
	//偶数个找到下一个作为mid
	if fastNode != nil {
		pre = mid
		mid = mid.Next
	}
	treeNode := new(TreeNode)
	pre.Next = nil
	treeNode.Val = mid.Val
	treeNode.Left = sortedListToBST(head)
	treeNode.Right = sortedListToBST(mid.Next)
	mid.Next = nil
	return treeNode
}