102. Binary Tree Level Order Traversal

    xiaoxiao2025-02-12  8

    Given a binary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).

    For example: Given binary tree [3,9,20,null,null,15,7],

    3 / \ 9 20 / \ 15 7

    return its level order traversal as:

    [ [3], [9,20], [15,7] ]

    题意: 给定一个二叉树,返回层次遍历的结果。结果为一个二维数组,每层的结点保存在一个数组中。

    思路: 同BFS实现层次遍历,关键在于如何分层处理——在while(!q.empty())内加一层循环,以q的size为循环长度,每次循环处理删除一个q中的结点。这样在内存循环结束时,q中保存的就都是下一层的结点。

    8ms

    class Solution { public: vector<vector<int> > levelOrder(TreeNode *root) { vector<vector<int> > res; if (root == NULL) return res; queue<TreeNode*> q; q.push(root); while (!q.empty()) { vector<int> oneLevel; //保存每层的结果 int size = q.size(); for (int i = 0; i < size; ++i) { TreeNode *node = q.front(); q.pop(); oneLevel.push_back(node->val); if (node->left) q.push(node->left); if (node->right) q.push(node->right); } res.push_back(oneLevel); } return res; } };

    此外,这道题还有递归的解法。核心在于需要一个二维数组,和一个变量level,当level递归到上一层的个数,新建一个空层,继续往里面加数字。可以想象为从左到右按列依次填充res的内容:

    8ms

    class Solution { public: vector<vector<int>> levelOrder(TreeNode* root) { vector<vector<int> > res; levelorder(root, 0, res); return res; } void levelorder(TreeNode *root, int level, vector<vector<int> > &res) { if (!root) return; if (res.size() == level) res.push_back({}); //如果递归的层数达到level,则在res中新建一个空层 res[level].push_back(root->val); if (root->left) levelorder(root->left, level + 1, res); if (root->right) levelorder(root->right, level + 1, res); } };
    转载请注明原文地址: https://ju.6miu.com/read-1296350.html
    最新回复(0)