1. 程式人生 > 實用技巧 >二叉樹BFS遍歷

二叉樹BFS遍歷

二叉樹定義

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

一、層序遍歷(使用佇列)

使用一個佇列貯存節點,如果根節點非空,將根節點入隊。

只要佇列非空,就迴圈,列印top節點,將左右子節點入隊,pop隊頭。

vector<int> BiTree_order(TreeNode* root)
{
    vector
<int> retvec; if (root == nullptr) return retvec; queue<TreeNode*> q; q.push(root); while (!empty(q)) { auto t = q.front(); q.pop(); retvec.push_back(t->val); if (t->left != nullptr) q.push(t->left); if (t->right != nullptr) q.push(t->right); }
return retvec; }

二、層序遍歷分層輸出

如何界定每一層的分界點?

在每一層開始的時候獲取佇列長度,當前佇列中的所有點都在同一層。

vector<int> BiTree_order(TreeNode* root)
{
    vector<vector<int>> retvec;
    if (root == nullptr)
        return retvec;
    queue<TreeNode*> q;
    auto t = root;
    q.push(root);
    while (!empty(q))
    {
        
int size = q.size(); vector<int> level; for (int i = 0; i < size; i++) { t = q.front(); q.pop(); level.push_back(t->val); if (t->left != nullptr) q.push(t->left); if (t->right != nullptr) q.push(t->right); } retvec.push_back(level); } return retvec; }