首页 > 其他 > 详细

2021.2.2 刷题(层序遍历)

时间:2021-02-02 22:11:13      阅读:26      评论:0      收藏:0      [点我收藏+]

题目链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
题目描述:给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

示例:
二叉树:[3,9,20,null,null,15,7],
技术分享图片

分析:
二叉树的层序遍历类似于图的广度优先搜索。
我们可以用一种巧妙的方法修改 BFS:

代码:

/**
 * 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) {
        queue<TreeNode*>  que;
        vector<vector<int>> result;

        if(root != NULL)
            que.push(root);
        while(!que.empty())
        {
            vector<int> vec;  //保存当前层元素
            int currentLevelSize = que.size(); //记录当前层的大小
            for(int i = 0; i < currentLevelSize; i++)
            {
                TreeNode *node = que.front();
                que.pop();
                vec.push_back(node->val);
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
           result.push_back(vec);
        }
        return result;
    }
};

2021.2.2 刷题(层序遍历)

原文:https://www.cnblogs.com/ZigHello/p/14364204.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!