牛客-二叉树层序遍历

最新推荐文章于 2023-09-06 05:04:22 发布

德鸭

最新推荐文章于 2023-09-06 05:04:22 发布

阅读量141

点赞数

分类专栏：牛客

本文链接：https://blog.csdn.net/qq_36908995/article/details/117818043

版权

二叉树层次遍历 TreeNode C++ levelOrder

关键词由CSDN通过智能技术生成

牛客专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目：

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */
class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > levelOrder(TreeNode* root) {
    vector<vector<int>> ans;
    queue<TreeNode*> que;
    queue<int>       tick; 
        
    if(root!=NULL)
    {
       que.push(root);
       tick.push(0);
    }
    else
    {
        return ans;
    }
        
    while(!que.empty())
    {
        TreeNode* q=que.front();
        int       p=tick.front();
        que.pop();
        tick.pop();
        
        if(ans.size()<p+1)
        {
            vector<int> newvector;
            ans.push_back(newvector);
        }
        ans[p].push_back(q->val);
        
        if(q->left!=NULL)
        {
            que.push(q->left);
            tick.push(p+1);
        }
        if(q->right!=NULL)
        {
            que.push(q->right);
            tick.push(p+1);
        }
    }
    return ans;
    }
};

记得每次进入新的一层时候，需要新建立一个vector