LeetCode题库解答与分析——#100. 相同的树SameTree

给定两个二叉树，写一个函数来检查它们是否相同。

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

 

示例 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

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

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

Example 1:

Input:     1         1
          / \       / \
         2   3     2   3

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

Output: true

Example 2:

Input:     1         1
          /           \
         2             2

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

Output: false

Example 3:

Input:     1         1
          / \       / \
         2   1     1   2

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

Output: false

个人思路：

同时利用深度优先遍历两棵树，每次两棵树对应的元素出栈时判断两个值是否相等，进栈时判断两个栈长度是否一致，有不同则立刻返回false，如果坚持到最后则返回true。

代码（Java）：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSameTree(TreeNode p, TreeNode q) {
        Stack<TreeNode> pstack=new Stack<TreeNode>();  
        Stack<TreeNode> qstack=new Stack<TreeNode>(); 
        if(p!=null) pstack.push(p);  
        if(q!=null) qstack.push(q); 
        while (!pstack.isEmpty()&&!qstack.isEmpty()) {  
            p=pstack.pop();  
            q=qstack.pop();
            
            if(p.val!=q.val){
                return false;
            }
  
            if(p.right!=null){
                pstack.push(p.right);
            }
            if(q.right!=null){
                qstack.push(q.right);
            }
            if(pstack.size()!=qstack.size()){
                return false;
            }
            
            if(p.left!=null){
                pstack.push(p.left);
            }
            if(q.left!=null){
                qstack.push(q.left);
            }
            if(pstack.size()!=qstack.size()){
                return false;
            }
        }
        return pstack.size()==qstack.size();
    }
}