题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路:
与逐层打印二叉树一样,只是多了层数判定,依据层数做判定输出。
按之字形顺序打印二叉树需要两个栈。我们在打印某一行节点时,把下一层的子节点保存到相应的栈里。保存层数,对temp_res
的结果保存时结合层数做正反序append
。
参考答案:
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def Print(self, pRoot):
# write code here
if not pRoot:
return []
current_layer = [pRoot]
res = []
layer_level = 0
while current_layer:
next_layer = []
layer_res = []
for node in current_layer:
layer_res.append(node.val)
if node.left:
next_layer.append(node.left)
if node.right:
next_layer.append(node.right)
current_layer = next_layer[:]
if layer_level % 2 == 1:
res.append(layer_res[::-1])
else:
res.append(layer_res[:])
layer_level += 1
return res
补充一种 C++ 版本,注意体会,代码具有对称美!
C++ Version:
/*
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;
}
stack<TreeNode* > depNodes[2];
depNodes[0].push(pRoot);
TreeNode* curNode;
while(!depNodes[0].empty() || !depNodes[0].empty()){
vector<int> depValues[2];
# 偶数行
while(!depNodes[0].empty()){
curNode = depNodes[0].top();
depValues[0].push_back(curNode->val);
if(curNode->left != NULL){
depNodes[1].push(curNode->left);
}
if(curNode->right != NULL){
depNodes[1].push(curNode->right);
}
depNodes[0].pop();
}
if(!depValues[0].empty()){
res.push_back(depValues[0]);
}
# 奇数行
while(!depNodes[1].empty()){
curNode = depNodes[1].top();
depValues[1].push_back(curNode->val);
if(curNode->right != NULL){
depNodes[0].push(curNode->right);
}
if(curNode->left != NULL){
depNodes[0].push(curNode->left);
}
depNodes[1].pop();
}
if(!depValues[1].empty()){
res.push_back(depValues[1]);
}
}
return res;
}
};
Note
- 这道题要十分注意,其中current_layer 的结果
current_layer = next_layer[:]
,即是完全复制,直接用current_layer = next_layer
会出现错误的。