剑指offer59：按之字形顺序打印二叉树：[[1], [3,2], [4,5,6,7]]

1 题目描述

　　请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

2 思路和方法

先给定一个二叉树的样式：

　　输出的样式是：[[1], [3,2], [4,5,6,7]]。包含以下信息： （1）每一层所包含的树节点；（2）偶数层的树节点需倒序。

　　思路： 面对要求的偶数层倒序，亦有两种解题思路，第一种是：获取到所有节点的值后，将偶数层的节点值倒序（先存right，再存left实现倒序）。第二种则是在获取节点的值的时候就倒序存入。定义两堆栈stack1和stack2，在遍历当前层节点的同时!stack1.empty() TreeNode *data=stack1.top();，存储下一层的节点（stack2.push(data->right),stack2.push(data->left)），以此类推，!stack2.empty() TreeNode *data=stack2.top();，存储下一层的节点（stack1.push(data->left),stack2.push(data->right)）。　　　

3 C++核心代码

 /*
 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>> result;
         if(pRoot==nullptr)
             return result;
         stack<TreeNode*> stack1,stack2;//分别存奇数和偶数层
         stack1.push(pRoot);
         while(!stack1.empty() || !stack2.empty()){
             if(!stack1.empty()){
                 vector<int> temp;
                 while(!stack1.empty()){
                     TreeNode *data=stack1.top();
                     stack1.pop();
                     temp.push_back(data->val);
                     if(data->left!=nullptr)
                         stack2.push(data->left);
                     if(data->right!=nullptr)
                         stack2.push(data->right);
                 }
                 result.push_back(temp);
             }
             if(!stack2.empty()){
                 vector<int> temp;
                 while(!stack2.empty()){
                     TreeNode *data=stack2.top();
                     stack2.pop();
                     temp.push_back(data->val);
                     if(data->right!=nullptr)
                         stack1.push(data->right);
                     if(data->left!=nullptr)
                         stack1.push(data->left);
                 }
                 result.push_back(temp);
             }
         }
         return result;
     }
 };

参考资料

https://blog.csdn.net/u010005281/article/details/79759926