每日一解 相同的树

题目 相同的树

给定两个二叉树，编写一个函数来检验它们是否相同。
如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例 1:

输入:       1         1
          / \       / \
         2   3     2   3

        [1,2,3],   [1,2,3]

输出: true

示例 2:

输入:      1          1
          /           \
         2             2

        [1,2],     [1,null,2]

输出: false

示例 3:

输入:       1         1
          / \       / \
         2   1     1   2

        [1,2,1],   [1,1,2]

输出: false

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/same-tree

思路

毕竟是简单题目，读清楚题目的时候基本心里就有数了。其实只要采用先序或者中序后序等等方式遍历两棵二叉树，一旦出现不同就直接抛出false即可。但是因为要抛出false，因此采用递归函数来解题就不太好立刻返回答案了。还好凡是递归均可写成循环，再用两个栈分别存储当前遍历的节点即可。

代码实现

因为很快就把代码敲出来了，那么直接上代码：

class Solution {
public:
	bool isSameTree(TreeNode* p, TreeNode* q) {
		stack<TreeNode*> p_data, q_data;
		p_data.push(p);
		q_data.push(q);
		while (!p_data.empty() && !q_data.empty()) {
			if (p_data.top() == NULL) {
				if (q_data.top() != NULL) {
					return false;
				}
				p_data.pop();
				q_data.pop();
			}
			else if (q_data.top() == NULL) {
				if (p_data.top() != NULL) {
					return false;
				}
				p_data.pop();
				q_data.pop();
			}
			else {
				if (p_data.top()->val != q_data.top()->val) {
					return false;
				}
				TreeNode* a = p_data.top();
				p_data.pop();
				p_data.push(a->right);
				p_data.push(a->left);
				a = q_data.top();
				q_data.pop();
				q_data.push(a->right);
				q_data.push(a->left);
			}
		}
		if (!p_data.empty() || !q_data.empty()) {
			return false;
		}
		return true;
	}
};

也许因为题解很多或者水平还不高吧，虽然空间成绩不咋地但是时间成绩是100，我个人还算满意的。