前言:
如果说代码有灵魂,那么它的灵魂一定是算法,因此,想要写出优美的程序,核心算法是必不可少的,少年,你渴望力量吗,想掌握程序的灵魂吗❓❗️那么就必须踏上这样一条漫长的道路,我们要做的,就是斩妖除魔,打怪升级!当然切记不可走火入魔,每日打怪,日日累积,终能成圣!今天就开启我们的斩妖之旅!✈️✈️
题目:
给定一个 n 叉树的根节点 root ,返回 其节点值的 前序遍历 。n 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。
示例1:
示例2:
注意事项:
- 节点总数在范围 [0, 104]内
- 0 <= Node.val <= 104
- n 叉树的高度小于或等于 1000
解法一:
思路:
首先开辟一个数组,用来存放N叉树前序遍历的结果,先将根节点压入数组,然后进行范围for(顺序遍历二叉树的每一个节点),将前序遍历的结果放入到tmp数组中,再使用范围for将tmp数组的值拷贝回原数组。最后返回原数组的值即可。
但是这样写的效率非常低,将ans数组拷贝到tmp数组,再将tmp数组拷贝回原数组,这样来来回回的拷贝效率实在是很低,所以我们可以考虑用封装来优化。
代码实现:
/*
// 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<int> preorder(Node* root) {
if(root == NULL) return vector<int>{};
vector<int> ans;//开辟一个数组用来记录前序遍历结果
ans.push_back(root -> val);//将前序遍历到的每个节点的值压入到数组中
for(auto x : root -> children)//范围for依次遍历N叉树的每个节点
{
vector<int> tmp = preorder(x);//用tmp数组接收前序遍历的结果
for(auto y : tmp) ans.push_back(y);//拷贝完成之后再将tmp数组元素拷贝回原数组
}
return ans;//返回前序遍历数组的结果即可
}
};
解法二:
思路:
以上是不使用封装解决前序遍历问题的方法,没有什么问题是一层封装解决不了的,如果有,那就两层。
1、我们在preorder函数中定义一个数组ans用来记录前序遍历结果,封装一个前序遍历的函数,将根节点和数组传ans入函数,其中数组传参是用引用传参(避免多一次拷贝)最后返回数组即可。
2、在函数内部,我们首先将遍历到的每个节点的值压入到数组ans当中,再使用范围for对N叉树的每个子孩子遍历,并且将前序遍历到的节点全部拷贝到ans数组中。
时间复杂度:O(N),其中 n 为 N 叉树的节点。每个节点恰好被遍历一次。
空间复杂度:O(N),递归过程中需要调用栈的开销,平均情况下为 O(logN),最坏情况下树的深度为 N−1,此时需要的空间复杂度为 O(N)。
代码实现:
/*
// 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:
void _preorder(Node *root, vector<int> &ans)//引用传参,少一次拷贝构造
{
if(root == NULL)
return;
ans.push_back(root -> val);//将前序遍历的节点值压入数组中
for(auto x : root -> children)//范围for便利
{
_preorder(x, ans);//将前序遍历结果也压入到ans数组中
}
return;
}
vector<int> preorder(Node* root) {
vector<int> ans;//记录前序遍历的结果
_preorder(root, ans);//进行前序遍历
return ans;//返回前序遍历的数组
}
};
今天第一次写的题也是比较简单的,主要是对数组拷贝的优化,将多次拷贝优化为在一个数组内操作。