题目:199. 二叉树的右视图
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
解答:
方法一:深度优先搜索
思路:
我们对树进行深度优先搜索,在搜索过程中,我们总是先访问右子树。那么对于每一层来说,我们在这层见到的第一个结点一定是最右边的结点。
算法:
这样一来,我们可以存储在每个深度访问的第一个结点,一旦我们知道了树的层数,就可以得到最终的结果数组。
// 深度优先搜索 depth-first search
typedef struct TreeNode TreeNode;
// 通过“深度优先搜索”得到returnSize,max_depth
// d用来更新max_depth的值
void dfs(TreeNode* root, int d, int* max_depth, int* ans, int* returnSize)
{
if (root==NULL)
return;
if (d > *max_depth)
{
*(ans + (*returnSize)++) = root->val;
*max_depth = d;
}
dfs(root->right, d + 1, max_depth, ans, returnSize);
dfs(root->left, d + 1, max_depth, ans, returnSize);
}
int* rightSideView(TreeNode* root, int* returnSize)
{
int* ans = malloc(sizeof(int) * 1e4);
*returnSize = 0;
int max_depth = -1;
dfs(root, 0, &max_depth, ans, returnSize);
return ans;
}
方法二:广度优先搜索
思路:
我们可以对二叉树进行层次遍历,那么对于每层来说,最右边的结点一定是最后被遍历到的。二叉树的层次遍历可以用广度优先搜索实现。
算法:
执行广度优先搜索,左结点排在右结点之前,这样,我们对每一层都从左到右访问。因此,只保留每个深度最后访问的结点,我们就可以在遍历完整棵树后得到每个深度最右的结点。除了将栈改成队列,并去除了rightmost_value_at_depth之前的检查外,算法没有别的改动。