算法笔记
本题是二叉树层序遍历的延伸,即N叉树的层序遍历,需要注意区分二者有什么异同点。
题目描述
解题步骤
1.类的变化: 在做类似题目时不仅要注意代码本身,也要关注结构体和类的写法。
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val) {
val = _val;
}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
2.层序遍历的区别: 二叉树的层序遍历在for循环中父节点出队后就会去让判断它的左右子节点是否为空,并让这些孩子入队,同理,N叉树也是一样的,只不过孩子多了一些,其余步骤不变。
代码部分
class Solution {
public:
vector<vector<int>> levelOrder(Node* root) {
vector<vector<int>> result;
queue<Node*> que;
if(root != NULL) {
que.push(root);
}
while(!que.empty()) {
int size = que.size();
vector<int> path;
for(int i = 0; i < size; i++) {
Node* node = que.front();
que.pop();
path.push_back(node->val);
for(int j = 0; j < node->children.size(); j++) {
if(node->children[j]) {
que.push(node->children[j]);
}
}
}
result.push_back(path);
}
return result;
}
};
结语
在习惯处理二叉树后,碰到N叉树可能会有点懵,但是问题不大,熟悉写法,理清思路这道题就可以很好解决。