有没有更好的方法来编写这个 if-else 块？

Question

我正在为二叉搜索树编写一些代码，其中每个节点都有一个left 和一个right 子指针。在我的delete 函数中，我有以下代码。

if (!current.left && !current.right) {
  if (parent.left === current) {
    parent.left = null;
  } else {
    parent.right = null;
  }
} else {
  if (parent.left === current) {
    parent.left = current.left ? current.left : current.right;
  } else {
    parent.right = current.left ? current.left : current.right;
  }
}

我有两次使用以下语法的块：

if (someCondition) {
  parent.left = x;
} else {
  parent.right = x;
}

有没有更简洁的方法来写这个（可能是单行）？我不确定是否可以在这里使用三元运算符语法，因为我在if 块中有parent.left = x，在else 块中有parent.right = x。我不喜欢这个函数中使用的所有这些if-else 块。

这里是完整的代码sn-p。

const getInOrderSuccessor = (root, node) => {
  let successorParent = null;
  let successor = null;
  let previous = null;
  let current = root;

  while (current) {
    if (node.value < current.value) {
      successorParent = previous;
      successor = current;
      previous = current;
      current = current.left;
    } else {
      previous = current;
      current = current.right;
    }
  }
  return [successor, successorParent];
};

const deleteNode = (root, value) => {
  let current = root;
  let parent = null;
  while (current) {
    if (value === current.value) break;
    parent = current;
    current = value < current.value ? current.left : current.right;
  }

  // If 2 child, deal with that first
  if (current.left && current.right) {
    const [successor, successorParent] = getInOrderSuccessor(root, current);
    current.value = successor.value;
    current = successor;
    parent = successorParent;
  }

  if (!current.left && !current.right) {
    if (parent.left === current) {
      parent.left = null;
    } else {
      parent.right = null;
    }
  } else {
    if (parent.left === current) {
      parent.left = current.left ? current.left : current.right;
    } else {
      parent.right = current.left ? current.left : current.right;
    }
  }
};

Answer 1

请试试这个。

当你有类似z = y?y:x 的东西时，你应该用|| 这样的z = y || x 运算符替换它 也可以写常用条件。

if (parent.left === current) {
  parent.left = (!current.left && !current.right) ? null : (current.left || current.right);
} else {
  parent.right = !(!current.left && !current.right) ? null : (current.left || current.right);
}

Answer 2

这是一个满足您标准的选项，尽管在风格方面我更喜欢您写的比这更好的:-)

function chooseWhichToMakeNull(object, condition) {
  const field = condition ? 'left' : 'right';
  object[field] = null;
}

const obj = { left: 5, right: 10 };
chooseWhichToMakeNull(obj, true);
console.log(obj);

Answer 3

除了使用parent.left = x 和parent.right = x 之外，您还可以将属性名称存储在一个变量中，然后改用parent[direction] = x。

对于您的特定情况，在 if 和 else 块中进行 parent.left === current 检查也有帮助。这意味着我们可以将它移到条件之外。

const direction = parent.left === current ? "left" : "right";
if (!current.left && !current.right) {
  parent[direction] = null;
} else {
  parent[direction] = current.left ? current.left : current.right;
}

您可以通过重新设计逻辑来进一步简化此操作。

if (!a && !b) {
  variable = null;
} else {
  variable = a ? a : b;
}

可以改成：

if (a) {
  variable = a;
} else if (b) {
  variable = b;
} else {
  variable = null;
}

也可以写成：

variable = a || b || null;

导致以下解决方案：

const direction = parent.left === current ? "left" : "right";
parent[direction] = current.left || current.right || null;