Binary Tree Right Side View

时间:2015-04-06 14:17:43   收藏:0   阅读:104

Given a binary tree, imagine yourself standing on the right side of it, return the values of the nodes you can see ordered from top to bottom.

For example:
Given the following binary tree,

   1            <---
 /   2     3         <---
 \       5     4       <---

You should return [1, 3, 4].

#include<iostream>
#include<vector>
#include<queue>
using namespace std;

struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
vector<int> rightSideView(TreeNode *root) {
	vector<int>Result;
	if (root == NULL)
		return Result;
	queue<pair<TreeNode*, int>>Queuenode;
	TreeNode* TmpHead = root;
	Queuenode.push(make_pair(TmpHead, 0));
	pair<TreeNode*, int>deletenode(NULL, 0);
	while (!Queuenode.empty())
	{
		TmpHead = Queuenode.front().first;
		int curlevel = Queuenode.front().second;
		if (deletenode.second != curlevel)
			Result.push_back(deletenode.first->val);
		if (TmpHead->left)
			Queuenode.push(make_pair(TmpHead->left, curlevel + 1));
		if (TmpHead->right)
			Queuenode.push(make_pair(TmpHead->right, curlevel + 1));
		deletenode = Queuenode.front();
		Queuenode.pop();
	}
	Result.push_back(deletenode.first->val);
	return Result;
}


 

评论(0
© 2014 mamicode.com 版权所有 京ICP备13008772号-2  联系我们:gaon5@hotmail.com
迷上了代码!