二叉樹中最低級別的所有葉節點的總和

Question

如何在二叉樹中以最小級別計算所有葉節點的總和。 如果不存在樹，則應返回-1。

例：

對於上述二叉樹，返回100（40 + 60）

（圖片來源： GeeksForGeeks ）

Answer 1

f(node, level):
   if node is null then
       return { Inf, -1 }
   if isLeaf(node) then 
       return { level, node.value }
   fleft <- f(node.left, level + 1)
   fright <- f(node.right, level + 1)
   fnode <- { min(fleft.first, fright.first), 0 }
   if fnode.first = fleft.first then
       fnode.second <- fnode.second + fleft.second
   if fnode.first = fright.first then
       fnode.second <- fnode.second + fright.second
   return fnode

函數返回一對值，其中first是最小葉子級別， second是該級別上葉子元素的總和。

Answer 2

您也可以使用地圖， minLeafSum是驅動函數，邏輯很簡單。 只需檢查代碼即可。

void getmin(map<int,vector<int>> &m,Node* root,int level)
{
    if(root==NULL)
        return;
    if(root->left==NULL && root->right==NULL)
    {
        m[level].push_back(root->data);
        return;
    }
    if(root->left!=NULL)
        getmin(m,root->left,level+1);
    if(root->right!=NULL)
        getmin(m,root->right,level+1);
}
int minLeafSum(Node* root)
{
    int ans=0;

    map<int,vector<int>> m;
    getmin(m,root,0);
    auto it = m.begin();
    for(int i=0; i< it->second.size() ;i++)
        ans += it->second[i];

    return ans ;
}