深圳幻海软件技术有限公司 欢迎您!

一篇讲明白对称的二叉树

2023-02-28

实现思路二叉树的镜像中我们知道了此问题的解决方案是前序遍历,那么我们可以修改下前序遍历算法,父节点遍历后,先遍历它的右子节点,再遍历它的左子节点,我们把这种算法称为:对称前序遍历。如下图所示的两棵树,我们分别列举下两种遍历的结果:树A:前序遍历:8,6,5,7,6,7,5对称前序遍历:8,6,5,7

实现思路

二叉树的镜像中我们知道了此问题的解决方案是前序遍历,那么我们可以修改下前序遍历算法,父节点遍历后,先遍历它的右子节点,再遍历它的左子节点,我们把这种算法称为:对称前序遍历。

如下图所示的两棵树,我们分别列举下两种遍历的结果:

树A:

  • 前序遍历:8, 6, 5, 7, 6, 7, 5
  • 对称前序遍历:8, 6, 5, 7, 6, 7, 5

树B:

  • 前序遍历:8, 6, 5, 7, 9, 7, 5
  • 对称前序遍历:8, 9, 5, 7, 6, 7, 5

经过对比后,我们发现树A的两种遍历方法得到的结果是一样的,那么它就是对称的;树B的结果不同,它就不是对称的。

如果有一颗不完全二叉树,它的所有节点都相同,他是对称的吗?

针对于这种情况,我们就需要将它缺省的null节点进行补齐了,补齐后的两种遍历结果为:

  • 前序遍历:7, 7, 7, null, null, 7, null, null, 7, 7, null, null, null
  • 对称前序遍历:7, 7, null, 7, null, null, 7, 7, null, null, 7, null, null

对比两个结果后,我们发现并不一样,那么它就不是对称的。

实现代码

  • 有了思路后,接下来我们看下代码实现,如下所示:
  • 从树的根节点出发,递归比对它的左子节点和右子节点

QQi">比对过程中:

二者都到达叶子节点,代表这棵树是对称的

任意一方到达叶子结点,代表这棵树不对称

节点值不同,这棵树不对称

export function SymmetricBinaryTree(node: BinaryTreeNode | null): boolean {
  return isSymmetrical(node, node);
}

function isSymmetrical(
  node: BinaryTreeNode | null | undefined,
  cloneNode: BinaryTreeNode | null | undefined
): boolean {
  // 到达叶子节点,两者都为nul代表节点相同
  if (node == null && cloneNode == null) {
    return true;
  }
  // 任意一方到达叶子节点,代表节点不同
  if (node == null || cloneNode == null) {
    return false;
  }
  // 节点值不同
  if (node.key != cloneNode.key) {
    return false;
  }

  // 分别比对树的左子节点和右子节点
  return (
    isSymmetrical(node.left, cloneNode.right) &&
    isSymmetrical(node.right, cloneNode.left)
  );
}
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.
  • 22.
  • 23.
  • 24.
  • 25.
  • 26.
  • 27.

接下来,我们以上个章节列举的例子为例,将其带入上述代码,验证下能否正确判断,如下所示:

const tree: BinaryTreeNode = {
  key: 8,
  left: {
    key: 6,
    left: { key: 5 },
    right: { key: 7 }
  },
  right: { key: 6, left: { key: 7 }, right: { key: 5 } }
};
const isSymmetric = SymmetricBinaryTree(tree);
console.log(tree, "是否为对称二叉树: ", isSymmetric);
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.

示例代码

本文所用代码完整版请移步:

  • SymmetricBinaryTree
  • symmetricBinaryTree-test.ts