【面试题】从上到下打印二叉树||

最新推荐文章于 2024-01-01 12:05:48 发布

头号理想

最新推荐文章于 2024-01-01 12:05:48 发布

阅读量131

点赞数

文章标签：二叉树 dfs 算法面试 stack

本文链接：https://blog.csdn.net/weixin_44302602/article/details/112105192

版权

算法专栏收录该内容

59 篇文章 0 订阅

订阅专栏

题目

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

在这里插入图片描述

层序遍历（广度优先）

这个题和【面试题】从上到下打印二叉树（二叉树的层序遍历）解法很相似
只是输出多了一个层

vector<vector<int>> levelOrder(TreeNode* root) {
	vector<vector<int>> ans;
	if (root == NULL) {
		return ans;
	}
	queue<TreeNode* > q;
	q.push(root);


	while (!q.empty()) {
		vector<int> temp;
		int count = q.size();

		while (count--) {
			TreeNode* t = q.front();
			temp.push_back(t->val);
			q .pop();
			if (t->left != NULL) {
				q.push(t->left);
			}
			if (t->right != NULL) {
				q.push(t->right);
			}
		}
		ans.push_back(temp);
	}
	return ans;
}

深度优先

我们按照深度遍历的方法遍历二叉树
然后深度每次加一
我们对应的数组就将其val填入数组中

 vector<vector<int>>ans;
void dfs(TreeNode* root, int depth) {
    if (!root) return;
    if (depth >= ans.size())ans.resize(depth + 1);
    ans[depth].push_back(root->val);
    dfs(root->left, depth + 1);
    dfs(root->right, depth + 1);
}
vector<vector<int>> levelOrder(TreeNode* root) {
    if (!root)return{};
    dfs(root, 0);
    return ans;
}