题目:
给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
解题思路:迭代法实现
层序遍历一般都是用广度优先 遍历,那么迭代法就是最直观的且最简单的。
借助一个队列,层层遍历,将每一层的节点入队,出队时由于队列先进先出的特点,也会按顺序出队。在出队前,我们先计算队列的长度len,这个长度len就相当于当前层一共有len个节点,所以我们只需要将len个节点出队即可,其他的节点都是下一层的,我们先不去管,所以在循环出队时,结束条件就是len=0。
源代码如下:
//迭代法
class Solution {
public:vector<vector<int>> levelOrder(TreeNode* root) {queue<TreeNode*> qq;//定义一个栈vector<vector<int>> res;//结果数组if(root==nullptr) return res;//root为空,直接返回空数组即可qq.push(root);//先将根节点入队while(!qq.empty())//循环结束条件使队列为空{vector<int> temp;//定义一个临时数组,用来存放出队的元素int len=qq.size();//计算当前队列的长度while(len)//长度len变为0时,说明当前层的节点已经全部出队了{TreeNode* node=qq.front();//队头元素qq.pop();//出队temp.push_back(node->val);//节点放入temp中//继续将节点的左右节点依次入队if(node->left!=nullptr) qq.push(node->left);if(node->right!=nullptr) qq.push(node->right);len--;//记得每出队一个元素,就要len-1}res.push_back(temp);//临时数组放入res中}return res;}
};
解题思路:递归法实现
递归也就是我们常说的深度优先遍历,需要设置一个index,用来记录遍历到哪一层了,将节点放入到自己的层中。
源代码如下:
//递归法
class Solution {
public:vector<vector<int>> res;//定义res结果数组void dfs(TreeNode* root,int index){if(!root) return ;//root为空,直接return//如果index超过数组的大小,那么就添加一个新的层//eg.原本res=[3],[2,5] index=2 res.size()=2//那么现在res=[3],[2,5],[]if(index>=res.size()){res.push_back(vector<int>());}//将节点的值放入到相应的层中res[index].push_back(root->val);//继续递归节点的左子树和右子树,记得index要不断更新,每次+1dfs(root->left,index+1);dfs(root->right,index+1);}vector<vector<int>> levelOrder(TreeNode* root) {dfs(root,0);return res;}
};