LeetCode 429. n-叉树的层序遍历(C++)

题目地址:力扣

思路:和二叉树类似,n叉树层序遍历也是一层一层输出节点,返回值应该是一个二维数组。我们可以用一个队列来存储每一层的节点,输出节点的时候将其孩子加入队列。同时使用一个变量来记录每一层的节点数量,当出队的节点数量等于该层节点数,就完成了当前层的遍历。以此类推,直到队列为空就说明遍历完了整棵树。

注意:n-叉树和二叉树唯一不同的地方就在于其孩子并非左右孩子两个,而是需要把当前节点的所有孩子加入队列。

class Solution {
public:
    vector> levelOrder(Node* root) {
        vector> res;
        // 若树非空
        if (root != nullptr)
        {
            // 先把根节点加入队列
            queue treeq;
            treeq.push(root);
            // 只要队列非空
            while (!treeq.empty())
            {
                // 记录本层节点数量和本层遍历结果
                int curLayer_cnt = treeq.size();
                vector curLayer;
                // 遍历本层所有节点
                while (curLayer_cnt !&#

你可能感兴趣的:(力扣刷题,leetcode,算法,职场和发展)