《剑指offer》32--从上往下打印二叉树[C++][Java]

最新推荐文章于 2022-06-02 08:42:58 发布

贫道绝缘子

最新推荐文章于 2022-06-02 08:42:58 发布

阅读量147

点赞数

分类专栏：剑指offer 文章标签：剑指offer 二叉树队列 BFS

本文链接：https://blog.csdn.net/qq_15711195/article/details/96497429

版权

剑指offer 专栏收录该内容

82 篇文章 1 订阅

订阅专栏

1 不分行

从上往下打印二叉树_牛客题霸_牛客网【牛客题霸】收集各企业高频校招笔面试题目，配有官方题解，在线进行百度阿里腾讯网易等互联网名企笔试面试模拟考试练习,和牛人一起讨论经典试题,全面提升你的技术能力https://www.nowcoder.com/practice/7fe2212963db4790b57431d9ed259701?tpId=13&tqId=23280&ru=/practice/1624bc35a45c42c0bc17d17fa0cba788&qru=/ta/coding-interviews/question-ranking

题目描述

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

解题思路

使用队列来进行层次遍历。

不需要使用两个队列分别存储当前层的节点和下一层的节点，因为在开始遍历一层的节点时，当前队列中的节点数就是当前层的节点数，只要控制遍历这么多节点数，就能保证这次遍历的都是当前层的节点。

【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> res;
        queue<TreeNode*> q;
        TreeNode* tmp;
        if(root) q.push(root);
        while(!q.empty()) {
            tmp = q.front();
            res.push_back(tmp->val);
            if(tmp->left) q.push(tmp->left);
            if(tmp->right) q.push(tmp->right);
            q.pop();
        }
        return res;
    }
};

【Java解法】

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> res = new ArrayList<>();
        if (root == null) {return res;}
        Queue<TreeNode> q = new LinkedList<>();
        TreeNode p;
        q.offer(root);
        while (!q.isEmpty()) {
            p = q.poll();
            if (p.left != null) {q.offer(p.left);}
            if (p.right != null) {q.offer(p.right);}
            res.add(p.val);
        }
        return res;
    }
}