给定一个二叉树,返回其节点值的锯齿形层次遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。
例如: 给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回锯齿形层次遍历如下:
[
[3],
[20,9],
[15,7]
]
此题与剑指offer第32题之字形打印二叉树一样
因为奇偶层的打印顺序不一样是相反的,可以用reverse来解决,但是海量数据时这个效率很低。
因为奇数层的打印顺序是从左到右,偶数层的打印顺序是从右到左,可以利用STL容器deque中push_back(),push_front(),front(),back(),pop(),popfront()来解决,实现前取后放,后取前放,因为deque队列中的元素,都是从左到右的,当flag==true;,从左到右打印,就从前边开始取出元素,下一层的元素从后边压入,同理,从右向左打印时,从后边取出元素,下一层的元素就从前边压入,注意先压入右子树的,后压入左子树,这样新的队列还是从左到右,然后继续,直至队列为空。
O(n)
O(n)
C++:
class Solution {
public:
vector<vector<int>> zigzaglevelOrder(TreeNode* root) {
vector<vector<int>> res;
if (root==NULL)
return res;
bool flag = true; //从左向右打印为true,从右向左打印为false
deque<TreeNode*> q;
q.push_back(root);
while (!q.empty())
{
int n = q.size();
vector<int> out;
TreeNode* node;
while (n>0)
{
if (flag) // 前取后放:从左向右打印,所以从前边取,后边放入
{
node = q.front();
q.pop_front();
if (node->left)
q.push_back(node->left); // 下一层顺序存放至尾
if (node->right)
q.push_back(node->right);
}
else //后取前放: 从右向左,从后边取,前边放入
{
node = q.back();
q.pop_back();
if (node->right)
q.push_front(node->right); // 下一层逆序存放至首
if (node->left)
q.push_front(node->left);
}
out.push_back(node->val);
n--;
}
flag = !flag;
res.push_back(out);
}
return res;
}
};
class Solution:
def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
res = []
if root is None:
return res
q = collections.deque()
q.append(root)
zigzag = True
while q:
n = len(q)
out = []
node = None
while n>0:
if zigzag:
node = q.popleft()
if node.left:
q.append(node.left)
if node.right:
q.append(node.right)
else:
node = q.pop()
if node.right:
q.appendleft(node.right)
if node.left:
q.appendleft(node.left)
out.append(node.val)
n-=1
res.append(out[:])
zigzag= not zigzag
return res