100. Same Tree（leetcode）

Given two binary trees, write a function to check if they are equal or not.

Two binary trees are considered equal if they are structurally identical and the nodes have the same value.

题意大概是：给定两个二叉树，编写一个函数检查它们是否相等。如果结构相同且节点具有相同的值，则两个二叉树被认为是相等的。

题目给出了树的节点类

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

我们最容易想到的，是用栈来实现：

 import java.util.*;
 public class Solution {
     static Stack<TreeNode> stack1=new Stack<TreeNode>();
     static Stack<TreeNode> stack2=new Stack<TreeNode>();
     public void toStack(TreeNode p,TreeNode q)
     {
         while(p.left!=null)
         {stack1.push(p);
          p=p.left;   }
          while(q.left!=null)
         {stack2.push(q);
          q=q.left;    }
         stack1.push(p);
         stack2.push(q);
 
     }
     public boolean isSameTree(TreeNode p, TreeNode q) {
         if((p==null&&q==null))
             return true;
         if(p==null||q==null)
             return false;
         while(!stack1.isEmpty())
             stack1.pop();
         while(!stack2.isEmpty())
             stack2.pop();
         TreeNode a,b;
         toStack(p,q);
             while(!stack1.isEmpty()&&!stack2.isEmpty())
             {
                 a=stack1.pop();
                 b=stack2.pop();
                   if(a.val!=b.val)//判断该节点的值是否相等
                      return false;  
 
                     if((a.right!=null)&&(b.right!=null))//都有右孩子，让他们入栈
                             toStack(a.right,b.right);
 
                     else if((a.right==null&&b.right!=null)||a.right!=null&&b.right==null)//有一个有右孩子，另外一个没有右孩子
                         return false;
             }
 
         if(stack1.isEmpty()&&stack2.isEmpty())
               return true;
         return false;
     }
 
 }

当然，还有更简单的，用递归搞定：

 public boolean isSameTree(TreeNode p, TreeNode q) {
     if(p == null && q == null) return true;
     if(p == null || q == null) return false;
     if(p.val == q.val)
         return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
     return false;
 }

更新：

上面两行

     if(p == null && q == null) return true;
     if(p == null || q == null) return false;

可以用一行代替：

if (p == NULL || q == NULL) return (p == q);