当前位置: 代码迷 >> 综合 >> leetcode102 Binary Tree Level Order Traversal(二叉树的层序遍历)
  详细解决方案

leetcode102 Binary Tree Level Order Traversal(二叉树的层序遍历)

热度:69   发布时间:2023-11-17 01:20:22.0

题目要求

给定一个二叉树,返回节点值的层序遍历结果。
(tips:从左到右,一层一层)

输入示例

For example:
给定一个二叉树 [3,9,20,null,null,15,7],3/ \9  20/  \15   7
return 层序遍历结果如下:
[[3],[9,20],[15,7]
]

解题思路

根据输入示例的结果形式可知,我们需要两个vector来辅助解答。
一个vector存放每一层的遍历结果,一个vector存放最后的整体结果。
根据层序遍历的要求(从左到右,一层一层的特点),我们选用队列来实现(先进先出),试想一下,每次我们取队首元素,然后把它的值放入当前层的vector中,再把它的孩子节点放到队列里进行迭代,只要队列中还有元素我们就进行循环操作,直到到达叶子结点。
, 难点是:什么时候我们结束当前一层遍历去进行下一层呢?
答案:这里通过当前层中-队列-里元素的个数就能知道当前层进行几次循环取值操作,用for来进行循环取值。比如第二层有9,20两个元素,要进行两个for循环(注意这里有个大坑点,不要在for循环中使用q.size()因为这个队列q在for循环中是不断改变的)

主要代码(c++)

/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode(int x) : val(x), left(NULL), right(NULL) {}* };*/
class Solution {
    
public:vector<vector<int>> levelOrder(TreeNode* root) {
    if(root==NULL) return {
    };vector<vector<int>>res; // 第一个vectorqueue<TreeNode*>q;q.push(root);while(!q.empty())// 迭代的条件{
    int size = q.size(); //这里要先保存下来队列的元素个数,不然在for中会改变vector<int>currentlevel; //第二个vectorfor(int i =0; i<size; ++i) //大坑点! 你可以把size换成q.size()试试看{
    TreeNode * tmp = q.front(); // 下面是常规写法 记录每一层的值q.pop();currentlevel.push_back(tmp->val);if(tmp->left !=NULL) q.push(tmp->left);if(tmp->right !=NULL) q.push(tmp->right);} res.push_back(currentlevel);}return res;}
};

主要代码python

# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = Noneclass Solution(object):def levelOrder(self, root):""":type root: TreeNode:rtype: List[List[int]]"""queue = []res = []if root is None:return resqueue.append(root)while len(queue)!=0:eachlevel = []size = len(queue)for i in range(size):temp = queue.pop(0)eachlevel.append(temp.val)if temp.left:queue.append(temp.left)if temp.right:queue.append(temp.right)res.append(eachlevel)return res

划重点:

有关层序遍历的迭代记得用 队列哈!!!

相似题目:
解答:leetcode107 Binary Tree Level Order Traversal 2 (二叉树的层序遍历2)

原题链接:https://leetcode.com/problems/binary-tree-level-order-traversal/

  相关解决方案