LeetCode #545. Boundary of Binary Tree

最新推荐文章于 2020-12-23 10:30:35 发布

原创最新推荐文章于 2020-12-23 10:30:35 发布 · 137 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

LeetCode 专栏收录该内容

625 篇文章

订阅专栏

本文介绍了一种算法，用于逆时针方向获取二叉树的边界节点值，包括左边界、叶子节点和右边界，确保不包含重复节点。通过两个示例详细解释了算法的工作原理。

题目描述：

Given a binary tree, return the values of its boundary in anti-clockwise direction starting from root. Boundary includes left boundary, leaves, and right boundary in order without duplicate nodes. (The values of the nodes may still be duplicates.)

Left boundary is defined as the path from root to the left-most node. Right boundary is defined as the path from root to the right-most node. If the root doesn't have left subtree or right subtree, then the root itself is left boundary or right boundary. Note this definition only applies to the input binary tree, and not applies to any subtrees.

The left-most node is defined as a leaf node you could reach when you always firstly travel to the left subtree if exists. If not, travel to the right subtree. Repeat until you reach a leaf node.

The right-most node is also defined by the same way with left and right exchanged.

Example 1

Input:
  1
   \
    2
   / \
  3   4

Ouput:
[1, 3, 4, 2]

Explanation:
The root doesn't have left subtree, so the root itself is left boundary.
The leaves are node 3 and 4.
The right boundary are node 1,2,4. Note the anti-clockwise direction means you should output reversed right boundary.
So order them in anti-clockwise without duplicates and we have [1,3,4,2].

Example 2

Input:
    ____1_____
   /          \
  2            3
 / \          / 
4   5        6   
   / \      / \
  7   8    9  10  
       
Ouput:
[1,2,4,7,8,9,10,6,3]

Explanation:
The left boundary are node 1,2,4. (4 is the left-most node according to definition)
The leaves are node 4,7,8,9,10.
The right boundary are node 1,3,6,10. (10 is the right-most node).
So order them in anti-clockwise without duplicate nodes we have [1,2,4,7,8,9,10,6,3].

class Solution {
public:
    vector<int> boundaryOfBinaryTree(TreeNode* root) {
        vector<int> result;
        if(root==NULL) return result;
        // 当只有根节点的时候，根节点也是叶节点，注意避免重复
        if(root->left||root->right) result.push_back(root->val);
        left_boundary(root->left,result);
        leaves(root,result);
        right_boundary(root->right,result);
        return result;
    }
    
    void left_boundary(TreeNode* root, vector<int>& result)
    {
        if(!root||(!root->left&&!root->right)) return;
        result.push_back(root->val);
        if(root->left) left_boundary(root->left, result);
        else left_boundary(root->right, result);
    }
    
    void right_boundary(TreeNode* root, vector<int>& result)
    {
        if(!root||(!root->left&&!root->right)) return;
        if(root->right) right_boundary(root->right, result);
        else right_boundary(root->left, result);
        result.push_back(root->val);
    }
    
    void leaves(TreeNode* root, vector<int>& result)
    {
        if(!root) return;
        if(!root->left&&!root->right) 
        {
            result.push_back(root->val);
            return;
        }
        leaves(root->left, result);
        leaves(root->right, result);
    }
};