leetcode102. 二叉树的层次遍历

题目

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

例如:
给定二叉树: [3,9,20,null,null,15,7],

3

/
9 20
/
15 7

返回其层次遍历结果：

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

思路

1.遍历每层节点时用队列存储每个节点的左右子节点
2.当层节点遍历完后，将节点值存入数组，然后开始遍历下层节点
3.此处需要层次打印，所以引入计数器统计当前层次的节点数量和下层节点的数量

时间复杂度：每个点进队出队各一次，故渐进时间复杂度为 O(n)。
空间复杂度：队列中元素的个数不超过 n 个，故渐进空间复杂度为 O(n)。

实现

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if (root == NULL) {
            return res;
        }
        func(root, res);
        return res;
    }
    void func(TreeNode* root, vector<vector<int>>& res) {
        queue<TreeNode*> temp;
        vector<int> temp1;
        int num = 0;
        int count = 0;
        temp.push(root);
        num++;
        while (!temp.empty()) {
            if (temp.front()->left != NULL) {
                temp.push(temp.front()->left);
                count++;
            }
            if (temp.front()->right != NULL) {
                temp.push(temp.front()->right);
                count++;
            }         
            temp1.push_back(temp.front()->val);
            temp.pop();
            num--;
            if (num == 0) {
                res.push_back(temp1);
                num = count;
                count = 0;
                temp1.clear();
            }
            
        }
        
    }
};