天天看點

二叉樹(4)----求二叉樹深度,遞歸和非遞歸

1、二叉樹定義

typedef struct BTreeNodeElement_t_ {
    void *data;
} BTreeNodeElement_t;

typedef struct BTreeNode_t_ {
    BTreeNodeElement_t *m_pElemt;
    struct BTreeNode_t_    *m_pLeft;
    struct BTreeNode_t_    *m_pRight;
} BTreeNode_t;
           

2、求二叉樹深度

定義:對任意一個子樹的根節點來說,它的深度=左右子樹深度的最大值+1

(1)遞歸實作

如果根節點為NULL,則深度為0

如果根節點不為NULL,則深度=左右子樹的深度的最大值+1

int  GetBTreeDepth( BTreeNode_t *pRoot)
{
    if( pRoot == NULL )
        return 0;

    int lDepth = GetBTreeDepth( pRoot->m_pLeft);
    int rDepth = GetBTreeDepth( pRoot->m_pRight);

    return ((( lDepth > rDepth )? lDepth: rDepth) + 1 );        
}
           

(2)非遞歸實作

借助隊列,在進行按層周遊時,記錄周遊的層數即可。

int GetBTreeDepth( BTreeNode_t *pRoot){
    if( pRoot == NULL )
        return 0;

    queue< BTreeNode_t *> que;
    que.push( pRoot );
    int depth = 0;
    while( !que.empty() ){
        ++depth;
        int curLevelNodesTotal = que.size();
        int cnt = 0;
        while( cnt < curLevelNodesTotal ){
            ++cnt;
            pRoot = que.front();
            que.pop();
            if( pRoot->m_pLeft )
                que.push( pRoot->m_pLeft);
            if( pRoot->m_pRight)
                que.push( pRoot->m_pRight);
        }
    }

    return;
}