需要加判断条件再进入右往左读取的过程,否则由于代码最后还有一个res.push_back(mid);就会出现结果有个空的vector出现。这个问题会在最后一步执行的是左往右循环后出现,因此增加一个进入条件。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> res;
if(!root) return res; //空树
stack<TreeNode*> l; //左到右遍历
stack<TreeNode*> r; //右到左遍历
l.push(root);
while(!l.empty() || !r.empty()){
vector<int> tmp;
while(!l.empty()){
tmp.push_back(l.top()->val);
if(l.top()->left)
r.push(l.top()->left);
if(l.top()->right)
r.push(l.top()->right);
l.pop();
}
res.push_back(tmp); tmp.clear();
if(!r.empty()){
while(!r.empty()){
tmp.push_back(r.top()->val);
if(r.top()->right)
l.push(r.top()->right);
if(r.top()->left)
l.push(r.top()->left);
r.pop();
}
res.push_back(tmp); tmp.clear();
}
}
return res;
}
};