剑指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

巴特西

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

1 题目描述

2 思路和方法

3 C++核心代码

参考资料

最新文章

热门文章