【LeetCode】102. Binary Tree Level Order Traversal(二叉树的层级遍历 完整代码)

题目如下:

这道题是求二叉树的层次遍历,也即宽度优先遍历,并将结果以二维数组形式输出。

因为树的每一层节点数可能不同,所以采用二维不定长的vector向量储存

采用递归方式求解,找完一层找下一层

输入在下面的程序前需要处理一下,采用先序建树,遇到叶子节点时将左右孩子置零,例如题目中的树,输入时输入:

3 9 0 0 20 15 0 0 7 0 0

完整代码如下: 

#include<bits/stdc++.h>

using namespace std;

vector< vector<int> > result;

typedef struct TreeNode{
	struct TreeNode* left;
	struct TreeNode* right;
	int elem;
};

TreeNode* BinaryTreeCreate() //先序建树
{
	int x;
	TreeNode *p;
	cin >> x;
	if(x == 0)
	{
		p = NULL;
	} 
	else
	{
		p = (TreeNode*)malloc(sizeof(TreeNode));
		p->left = BinaryTreeCreate();
		p->elem = x;
		p->right = BinaryTreeCreate();
	}
	return p;
}

int DepthOfTree(TreeNode* root) //求树的深度
{
	if(!root) return 0;
	int leftDepth = DepthOfTree(root->left);
	int rightDepth =  DepthOfTree(root->right);
	return leftDepth > rightDepth ? leftDepth+1 : rightDepth+1;
}



void LevelOrderTraversal(TreeNode* root, int level) //层级遍历
{

	if(!root) return;
	if(level > DepthOfTree(root)+1) return;
	if(result.size() == level)
        result.push_back(vector<int>());

	result[level].push_back(root->elem); //同一层的存到同一维度
	LevelOrderTraversal(root->left, level+1);
	LevelOrderTraversal(root->right, level+1);
} 

int main()
{
	TreeNode* inorder = BinaryTreeCreate();
	LevelOrderTraversal(inorder, 0);
	for(int i = 0; i < DepthOfTree(inorder); i++)
	{                   
		for(int j = 0; j < result[i].size(); j++)
		{
			cout << result[i][j] << ' ';
		}
		cout<<endl;
	}	
	return 0;
}


 运行结果如下:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

McGregorWwww

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值