树有几层知道的话可以这样

把二叉树打印成多行

http://www.nowcoder.com/questionTerminal/445c44d982d04483b04a54f298796288

vector<vector<int> > Print(TreeNode* pRoot) {
    vector<vector<int> > ret;
    if(!pRoot)
        return ret;
    queue<TreeNode*> q;
    q.push(pRoot);
    int cnt = 0;
    int base = 1;
    vector<int> tmp;
    for(int i = 0;i <4096;i ++)
    {
        TreeNode* p = q.front();
        q.pop();
        cnt ++;
        if(p)
            tmp.push_back(p->val);//cout<<p->val<<" ";
        if(cnt==base)
        {
            cnt = 0;
            base = (base<<1);
//            cout<<endl;
            if(tmp.size()!= 0)
            {
                vector<int> t;
                for(int i= 0;i < tmp.size();i ++)
                {
                    t.push_back(tmp[i]);
                }
                ret.push_back(t);
                tmp.clear();
            }
        }
        if(p)
        {
            q.push(p->left);
            q.push(p->right);
        }else
        {
            q.push(NULL);
            q.push(NULL);
        }
    }

    return ret;
}
全部评论

相关推荐

AI牛可乐:哇,听起来你很激动呢!杭州灵枢维度科技听起来很厉害呀~你逃课去白马培训,老冯会同意吗?不过既然你这么感兴趣,肯定是有原因的吧! 对了,想了解更多关于这家公司或者求职相关的问题吗?可以点击我的头像私信我哦,我可以帮你更详细地分析一下!
你都用vibe codi...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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