qidao123.com技术社区-IT企服评测·应用市场

标题: 层序遍历重建二叉树，绘制二叉树图片 [打印本页]
作者: 农妇山泉一亩田    时间: 2025-4-28 13:00
标题: 层序遍历重建二叉树，绘制二叉树图片
在力扣刷二叉树相关题目时，输入一般都是完全层序遍历，我习惯在自己电脑上调试代码，因此才编写下面代码将完全层序遍历数据重建二叉树对象。
生成的结果二叉树一般也只会给出完全层序遍历，无法直观的感受二叉树现真相况，因此我编写代码将二叉树对象生成svg图片，刷二叉树相关题目更清晰直观了。
力扣原题：https://leetcode.cn/problems/w6cpku/ ，效果图如下：

代码如下：
  1. package main
  3. import (
  4.         "fmt"
  5.         "math"
  6.         "os"
  7.         "strconv"
  8.         "strings"
  9. )
  11. type TreeNode struct {
  12.         Val   int
  13.         Left  *TreeNode
  14.         Right *TreeNode
  15. }
  17. // 根据完全层序遍历构建二叉树
  18. func buildTree(s string) *TreeNode {
  19.         nums := strings.Split(strings.ReplaceAll(s, " ", ""), ",")
  20.         if len(nums) == 0 {
  21.                 return nil
  22.         }
  24.         n, err := strconv.Atoi(nums[0])
  25.         if err != nil {
  26.                 return nil
  27.         }
  29.         var (
  30.                 root  = &TreeNode{Val: n}
  31.                 queue = []*TreeNode{root}
  32.                 i     = 1
  33.         )
  34.         for i < len(nums) {
  35.                 node := queue[0]
  36.                 queue = queue[1:]
  37.                 if n, err = strconv.Atoi(nums[i]); err == nil {
  38.                         node.Left = &TreeNode{Val: n}
  39.                         queue = append(queue, node.Left)
  40.                 }
  41.                 i++
  42.                 if i < len(nums) {
  43.                         if n, err = strconv.Atoi(nums[i]); err == nil {
  44.                                 node.Right = &TreeNode{Val: n}
  45.                                 queue = append(queue, node.Right)
  46.                         }
  47.                 }
  48.                 i++
  49.         }
  50.         return root
  51. }
  53. // 生成完全层序遍历
  54. func levelOrder(root *TreeNode) string {
  55.         if root == nil {
  56.                 return ""
  57.         }
  58.         var (
  59.                 result strings.Builder
  60.                 queue  = []*TreeNode{root}
  61.         )
  62.         for len(queue) != 0 {
  63.                 node := queue[0]
  64.                 queue = queue[1:]
  65.                 if node == nil {
  66.                         result.WriteString("null,")
  67.                 } else {
  68.                         result.WriteString(strconv.Itoa(node.Val))
  69.                         result.WriteByte(',')
  70.                         queue = append(queue, node.Left)
  71.                         queue = append(queue, node.Right)
  72.                 }
  73.         }
  74.         return strings.TrimRightFunc(result.String(), func(r rune) bool {
  75.                 return r == 'n' || r == 'u' || r == 'l' || r == ','
  76.         })
  77. }
  79. // 生成二叉树的SVG图片
  80. func generateSVG(root *TreeNode) string {
  81.         var treeHeight func(*TreeNode) int
  82.         treeHeight = func(root *TreeNode) int {
  83.                 if root == nil {
  84.                         return 0
  85.                 }
  86.                 return max(treeHeight(root.Left), treeHeight(root.Right)) + 1
  87.         }
  89.         var (
  90.                 height = treeHeight(root)
  91.                 width  = int(math.Pow(2, float64(height))) * 50
  93.                 draw func(node *TreeNode, x, y, level int)
  95.                 svg = &strings.Builder{}
  96.         )
  97.         fmt.Fprintf(svg, `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 %d %d">`, width, height*100)
  99.         draw = func(node *TreeNode, x, y, level int) {
  100.                 if node == nil {
  101.                         return
  102.                 }
  104.                 nextLevel := level + 1
  105.                 offset := int(math.Pow(2, float64(height-nextLevel-1))) * 50
  107.                 if node.Left != nil {
  108.                         leftX := x - offset
  109.                         leftY := y + 100
  110.                         fmt.Fprintf(svg, `<line x1="%d" y1="%d" x2="%d" y2="%d" stroke="black" stroke-width="2" />`, x, y, leftX, leftY)
  111.                         draw(node.Left, leftX, leftY, nextLevel)
  112.                 }
  114.                 if node.Right != nil {
  115.                         rightX := x + offset
  116.                         rightY := y + 100
  117.                         fmt.Fprintf(svg, `<line x1="%d" y1="%d" x2="%d" y2="%d" stroke="black" stroke-width="2" />`, x, y, rightX, rightY)
  118.                         draw(node.Right, rightX, rightY, nextLevel)
  119.                 }
  121.                 fmt.Fprintf(svg, `<circle cx="%d" cy="%d" r="20" fill="lightblue" />`, x, y)
  122.                 fmt.Fprintf(svg, `<text x="%d" y="%d" text-anchor="middle" dominant-baseline="middle">%d</text>`, x, y, node.Val)
  123.         }
  125.         draw(root, width/2, 50, 0)
  126.         svg.WriteString(`</svg>`)
  127.         return svg.String()
  128. }
  130. func main() {
  131.         root := buildTree("4,1,6,0,2,5,7,null,null,null,3,null,null,null,8")
  133.         svg := generateSVG(root)
  135.         err := os.WriteFile("binary_tree.svg", []byte(svg), 0666)
  136.         if err != nil {
  137.                 fmt.Println("Error writing to file:", err)
  138.                 return
  139.         }
  141.         fmt.Println("SVG file created successfully!")
  142. }
复制代码
出处：https://www.cnblogs.com/janbar本文版权归作者和博客园所有，欢迎转载，转载请标明出处。喜欢我的文章请[关注我]吧。假如您觉得本篇博文对您有所劳绩，可点击[推荐]并[收藏]
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。



欢迎光临 qidao123.com技术社区-IT企服评测·应用市场 (https://dis.qidao123.com/) Powered by Discuz! X3.4