二叉樹中相距最遠的兩個節點之間的距離

朗潤發表於2014-08-18

遞迴解法:

(1)如果二叉樹為空,返回0,同時記錄左子樹和右子樹的深度,都為0

(2)如果二叉樹不為空,最大距離要麼是左子樹中的最大距離,要麼是右子樹中的最大距離,要麼是左子樹節點中到根節點的最大距離+右子樹節點中到根節點的最大距離,同時記錄左子樹和右子樹節點中到根節點的最大距離。



intGetMaxDistance(BinaryTreeNode* pRoot,int& maxLeft,int& maxRight)

{

    //maxLeft, 左子樹中的節點距離根節點的最遠距離

    //maxRight, 右子樹中的節點距離根節點的最遠距離

    if(pRoot== NULL)

    {

        maxLeft =0;

        maxRight =0;

        return0;

    }

    int maxLL, maxLR, maxRL, maxRR;

    int maxDistLeft, maxDistRight;

    if(pRoot->m_pLeft!= NULL)

    {

        maxDistLeft=GetMaxDistance(pRoot->m_pLeft, maxLL, maxLR);

        maxLeft = max(maxLL, maxLR)+1;

    }

    else

    {

        maxDistLeft=0;

        maxLeft =0;

    }

    if(pRoot->m_pRight!= NULL)

    {

        maxDistRight=GetMaxDistance(pRoot->m_pRight, maxRL, maxRR);

        maxRight = max(maxRL, maxRR)+1;

    }

    else

    {

        maxDistRight=0;

        maxRight =0;

    }

    return max(max(maxDistLeft,maxDistRight), maxLeft+maxRight);

}


相關文章