剑指Offer 从上往下打印二叉树

Posted 2020-11-21 hyouka

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

 算法知识视频讲解

题目描述

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

给出代码：

/*
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) {

    }
};

　　

借助队列进行bfs遍历

AC代码：

class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        queue<TreeNode*> que;

        vector<int> vec;
        if(root == NULL) return vec;
        que.push(root);

        while(!que.empty()) {
            TreeNode* tr;
            tr = que.front();
            que.pop();

            if(tr == NULL) continue;

            vec.push_back(tr->val);
            que.push(tr->left);
            que.push(tr->right);
        }
        return vec;
    }
};