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()
}