LeetCode 545----Boundary of Binary Tree

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.

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].

算法分析

本题的主要难点是如何判断一个节点是在Left boundary上的、在Right boundary 上的还是一颗普通的节点。

为了将叶节点加入List中，首先想到要用 DFS 算法。但如果仅仅使用DFS算法，那么 left boundary 和 right boundary 上的节点就无法加入到List中了。因此，需要设计两个包装函数GetLeftPath 和 GetRightPath,在这两个函数中，通过判断选择递归地使用本身函数或者调用DFS算法函数。GetLeftPath函数默认传入的节点为 Left boundary 上的节点，并对该节点的 left 节点继续调用GetLeftPath函数，然后对该节点的右节点调用DFS算法函数。GetRightPath道理相同。

Java算法实现：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {

    public List<Integer> boundaryOfBinaryTree(TreeNode root) {
    	List<Integer>list=new ArrayList<>();
        if(root==null){
        	return list;
        }
        list.add(root.val);

        GetLeftPath(root.left,list);//add left boundary node and leaves node
        GetRightPath(root.right, list);// add right boundary node and leaves node

        return list;
    }

    public void GetLeftPath(TreeNode left,List<Integer>list){
    	if(left!=null){
    		list.add(left.val);// add the left boundary node
    		if(left.left!=null){
    			GetLeftPath(left.left, list);
        		DFS(left.right,list);
    		}
    		else{// according to the rule, if the node has no left subtree,then the left path goes to right
    			GetLeftPath(left.right, list);
    		}
    	}
    }

    public void GetRightPath(TreeNode right,List<Integer>list){
    	if(right!=null){
    		if(right.right!=null){
    			DFS(right.left,list);
        		GetRightPath(right.right, list);
    		}
    		else{
    			//according to the rule,if the node has no right subtree,then the right path goes to left
    			GetRightPath(right.left, list);
    		}
    		list.add(right.val);
    	}
    }

    public void DFS(TreeNode node,List<Integer>list){
    	if(node!=null){
    		if(node.left==null&&node.right==null){
    			list.add(node.val);
    		}
    		else{
    			DFS(node.left, list);
    			DFS(node.right,list);
    		}
    	}
    }
}