二叉樹BFS遍歷
阿新 • • 發佈:2020-12-23
二叉樹定義
/** * 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; }