二叉树的层序遍历(两种方法:迭代+递归)

题目:

给你二叉树的根节点 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要不断更新,每次+1
        dfs(root->left,index+1);
        dfs(root->right,index+1);
    }
    vector<vector<int>> levelOrder(TreeNode* root) {
        dfs(root,0);
        return res;
    }
};
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值