iOS數據結構與算法實戰 二叉樹的算法實戰 Binary Tree Paths

來源

公衆號：人魔七七

Github：github.com/renmoqiqi/1…

題目大意

Given a binary tree, return all root-to-leaf paths.

For example, given the following binary tree:


   1
 /   \
2     3
 \
  5
  
All root-to-leaf paths are:

["1->2->5", "1->3"]
複製代碼

靈感思路

給咱們一個二叉樹，讓咱們返回全部根到葉節點的路徑。咱們能夠採用遞歸的思路，不停的DFS到葉結點，若是遇到葉結點的時候，那麼此時一條完整的路徑已經造成，咱們加上當前的葉結點後變成的完整路徑放到數組中。

須要注意的是對空節點的判斷，以及遞歸函數回溯時候對一些對象的影響。

主要代碼

- (void)printPathsRecurTreeNode:(DSTreeNode *)treeNode path:(NSString *)path results:(NSMutableArray <NSString *>*)results
{

    //1
    if (treeNode == nil) {
        return;
    }
    //2
    if (treeNode.leftChild == nil && treeNode.rightChild == nil)
    {
        NSString *resultsStr = [NSString stringWithFormat:@"%@%@",path,treeNode.object];
        [results addObject:resultsStr];

    }
    else
    {
        //3
        if (treeNode.leftChild != nil)
        {
            NSString *resultsStr = [NSString stringWithFormat:@"%@%@",path,[NSString stringWithFormat:@"%@->",treeNode.object]];

            [self printPathsRecurTreeNode:treeNode.leftChild path:resultsStr results:results];
        }
        //4
        if (treeNode.rightChild != nil )
        {
            NSString *resultsStr = [NSString stringWithFormat:@"%@%@",path,[NSString stringWithFormat:@"%@->",treeNode.object]];
            [self printPathsRecurTreeNode:treeNode.rightChild path:resultsStr results:results];
        }
    }
    

}
複製代碼

代碼解釋

1. 若是節點是空則返回。
2. 若是當前節點是葉子節點則把這個完整路徑加到數組裏。
3. 若是當前節點存在左孩子節點，則繼續DFS直到葉子節點。
4. 若是當前節點存在右孩子節點，則繼續DFS直到葉子節點。

GitHubDemo：

github.com/renmoqiqi/1…