ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

层序建立二叉树并输出,先序遍历二叉树并输出

2022-07-01 16:02:47  阅读:110  来源: 互联网

标签:输出 TreeNode int 层序 que 二叉树 ans root


func main() {
    //按数组层序建立二叉树,之后层序输出二叉树
    root:=createTree(0,[]int{1,2,3,4,5,6})
    ans:=printTree(root)
    fmt.Println(ans)
    
    //先序建立二叉树,之后先序输出二叉树
    var tmp *TreeNode
    for i:=0;i<6;i++{
        tmp=insert(tmp,i)
    }
    print(tmp)
}
type TreeNode struct {
    Val int
    Left *TreeNode
    Right *TreeNode
}
// 按数组建立二叉树
func createTree(index int,nums []int) *TreeNode{
    if index>=len(nums){
        return nil
    }
    root:=&TreeNode{
        Val:   nums[index],
    }
    root.Left=createTree(index*2+1,nums)
    root.Right=createTree(index*2+2,nums)
    return root
}

// 二叉树的层序遍历
func printTree(root *TreeNode) [][]int{
    que:=make([]*TreeNode,0)
    que=append(que,root)
    ans:=make([][]int,0)
    if root==nil{
        return ans
    }
    for len(que)!=0{
        size:=len(que)
        ans2:=make([]int,0)
        for i:=0;i<size;i++{
            front:=que[0]
            ans2=append(ans2,front.Val)
            que=que[1:]
            if front.Left!=nil{
                que=append(que,front.Left)
            }
            if front.Right!=nil{
                que=append(que,front.Right)
            }
        }
        ans=append(ans,ans2)
    }
    return ans
}
//先序建立二叉树
func insert(root *TreeNode,x int) *TreeNode{
    if root==nil{
        tmp:=&TreeNode{
            Val:x,
        }
        return tmp
    }
    if x<=root.Val{//这里是按照先序遍历的方式建立二叉树
        root.Left=insert(root.Left,x)
    }else{
        root.Right=insert(root.Right,x)
    }
    return root
}
//二叉树的先序遍历
func print(root *TreeNode){
    if root==nil{
        return
    }
    fmt.Printf("%v ",root.Val)
    if root.Left!=nil{
        print(root.Left)
    }
    if root.Right!=nil{
        print(root.Right)
    }
}

 

标签:输出,TreeNode,int,层序,que,二叉树,ans,root
来源: https://www.cnblogs.com/-citywall123/p/16434923.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有