Leetcode 102. 二叉树的层序遍历 C++

Leetcode 102. 二叉树的层序遍历

题目

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

测试样例

二叉树：[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

题解

可以通过BFS获取结果，也可以通过递归算法传递层数实现。代码方法为后者，前者用一个队列便可以实现

代码

	vector<vector<int>> ans;
    void search(TreeNode* root,int length){
        if(length > ans.size()) ans.push_back({});	//第一次遍历到该深度
        ans[length-1].push_back(root->val);
        if(root->left)  search(root->left,length+1);
        if(root->right) search(root->right,length+1);
    }
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(root == NULL)    return ans;
        else ans.push_back({});
        search(root,1);
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。