Re: [閒聊] 每日leetcode

[JIWP (神楽めあ的錢包)]

1382. Balance a Binary Search Tree

給你一個二元搜尋樹

想辦法把它變成深度不相差超過1

思路:

先把所有節點的值由記錄下來

再去做成題目要求的樣子

太久沒寫BST，差點忘記要怎麼寫

寫起來有夠卡

golang code :
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func balanceBST(root *TreeNode) *TreeNode {
        var arr []int
        sortarray(root, &arr)
        return dfs(arr)
}
func sortarray(node *TreeNode, arr *[]int) {
        if node == nil {
                return
        }
        sortarray(node.Left, arr)
        *arr = append(*arr, node.Val)
        sortarray(node.Right, arr)
}
func dfs(arr []int) *TreeNode {
        if len(arr) == 0 {
                return nil
        }
        res := &TreeNode{}
        res.Val = arr[len(arr)/2]
        res.Left = dfs(arr[:len(arr)/2])
        res.Right = dfs(arr[len(arr)/2+1:])
        return res
}


--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 223.140.184.232 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1719394102.A.522.html
→ yam276: :O
→ yam276: 今天忘了寫題目了
推 DJYOMIYAHINA: 法國我
→ JIWP: 我要法克你
推 sustainer123: 我本來是想能不能邊遍歷邊改
→ sustainer123:  後來想想直接紀錄比較快
→ JIWP: 我覺得沒辦法
推 oin1104: 模型 送我 我快哭了