其他分享
首页 > 其他分享> > 层序建立二叉树并输出,先序遍历二叉树并输出

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

作者:互联网

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