求职面试问题变得更难，检查除根以外的所有内容？

Question

我在一次采访中遇到了这个问题，虽然起初看起来很容易，但当我深入研究时，对于大量的边缘情况来说，这对我来说几乎是不可能的。

要求：

想象一棵树，其中每个节点可以有 0 到 10 个孩子，每个孩子都有自己的权重，编写一个执行以下操作的函数：

1. 如果根没有子节点，则返回 -1；
2. 否则，返回树的所有子节点（根节点除外）的权重总和。

我尝试了类似的方法：

int my_func(node *root) {
    if (root->isleaf())
    {
        return -1;
    }
    int result = 0;
    for (int i = 0; i < root->num_of_children(); ++i)
    {
        result += my_func(root->child[i]);
    }
    return result;
}

但这真的很糟糕，原因有两个：

1. 我并不是所有孩子的体重相加。

2. 当我到达叶子孩子时，我正在求和 -1 而我应该加 0。

Answer 1

创建一个新函数，委托给您的函数处理根没有子节点的情况，并在计算总和后删除根的权重。您还忘记在原始函数中添加权重。您可以通过在添加其余部分之前将 result 设置为 root->weight 来解决此问题。

int sumOfWeightsExceptRoot(node* root) {
  if (!root || root->isLeaf())
    return -1;
  return my_func(root) - root->weight;
}

int my_func(node* root) {
  if (!root)
    return 0;
  int result = root->weight;
  for (int i = 0; i < root->num_of_children(); ++i) {
    result += my_func(root->child[i]);
  }
  return result;
}

迭代版本：

int sumOfWeightsExceptRoot(node* root) {
  if (!root || root->isLeaf())
    return -1;

  std::stack<node*> s;
  s.push(root);

  int result = -root->weight;
  while (!s.empty()) {
    node* ptr = s.top(); s.pop();
    result += ptr->weight;
    for (int i = 0; i < ptr->num_of_children(); i++) {
      s.push(ptr->child[i]);
    }
  }
  return result;
}