填充樹的所有節點的數據字段

Question

二叉樹的節點有兩個指針，'left'和'right'，以及兩個數據字段'leftcount'和'rightcount'。 'leftcount'指定節點左側子樹中的節點數量，'rightcount'指定節點右側子樹中的節點數量。編寫一個算法來填充樹的所有節點的數據字段。

我在接受採訪時被問到這個問題。我想出了一個基於樹的後序遍歷的解決方案。有人可以請指導我這一點。

Answer 1

這應該工作（我相信）：

int populateCounters(Node* node) { 
    if(node == NULL) return 0; 
    node->leftCount = populateCounters(node->left); 
    node->rightCount = populateCounters(node->right); 
    return node->leftCount + node->rightCount + 1; 
}