LeetCode 545. 二叉树的边界(前序+后序)*

文章目录

    • 1. 题目
    • 2. 解题

1. 题目

给定一棵二叉树,以逆时针顺序从开始返回其边界。
边界按顺序包括左边界叶子结点右边界而不包括重复的结点。 (结点的值可能重复)

左边界的定义是从根到最左侧结点的路径。
右边界的定义是从根到最右侧结点的路径。
若根没有左子树或右子树,则根自身就是左边界或右边界。
注意该定义只对输入的二叉树有效,而对子树无效。

最左侧结点的定义是:在左子树存在时总是优先访问,
如果不存在左子树则访问右子树。
重复以上操作,首先抵达的结点就是最左侧结点。

最右侧结点的定义方式相同,只是将左替换成右。

示例 1
输入:
  1
   \
    2
   / \
  3   4
输出:
[1, 3, 4, 2]
解析:
根不存在左子树,故根自身即为左边界。
叶子结点是34。
右边界是124。注意逆时针顺序输出需要你输出时调整右边界顺序。
以逆时针顺序无重复地排列边界,得到答案[1,3,4,2]。

示例 2
输入:
    ____1_____
   /          \
  2            3
 / \          / 
4   5        6   
   / \      / \
  7   8    9  10  
       
输出:
[1,2,4,7,8,9,10,6,3]

解析:
左边界是结点1,2,4(根据定义,4是最左侧结点)
叶子结点是结点4,7,8,9,10。
右边界是结点1,3,6,10(10是最右侧结点)
以逆时针顺序无重复地排列边界,得到答案 [1,2,4,7,8,9,10,6,3]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/boundary-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

class Solution {
	vector<int> ans;
public:
    vector<int> boundaryOfBinaryTree(TreeNode* root) {
    	if(!root) return {};
    	ans.push_back(root->val);
    	dfs(root->left, -1);
    	dfs(root->right, 1);
    	return ans;
    }
    void dfs(TreeNode* root, int dir)
    {
    	if(!root) return;
    	if(dir == -1)//左边界
    	{
    		ans.push_back(root->val);//前序遍历
    		if(root->left)
    		{	
    			dfs(root->left,-1);//有左节点,进入左节点找
    			dfs(root->right, 0);//舍弃右节点
    		}
    		else
    			dfs(root->right,-1);//没有左节点,当成左边界
    	}
    	else if(dir == 1)//右边界
    	{
    		// ans.push_back(root->val);//写这里错了
    		if(root->right)
    		{
    			dfs(root->left,0);
    			dfs(root->right, 1);
    		}
    		else
    			dfs(root->left, 1);
    		ans.push_back(root->val);//应该是后序遍历
    	}
    	else//dir==0,非左右边界,找叶子节点
    	{
    		if(!root->left && !root->right)
    			ans.push_back(root->val);
    		dfs(root->left, 0);
    		dfs(root->right, 0);
    	}
    }
};

12 ms 21 MB


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!

你可能感兴趣的:(LeetCode)