二叉树的S型遍历(双栈)

思路:维护两个栈,逻辑如下:

  • 奇数层的遍历明显从右到左,其下一层反向所以这层入栈的子树也应该从右到左
  • 偶数层反向
    注意在维护STL的时候指针参数地传递,关于对象的赋值要处理好,可以用指针减少不必要的麻烦!
  • 在处理遍历方向时候通过根节点加入不同的栈能够完成顺序地完全对称遍历,也就是本来是之字形遍历变为S型遍历

Code

/* 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> > ans;
        ans.clear();
        stack<TreeNode*> st1, st2;
        while (!st1.empty()) st1.pop();
        while (!st2.empty()) st2.pop();
         if (!pRoot) return ans;
        st1.push(pRoot);
        while (!st1.empty() || !st2.empty()) {
            vector<int> tmp;
            tmp.clear();
            if (!st1.empty()) {
                while (!st1.empty()) {
                    TreeNode* vis = st1.top();
                    st1.pop();
                    tmp.push_back(vis->val);                    
                    if (vis->left) st2.push(vis->left);
                    if (vis->right) st2.push(vis->right);
                }
            } else {
                while (!st2.empty()) {
                    TreeNode* vis = st2.top();
                    st2.pop();
                    tmp.push_back(vis->val);                    
                    if (vis->right) st1.push(vis->right);
                    if (vis->left) st1.push(vis->left);
                }
            }
            ans.push_back(tmp);
        }
        return ans;
    }
     
};

全部评论

相关推荐

面了这么多场试,总有公司总喜欢压力面一个小时面试+手撕,哪里不会就点哪里,说了不会不会还继续追着问不尊重求职者,稍微有些细节记不清了,就开始怀疑项目真实性以及人格让求职者开摄像头但是自己不开,说话声音还贼小,pardon几次就开始不耐烦的不知道这个算不算,手撕的时候,面试官人跑了。。。最后快结束才来
一纸丿繁华丶:你换位思考一下,自己在职场被领导push麻了,身心俱疲,现在有个机会让你放松一下,体验一把上位者的感觉,还能看着那些高学历人才、未来自己的竞争者,抓耳挠腮、手足无措的样子,没给你当场笑出来就不错了,理解一下面试官吧。
点赞 评论 收藏
分享
兄弟们你们进大厂靠的是什么项目啊
DOTPHTP:课设改。其实项目什么的如果不是实习里面的生产项目的话,建议✍️那种自己想要做的。突出个人自驱力,而不是为了找工作不得不随波逐流这种
点赞 评论 收藏
分享
05-19 19:57
蚌埠学院 Python
2237:Gpa70不算高,建议只写排名,个人技能不在多而在精,缩到8条以内。项目留一个含金量高的,减少间距弄到一页,硕士简历也就一页,本科不要写很多
点赞 评论 收藏
分享
喜欢核冬天的哈基米很想上市:会爆NullPointerException的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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