IT公司100题-16-层遍历二元树

问题描述：

层遍历二叉树，同一层从左往右打印。

定义二元查找树的结点为：

typedef struct BSTreeNode {
    int data;
    BSTreeNode *left;
    BSTreeNode *right;
} Node;

例如输入二叉树：

   6
 /  
 4    12
/     /
2 5 8  16

输出：6 4 12 2 5 8 16。

 

分析：

二叉树的广度优先遍历。

 

代码实现：

// 16.cc
#include <deque>
#include <iostream>
using namespace std;

typedef struct BSTreeNode {
    int data;
    BSTreeNode *left;
    BSTreeNode *right;
} Node;

// 创建二元查找树
void add_BSTree_node(Node* &p_current, int data) {
    if (NULL == p_current) {
        Node *node = new Node();
        node->left = NULL;
        node->right = NULL;
        node->data = data;
        p_current = node;
    } else {
        if (p_current->data > data)
            add_BSTree_node(p_current->left, data);
        else if (p_current->data < data)
            add_BSTree_node(p_current->right, data);
        else
            cout << "The data has already in the Node.";
    }
}

// 层遍历
void level_order(Node* root) {
    deque<Node*> s;
    s.push_back(root);
    Node* p;
    
    while (!s.empty()) {
        p = s.front();
        s.pop_front();
        cout << p->data << " ";

        if (p->left != NULL)
            s.push_back(p->left);
        if (p->right != NULL)
            s.push_back(p->right);
    }
    cout << endl;
}

int main() {
    Node *root = NULL;
    add_BSTree_node(root, 6);
    add_BSTree_node(root, 4);
    add_BSTree_node(root, 12);
    add_BSTree_node(root, 2);
    add_BSTree_node(root, 5);
    add_BSTree_node(root, 8);
    add_BSTree_node(root, 16);

    level_order(root);
    return 0;
}