No.93 - LeetCode199 - 树的右边界

最新推荐文章于 2024-09-22 20:07:10 发布

原创最新推荐文章于 2024-09-22 20:07:10 发布 · 96 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种求解树的右视图的算法，通过深度优先搜索(DFS)遍历二叉树，记录每一层最右侧节点的值，最终得到从顶部到底部的最右侧节点值列表。此算法的时间复杂度为O(N)，其中N为树中节点的数量。

所有求树的边界，最优都是O(N)的。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* now,vector<int>& ans,int level){
        if(level >= ans.size()) ans.push_back(now->val);
        else ans[level] = now->val;
        if(now->left != NULL) dfs(now->left,ans,level+1);
        if(now->right != NULL) dfs(now->right,ans,level+1);
    }
    vector<int> rightSideView(TreeNode* root) {
        vector<int> ans;
        if(root != NULL) dfs(root,ans,0);
        return ans;
    }
};