相同的树（C++解法）

题目

给你两棵二叉树的根节点 p 和 q ，编写一个函数来检验这两棵树是否相同。

如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例 1：

输入：p = [1,2,3], q = [1,2,3]
输出：true

示例 2：

输入：p = [1,2], q = [1,null,2]
输出：false

示例 3：

输入：p = [1,2,1], q = [1,1,2]
输出：false

C++代码

#include <iostream>
using namespace std;

//定义二叉树结构体
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode() : val(0), left(nullptr), right(nullptr) {}
	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
	TreeNode(int x, TreeNode* left, TreeNode* right): val(x), left(left), right(right) {}
};

/*
* 相同的树问题
* 分四种情况，如果两棵树都为空，则返回true
* 如果只有一棵树为空，返回false
* 如果两棵树的根节点值不同，返回false
* 如果两棵树不为空且值相同，迭代判断左右子树是否相同
*/
bool isSameTree(TreeNode* p, TreeNode* q) {
	if (q == nullptr && q == nullptr) {
		return true;
	}
	else if (q == nullptr || q == nullptr) {
		return false;
	}
	else if (q->val != q->val) {
		return false;
	}
	else {
		return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
	}
}

int main() {
	TreeNode* p1 = new TreeNode(1);
	TreeNode* p2 = new TreeNode(2);
	TreeNode* p3 = new TreeNode(3);
	TreeNode* q1 = new TreeNode(1);
	TreeNode* q2 = new TreeNode(2);
	TreeNode* q3 = new TreeNode(3);
	p1->left = p2;
	p1->right = p3;
	p2->left = nullptr;
	p2->right = nullptr;
	p3->left = nullptr;
	p3->right = nullptr;
	
	q1->left = q2;
	q1->right = q3;
	q2->left = nullptr;
	q2->right = nullptr;
	q3->left = nullptr;
	q3->right = nullptr;

	TreeNode* p = p1;
	TreeNode* q = q1;
	bool ans = isSameTree(p, q);
	cout << boolalpha << ans << endl;

	delete p1, p2, p3, q1, q2, q3;
	return 0;
}

分析

相同的树问题，分四种情况，如果两棵树都为空，则返回 true，如果只有一棵树为空，返回 false，如果两棵树的根节点值不同，返回 false，如果两棵树不为空且值相同，迭代判断左右子树是否相同。