前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >算法篇:树之转换为二叉搜索树

算法篇:树之转换为二叉搜索树

作者头像
灰子学技术
发布2020-08-17 15:18:35
2290
发布2020-08-17 15:18:35
举报
文章被收录于专栏:灰子学技术灰子学技术

算法:

核心思想是利用二分法,不过有序数组和有序链表找到中间节点的方法不一致。

代码语言:javascript
复制
1.对有序数组或者有序链表来说,把中间节点当作根节点
2. 左边数组的值都小于根节点,作为左子树;
 右边数组的值都大于根节点,作为右子树。
3. 递归处理左子树和右子树,一直到只剩下一个节点。

题目1:

https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree/

代码实现:

代码语言:javascript
复制
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func sortedArrayToBST(nums []int) *TreeNode {
    if len(nums) == 0 {
        return nil
    }
    mid := len(nums)/2
    root := new(TreeNode)
    root.Val = nums[mid]
    root.Left = sortedArrayToBST(nums[:mid])
    root.Right = sortedArrayToBST(nums[mid+1:])
    return root
}
// 算法:核心思想是利用二分法,对有序数组来说,把中间节点当作根节点,
// 左边数组的值都小于根节点,作为左子树;
// 右边数组的值都大于根节点,作为右子树。
// 递归处理左子树和右子树,一直到只剩下一个节点。

执行结果:

题目2:

https://leetcode-cn.com/problems/convert-sorted-list-to-binary-search-tree/

代码实现:

代码语言:javascript
复制
/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func sortedListToBST(head *ListNode) *TreeNode {
    if head == nil {
        return nil
    }
    if head.Next == nil {
        return &TreeNode{Val:head.Val}
        
    }
    // 小技巧:方便一次循环找到中间节点的前序节点,哨兵
    s := new(ListNode) 
    s.Next = head
    f := head
    for f != nil && f.Next != nil { 
        // 精髓:快慢指针,1步,2步正好是二等分,可以延伸出三等份,n等分
        s = s.Next
        f = f.Next.Next
    }
    
    res := new(TreeNode)
    res.Val = s.Next.Val
    r := s.Next.Next
    s.Next = nil // 左半部分单独成一个链表
    res.Left = sortedListToBST(head)
    res.Right = sortedListToBST(r)
    return res
}
// 算法:利用二分法,这里是采用了链表二分法的常规做法,
// 找到中间节点之后,将链表一分为二,左边的继续构造左子树,右边的为右子树
// 递归处理,直到所有节点都处理完。

执行结果:

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2020-08-13,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 灰子学技术 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档