首页 > 其他 > 详细

Leetcode_Binary Tree Level Order Traversal

时间:2015-02-09 16:02:05      阅读:128      评论:0      收藏:0      [点我收藏+]

广度遍历二叉树,且要分别记录每一层的数据,方法有二。

1.迭代法

设置两个队列,其中一个队列用于存放上一层的节点,另一个队列用于存放下一层的节点。轮番使用。

class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
		queue<TreeNode *> q1,q2;
		vector<vector<int>> result;
		if(!root) return result;
		vector<int> path;
		q1.push(root);
		while(!q1.empty()||!q2.empty())
		{
			path.clear();
			while(!q1.empty())
			{
				TreeNode *node=q1.front();
				path.push_back(node->val);
				if(node->left)
					q2.push(node->left);
				if(node->right)
					q2.push(node->right);
				q1.pop();
			}
			if(!path.empty())
				result.push_back(path);
			path.clear();
			while(!q2.empty())
			{
				TreeNode *node=q2.front();
				path.push_back(node->val);
				if(node->left)
					q1.push(node->left);
				if(node->right)
					q1.push(node->right);
				q2.pop();
			}
			if(!path.empty())
				result.push_back(path);
		}
       return result; 
    }
};
2.递归法(来源于网上)

class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
		vector<vector<int>> result;
		traverse(root,1,result);
		return result;
        
    }
private:
	void traverse(TreeNode *root,int level,vector<vector<int>> &result)
	{
		if(!root) return;
		if(level>result.size())
			result.push_back(vector<int>());
		result[level-1].push_back(root->val);
		traverse(root->left,level+1,result);
		traverse(root->right,level+1,result);
	}
};



Leetcode_Binary Tree Level Order Traversal

原文:http://blog.csdn.net/yinqiaohua/article/details/43671067

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!