题目描述:
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
输入:
{8,6,10,5,7,9,11}
输出:
[[8],[10,6],[5,7,9,11]]
解法1:
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int>>result;
if(pRoot==nullptr)return result;
stack<TreeNode*>left_right_st;
stack<TreeNode*>right_left_st;
left_right_st.push(pRoot);
while(left_right_st.size()||right_left_st.size()){
if(!left_right_st.empty()){
vector<int>temp;
TreeNode* node;
while(!left_right_st.empty()){
node=left_right_st.top();
temp.push_back(node->val);
if(node->left)//先左在右
right_left_st.push(node->left);
if(node->right)
right_left_st.push(node->right);
left_right_st.pop();
}
result.push_back(temp);
}
if(!right_left_st.empty()){
vector<int>temp;
TreeNode*node;
while(!right_left_st.empty()){
node=right_left_st.top();
temp.push_back(node->val);
if(node->right)//这里是先右在左
left_right_st.push(node->right);
if(node->left)
left_right_st.push(node->left);
right_left_st.pop();
}
result.push_back(temp);
}
}
return result;
}
};