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…