leetcode之binary-tree-zigzag-level-order-traversal
题目
Given a binary tree, return the zigzag level order traversal of its nodes’ values. (ie, from left to right, then right to left for the next level and alternate between).
For example:
Given binary tree{3,9,20,#,#,15,7},
return its zigzag level order traversal as:
confused what"{1,#,2,3}"means? > read more on how binary tree is serialized on OJ.
题意
给定一个二叉树,令它从上而下输出一个二维数组,这个二维数组在奇数层是从左到右输出,在偶数曾是从右到左输出。
解题思路
这里很简单,利用queue队列来进行层次遍历,记录层次遍历时的level数,判断当前是奇数层还是偶数层,如果是偶数层,则直接反转该层后,再添加到二维数组中去。
C++实现代码
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(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> > zigzagLevelOrder(TreeNode *root) {
queue<TreeNode*> queue;
int level=0;
vector<vector<int>> vv;
if(root == nullptr)
return vv;
queue.push(root);
while(!queue.empty()){
vector<int> v;
int num=queue.size();
for(int i=0;i<num;i++){
TreeNode *node = queue.front();
queue.pop();
v.push_back(node->val);
if(node->left)
queue.push(node->left);
if(node->right)
queue.push(node->right);
}
level++;
if(level%2 ==0)
reverse(v.begin(),v.end());
vv.push_back(v);
}
return vv;
}
};