题解 | #按之字形顺序打印二叉树#

按之字形顺序打印二叉树

http://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

class Solution {
public:
map<TreeNode,int> m;
#遍历二叉树,使用map存储每个节点的高度,此处使用前序遍历
void biaoji(TreeNode
root,int n){
if(!root) return;
m[root] = n;
n++;
biaoji(root->left,n);
biaoji(root->right,n);
return;
}
#层序遍历读取每个节点,存储于v中
void bianli(TreeNode* root,vector<TreeNode> &v){
queue<TreeNode
> q;
TreeNode* temp;
q.push(root);
while(!q.empty()){
temp = q.front();
v.push_back(temp);
q.pop();
if(temp->left) q.push(temp->left);
if(temp->right) q.push(temp->right);
}
return;
}
#遍历v,以高度为分界,m[v[i]]为奇数直接push参数val至temp中;m[v[i]]为偶数利用栈存取后取出。
vector<vector<int> > Print(TreeNode* pRoot) {
vector<TreeNode*> v;
vector<int> temp;
vector<vector<int>>vv;
stack<int> s;
if(!pRoot) return {};
biaoji(pRoot,1);
bianli(pRoot, v);
for(int i=0;i<v.size();i++){
if(m[v[i]]%2==1){
temp.push_back(v[i]->val);}
else{
s.push(v[i]->val);
}
if(m[v[i]]!=m[v[i+1]]){
while(!s.empty()){
temp.push_back(s.top());
s.pop();
}
vv.push_back(temp);
temp.clear();
}
}
return vv;
}</int></int></int></int>

};

全部评论

相关推荐

点赞 评论 收藏
分享
10-14 21:00
门头沟学院 Java
吃花椒的狸猫:这个人说的倒是实话,特别是小公司,一个实习生哪里来的那么多要求
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务