《剑指offer》面试题 26：树的子结构（C++实现）

题目

输入两棵二叉树A和B，判断B是不是A的子结构。

代码

#pragma once

// 题目：输入两棵二叉树A和B，判断B是不是A的子结构。

struct BinaryTreeNode
{
	double m_dbValue;
	BinaryTreeNode* m_pLeft;
	BinaryTreeNode* m_pRight;
};

/**
 * 1 递归 查找A中是否有B的根节点
 * 2 递归查找是否有左右子节点
 * 注意判断浮点数是否相等
 */

// 1 前序遍历 查找A中是否有B的根节点
bool hasSubTree(BinaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
	bool result = false;
	if (pRoot1 != nullptr || pRoot2 != nullptr)
	{
		if (doubleEqual(pRoot1->m_dbValue, pRoot2->m_dbValue))
			result = doesTree1HasTree2(pRoot1, pRoot2);
		if (!result)
			result = hasSubTree(pRoot1->m_pLeft, pRoot2);
		if (!result)
			result = hasSubTree(pRoot1->m_pRight, pRoot2);
	}

	return result;
}

// 2 递归查找是否有左右子节点
bool doesTree1HasTree2(BinaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
	if (pRoot2 == nullptr)
		return true;
	if (pRoot1 == nullptr)
		return false;

	if (!doubleEqual(pRoot1->m_dbValue, pRoot2->m_dbValue))
		return false;
	
	return doesTree1HasTree2(pRoot1->m_pLeft, pRoot2->m_pLeft) && doesTree1HasTree2(pRoot1->m_pRight, pRoot2->m_pRight);
}

bool doubleEqual(double num1, double num2)
{
	if ((num1 - num2 > -0.0000001) && (num1 - num2 < 0.0000001))
		return true;
	return false;
}