等价二叉树

题目：

两个二叉树结构相同，且对应结点的值相同，我们称这两个二叉树等价.

 

例如：以下两个二叉树等价

        1           1

       /  \         /  \

      2   3       2   3

而以下两个则不等价

        1           1

       /  \         /  \

      2   3       3   2

以下两个也不等价

        1           1

       /  \         /  \

      2   3       2   2

 

给出两个二叉树p和q，判断它们是否等价.

 

p和q的结点数不多于100000，每个结点的数值在1和1000000000之间.

 

请为下面的Solution类实现解决上述问题的isEqual函数，函数的两个参数p和q分别代表两个二叉树的根节点，如果以p和q为根的二叉树等价则函数返回true，否则返回false.

 

/**

  Definition for a binary tree node.

  struct TreeNode {

      int val;

      TreeNode *left;

      TreeNode *right;

      TreeNode(int x) : val(x), left(NULL), right(NULL) {}

  };

 */

class Solution {

public:

       bool isEqual(TreeNode* p, TreeNode* q) {

             

    }

};

 

解题思路：

使用分治策略，几种边界条件有：

1. 两个节点都为空，返回true

2. 只有其中一个节点有两个孩子，返回false

3. 只有其中一个节点有一个孩子，返回false

4. 两个节点都只有一个孩子但不在同一边，返回false

5. 两个节点都是叶子，比较叶子的值是否相等；

将这个问题分解为子问题：

isEqual(p, q) = p->val == q->val && isEqual(p->left, q->left) && isEqual(p->right, q->right)

代码：

class Solution {
public:  
	bool isLeaf(TreeNode *p){
		return p->left == 0 && p->right == 0;
	}
	
	bool onechild(TreeNode *p){
		return (p->left == 0 && p->right != 0) || (p->left != 0 && p->right == 0);
	}
	
	bool twoch(TreeNode *p){
		return (p->left != 0 && p->right != 0);
	}
	
	bool isEqual(TreeNode* p, TreeNode* q){
		if(p == NULL && q == NULL)
			return true;
		
		if((twoch(p) && !twoch(q)) || (twoch(q) && !twoch(p)))
			return false;
			
		if((onechild(p) && !onechild(q)) || (onechild(q) && !onechild(p)))
			return false;
			
		if(onechild(p) && onechild(q)){
			if((p->left == NULL && q->right == NULL) || (p->right == NULL && q->left == NULL))
			return false;
		}
		
		if(isLeaf(p) && isLeaf(q))
			return p->val == q->val;
		
		return p->val == q->val && isEqual(p->left, q->left) && isEqual(p->right, q->right);
	}

};

不过后来发现这样写太复杂了，其实不用考虑这么多边界条件，只有（1）当两个节点都为空时返回true，（2）有且只有一个节点为空时返回false，（3）当两个节点val值相等时递归返回isEqual(p->left, q->left) && isEqual(p->right, q->right)，其余情况返回false。这样思路就十分清晰了。

代码：

class Solution {
public:  
	bool isEqual(TreeNode* p, TreeNode* q){
		if(p == NULL && q == NULL)
			return true;
		if((p == NULL && q != NULL) || (p != NULL && q == NULL))
			return false;
		if(p->val == q->val)
			return isEqual(p->left, q->left) && isEqual(p->right, q->right);
		return false;
	}
};