TypeScript编译器在if分支中无法推断root2非空类型求助

我明白你遇到的困惑了——你的代码逻辑确实覆盖了所有可能的分支情况，能通过LeetCode测试也证明了这一点，但TypeScript的类型检查器有时候需要更明确的信号，才能确认某个变量一定是非空的。

问题原因 #

你的最后一个return语句，虽然前面的三个if已经排除了：

• !root1 && !root2（返回null）
• root1 && root2（合并节点返回）
• root1 && !root2（返回root1的复制）

理论上剩下的唯一情况就是!root1 && root2，但TypeScript的控制流分析并没有智能到能完全推断出这一步。它仍然认为root2可能为null，所以当你访问root2.val、root2.left和root2.right时，就会抛出类型错误。

几种解决办法 #

1. 非空断言（快速解决）

直接给root2加上非空断言符号!，告诉TypeScript：“我保证这里的root2绝对不是null”。这种方式改动最小，适合你当前的代码结构：

const mergeTrees = (
  root1: TreeNode | null,
  root2: TreeNode | null
): TreeNode | null => {
  if (!root1 && !root2) return null;
  if (root1 && root2)
    return new TreeNode(
      root1.val + root2.val,
      mergeTrees(root1.left, root2.left),
      mergeTrees(root1.right, root2.right)
    );
  if (root1 && !root2)
    return new TreeNode(
      root1.val,
      mergeTrees(root1.left, null),
      mergeTrees(root1.right, null)
    );
  // 加上非空断言
  return new TreeNode(
    root2!.val,
    mergeTrees(null, root2!.left),
    mergeTrees(null, root2!.right)
  );
}

2. 明确最后一个分支的条件判断

把最后一个默认返回改成明确的if判断，让TypeScript清晰知道进入这个分支时root2一定非空：

const mergeTrees = (
  root1: TreeNode | null,
  root2: TreeNode | null
): TreeNode | null => {
  if (!root1 && !root2) return null;
  if (root1 && root2)
    return new TreeNode(
      root1.val + root2.val,
      mergeTrees(root1.left, root2.left),
      mergeTrees(root1.right, root2.right)
    );
  if (root1 && !root2)
    return new TreeNode(
      root1.val,
      mergeTrees(root1.left, null),
      mergeTrees(root1.right, null)
    );
  // 明确判断当前分支的情况
  if (!root1 && root2)
    return new TreeNode(
      root2.val,
      mergeTrees(null, root2.left),
      mergeTrees(null, root2.right)
    );
  // 兜底返回（实际永远不会走到，只是让TypeScript的类型检查更严谨）
  return null;
}

3. 简化代码逻辑（更优雅的写法）

用可选链?.和空值合并??来简化代码，这种写法不仅更简洁，还能让TypeScript自动处理类型推断，从根源上避免报错：

const mergeTrees = (
  root1: TreeNode | null,
  root2: TreeNode | null
): TreeNode | null => {
  if (!root1 && !root2) return null;
  
  // 用空值合并处理节点不存在的情况，默认取0
  const val = (root1?.val ?? 0) + (root2?.val ?? 0);
  // 可选链获取左右子树，不存在则传null
  const left = mergeTrees(root1?.left ?? null, root2?.left ?? null);
  const right = mergeTrees(root1?.right ?? null, root2?.right ?? null);
  
  return new TreeNode(val, left, right);
}

这个版本的逻辑和你原来的代码完全一致，但TypeScript能完美识别所有变量的类型，不会再有报错。

内容的提问来源于stack exchange，提问作者Seuta K