题目描述:
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
分析:
我们都知道二叉树的层次遍历用的是队列。
但是这个的子树的结点是先进后出的,所以我们使用栈更加方便。
而且为了方便,我用了两个栈,因为不是每一次都是先进左子树再进右子树的。
代码:
1 /* 2 struct TreeNode { 3 int val; 4 struct TreeNode *left; 5 struct TreeNode *right; 6 TreeNode(int x) : 7 val(x), left(NULL), right(NULL) { 8 } 9 }; 10 */ 11 class Solution { 12 public: 13 vector<vector<int> > Print(TreeNode* pRoot) { 14 vector<vector<int> > res; 15 if(pRoot == NULL) return res; 16 stack<TreeNode*> s1, s2; 17 s1.push(pRoot); 18 while(1) { 19 vector<int> v1; 20 while(!s1.empty()) { 21 TreeNode* top = s1.top(); 22 v1.push_back(top->val); 23 if(top->left) s2.push(top->left); 24 if(top->right) s2.push(top->right); 25 s1.pop(); 26 } 27 res.push_back(v1); 28 if(s2.empty()) break; 29 vector<int> v2; 30 while(!s2.empty()) { 31 TreeNode* top = s2.top(); 32 v2.push_back(top->val); 33 if(top->right) s1.push(top->right); 34 if(top->left) s1.push(top->left); 35 s2.pop(); 36 } 37 res.push_back(v2); 38 if(s1.empty()) break; 39 } 40 return res; 41 } 42 };