429. N 叉树的层序遍历 c++ 队列

给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。

树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。

429. N 叉树的层序遍历 c++ 队列_第1张图片

输入:root = [1,null,3,2,4,null,5,6]
输出:[[1],[3,2,4],[5,6]]

429. N 叉树的层序遍历 c++ 队列_第2张图片 

 

输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]

 

提示:

  • 树的高度不会超过 1000
  • 树的节点总数在 [0, 10^4] 之间

可能难点:如何记录每层?

设置一个int size=queue.size();令size--到0时为该层结束。

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector children;

    Node() {}

    Node(int _val) {
        val = _val;
    }

    Node(int _val, vector _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
public:
    vector> levelOrder(Node* root) {
        vector> res;
        if(!root) return res;
        int size;
        queue que;
        Node* p;
        vector q;
        que.push(root);
        while(!que.empty()){

            size=que.size();
            vector path;
            while(size--){
                p=que.front();
                path.push_back(p->val);
                que.pop();
                q=p->children;
                for(Node* item:q){
                    que.push(item);
                }
            }
            res.push_back(path);
        }

        return res;
    }
};

你可能感兴趣的:(代码题,c++,层序遍历)