C++详解Leetcode:102. Binary Tree Level Order Traversal

原题

思想

这一题主要考查二叉树的层次遍历，最常用的方法就是用队列去做，但是需要将二叉树的节点值存到一个二维数组中去，然后返回，这需要在队列存储时候做一些操作，通过NULL来标记二叉树的层次

代码和详解

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
};

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        //分配一个二维向量
        vector<vector<int>> result;
        //用队列来存储二叉树结点值
        queue<TreeNode *> q;
        //如果根节点为空，则返回
        if (!root)
        {
            return result;
        }
        //将根节点先放入队列中
        q.push(root);
        //再将NULL放入队列中，NULL是为了标记那一层的作用
        q.push(NULL);

        //cur是为了存储每一层的节点值
        vector<int> cur;
        //循环进行二叉树层次遍历，直到队列为空
        while (!q.empty())
        {
            //去除队列的第一个值
            TreeNode* t = q.front();
            //出队列
            q.pop();

            //如果t==NULL则证明当前层的遍历已经结束，需要将这一层存在cur中的值放到result中
            if (t == NULL)
            {
                result.push_back(cur);
                //将cur大小重新置为0
                cur.resize(0);
                //如果队列不为空，证明还没有遍历结束，继续讲NULL入队列作为标记
                if (q.size() > 0)
                {
                    q.push(NULL);
                }
            }
            else
            {
                //将当前的节点值存到cur中
                cur.push_back(t->val);
                //如果左孩子不为空，则入队列
                if (t->left)
                {
                    q.push(t->left);
                }
                //如果右孩子不为空，则入队列
                if (t->right)
                {
                    q.push(t->right);
                }
            }
        }
        //返回结果
        return result;
    }
};