首页 > 代码库 > leetcode Binary Tree Level Order Traversal
leetcode Binary Tree Level Order Traversal
给定一个数，广度优先输出记录。例如：
Given binary tree {3,9,20,#,#,15,7},
3 / 9 20 / 15 7
return its level order traversal as:
[ [3], [9,20], [15,7]]
思考了一下，居然被我想到了用队列来存储每一层的数据。嘿嘿。
用两个队列，一个队列用来输出，一个队列用来记录下一层的所有左右节点，然后循环使用两个队列即可。
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public: vector<vector<int> > levelOrder(TreeNode *root) { vector<vector<int> > ans; if (!root) return ans; vector<int> tmp; TreeNode *lf, *ri, *p; queue<TreeNode *> q1, q2; q1.push(root); while(!q1.empty() || !q2.empty()) { while(!q1.empty()) { p = q1.front(); q1.pop(); tmp.push_back(p -> val); if (p -> left) q2.push(p -> left); if (p -> right) q2.push(p -> right); } ans.push_back(tmp); tmp.clear(); if (q1.empty() && q2.empty()) return ans; // q1完，判断是否都为空 while(!q2.empty()) { p = q2.front(); q2.pop(); tmp.push_back(p -> val); if (p -> left) q1.push(p -> left); if (p -> right) q1.push(p -> right); } ans.push_back(tmp); tmp.clear(); } return ans; }};
一开始使用两个flag标记，来判断是否进入了中间的两个while是否进入，用来判断是否要pushback。后面发现只要在上面注释部分判断是否都为空就行。
当然，我用到了两个队列。可不可以只用一个呢。
用一个队列的话就是用NULL来判断分层。每次完后输入一个NULL。例如：
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public: vector<vector<int> > levelOrder(TreeNode *root) { // IMPORTANT: Please reset any member data you declared, as // the same Solution instance will be reused for each test case. vector<vector<int> >res; if(root == NULL)return res; queue<TreeNode*> myqueue; myqueue.push(root); myqueue.push(NULL);//NULL是层与层之间间隔标志 vector<int> level; while(myqueue.empty() == false) { TreeNode *p = myqueue.front(); myqueue.pop(); if(p != NULL) { level.push_back(p->val); if(p->left)myqueue.push(p->left); if(p->right)myqueue.push(p->right); } else { res.push_back(level); if(myqueue.empty() == false) { level.clear(); myqueue.push(NULL); } } } return res; }};
leetcode Binary Tree Level Order Traversal
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。