剑指offer系列22之从上往下打印二叉树

最新推荐文章于 2024-07-14 14:13:59 发布

Cheng Qian

最新推荐文章于 2024-07-14 14:13:59 发布

阅读量91

点赞数

分类专栏：剑指offer coding

本文链接：https://blog.csdn.net/xiayangmian4130/article/details/88049301

版权

coding 同时被 2 个专栏收录

35 篇文章 0 订阅

订阅专栏

剑指offer

34 篇文章 0 订阅

订阅专栏

要求

时间限制：1秒
空间限制：32768K
热度指数：284958

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

思路

利用一个队列进行二叉树的层次遍历。

python实现

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回从上到下每个节点值列表，例：[1,2,3]
    def PrintFromTopToBottom(self, root):
        # write code here
        
        res = []
        if not root:
            return res
        queue = []
        queue.append(root)
        while len(queue)>0:
            temp = queue.pop(0)
            res.append(temp.val)
            if temp.left: 
                queue.append(temp.left)
            if temp.right:
                queue.append(temp.right)
        return res

运行时间：29ms
占用内存：5740k

C++实现

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        vector<int> nodes;
        if (!root)
            return nodes;
        queue<TreeNode*> qu;
        qu.push(root);
        TreeNode* temp;
        
        while (qu.size()){
            temp = qu.front();
            nodes.push_back(temp->val);
            qu.pop();
            if (temp->left) qu.push(temp->left);
            if (temp->right) qu.push(temp->right);
        }
        return nodes;
    }
};