【刷题】199. 二叉树的右视图——给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。

题目: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之前的检查外,算法没有别的改动。

猜你喜欢

转载自blog.csdn.net/m0_46613023/article/details/114022107