bfs 和 队列

学过数据结构的都知道,二叉树的层次遍历。

层次遍历利用的数据结构是队列。

那么,

思考一下

为什么层次遍历,要用到队列,而不是其他的数据结构,比如栈呢?换句话说,队列在二叉树的层次遍历过程中起到了什么作用呢?

队列在二叉树层次遍历中的作用

我们知道,二叉树的结构如下:

/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/

从结构中大致可以知道,二叉树的根跟左右还是是采用链表的形式联系的,父节点,保存了左右孩子的指针,所以,我们知道了父节点,必然也就知道了左右孩子的信息。

但是,在一个二叉树中,同层之间的信息通过什么联系在一起的呢?

7

5 8

2 6 9 12

比如上图中的2,6,9,12,这时层次遍历的时候,我们如何保持同层之间的顺序呢?这个就应该是靠队列的作用了。

队列的一个特点是,元素是先进先出,我们可以把同层的兄弟元素依次入队,然后,在出队的过程中再遍历左右孩子,循环以上操作即可。

通过数组创建二叉树vs二叉树的层次遍历

今天,做leetcode上的一道二叉树层次遍历的题目,由于需要在本地调试,所以想在本地创建一个二叉树,于是想到了,通过数组去创建一个二叉树,也就是将一个数组转化为完全二叉树。

最后发现,原来,通过数组创建二叉树和二叉树的层次遍历其实存在着某种联系,他们是一个相反的过程。

也就是通过数组创建二叉树是将数组转化为二叉树,而二叉树的层次遍历其实是将二叉树转化为数组的形式。

确实很妙。

下面给出代码

python
#!/usr/bin/python
#-*- coding: UTF-8 -*-
class node():
def __init__(self,k=None,l=None,r=None):
self.key=k;
self.left=l;
self.right=r;
def create_tree_by_array(input_list):
"""

通过数组创建二叉树,

其实是层次遍历

需要利用节点队列

"""
def _createNode(item):
if item is '#':
return None
else:
return node(k=item)
queue = []
root = _createNode(input_list.pop(0))
queue.append(root)
while len(input_list) > 0:
curNode = queue.pop(0)
curNode.left = _createNode(input_list.pop(0))
curNode.right = _createNode(input_list.pop(0))
if curNode.left:
queue.append(curNode.left)
if curNode.right:
queue.append(curNode.right)
return root
def traverse_tree_by_level(root):
"""

层次遍历

需要利用节点队列

"""
queue = []
output_list = []
queue.append(root)
output_list.append(root.key)
while len(queue) > 0:
curNode = queue.pop(0)
if curNode.left:
output_list.append(curNode.left.key)
queue.append(curNode.left)
if curNode.right:
output_list.append(curNode.right.key)
queue.append(curNode.right)
print(output_list)
def inorder(root): #中序遍历
if root is None:
return ;
else:
inorder(root.left);
print(root.key,)
inorder(root.right);
tree_data_list = [3,9, 20, '#','#', 15,7]
root=None; # 测试代码
root=create_tree_by_array(tree_data_list);
inorder(root);
traverse_tree_by_level(root)
golang
package main
import (
"fmt"
)
type Node struct {
Value int
Left *Node
Right *Node
}
func CreateNode(item int) *Node {
if item == -1 {
return nil
}
return &Node{item, nil, nil}
}
func CreateByBreadthFirstSearch(inputList []int) *Node {
queue := []*Node{}
root := CreateNode(inputList[0])
queue = append(queue, root)
inputList = inputList[1:]
for len(inputList) > 0 {
curNode := queue[0]
queue = queue[1:]
curNode.Left = CreateNode(inputList[0])
if curNode.Left != nil {
queue = append(queue, curNode.Left)
}
inputList = inputList[1:]
curNode.Right = CreateNode(inputList[0])
if curNode.Right != nil {
queue = append(queue, curNode.Right)
}
inputList = inputList[1:]
}
return root
}
func (node *Node) BreadthFirstSearch() {
if node == nil {
return
}
output_list := []int{}
queue := []*Node{node}
for len(queue) > 0 {
curNode := queue[0]
queue = queue[1:]
output_list = append(output_list, curNode.Value)
if curNode.Left != nil {
queue = append(queue, curNode.Left)
}
if curNode.Right != nil {
queue = append(queue, curNode.Right)
}
}
for _, v := range output_list {
fmt.Print(v, " ")
}
}
func main() {
input_list := []int{3, 7, 9, -1, -1, 20, 4}
root := CreateByBreadthFirstSearch(input_list)
root.BreadthFirstSearch()
}