目录
1.层序遍历
与前中后序遍历使用的栈相对应,这里使用队列来实现层序遍历,需要注意的一点细节就是,我们如何确定一层的节点数是多少,很简单,在取之前用size取出当前队列的大小即可
代码如下所示:
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
class Solution {
public:
/**
*
* @param root TreeNode类
* @return int整型vector<vector<>>
*/
vector<vector<int> > levelOrder(TreeNode* root) {
// write code here
vector<vector<int>> res;
if(root==NULL) return res;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
int nums=q.size();
vector<int> tmp;
for(int i=0;i<nums;++i){
TreeNode *node=q.front();
q.pop();
tmp.push_back(node->val);
if(node->left){
q.push(node->left);
}
if(node->right){
q.push(node->right);
}
}
res.push_back(tmp);
}
return res;
}
};
2.之字型遍历
这里直接对节点进行处理是不大方便的,我们可以先普通层序遍历,然后按照奇偶层来对临时的数组来判断是否需要翻转,代码如下所示:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int>> res;
if(pRoot==NULL) return res;
queue<TreeNode*> q;
q.push(pRoot);
bool l2r=true;
while(!q.empty()){
int nums=q.size();
vector<int> tmp;
l2r=!l2r;
for(int i=0;i<nums;++i){
TreeNode *node=q.front();
q.pop();
tmp.push_back(node->val);
if(node->left){
q.push(node->left);
}
if(node->right){
q.push(node->right);
}
}
if(l2r){
reverse(tmp.begin(), tmp.end());
}
res.push_back(tmp);
}
return res;
}
};