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();
    }
}



猜你喜欢

转载自blog.csdn.net/weixin_38385524/article/details/79864850