[leetcode]解决Same Tree的一点小心得

本次选择的题目是

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.

Solution:
首先我想到一个直接的方法,利用栈,按同样的顺序将p和q的各个节点push进去,再pop出来对比它们的val值,当然不是先全部push完再pop,而是先左再右,pop出来左节点,然后左节点一样了,push左节点的两个子节点,再pop出来右节点,检测右节点的val,push右节点的两个子节点,然后栈里还有左节点的两个子节点和右节点的两个子节点(假设它有,这里有4种情况,都没有,都有,都有左子节点,都有右子节点)一一对比。

/**
 * 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) {
         if(p == null && q == null){  
        return true;  
    }else if(p != null && q != null){  
        Stack<TreeNode> stack1 = new Stack<TreeNode>();  
        Stack<TreeNode> stack2 = new Stack<TreeNode>();  
        stack1.push(p);  
        stack2.push(q);  
        while(!stack1.empty() && !stack2.empty()){  
            TreeNode temp1 = stack1.pop();  
            TreeNode temp2 = stack2.pop();  
            if(temp1.val!=temp2.val){  
                return false;  
            }  
            if(temp1.right!=null&&temp1.left!=null&&temp2.right!=null&&temp2.left!=null){  
                stack1.push(temp1.left);  
                stack1.push(temp1.right);  
                stack2.push(temp2.left);  
                stack2.push(temp2.right);  
            }else if(temp1.right!=null&&temp1.left==null&&temp2.right!=null&&temp2.left==null){  
                stack1.push(temp1.right);  
                stack2.push(temp2.right);  
            }else if(temp1.right==null&&temp1.left!=null&&temp2.right==null&&temp2.left!=null){  
                stack1.push(temp1.left);  
                stack2.push(temp2.left);  
            }else if(temp1.right==null&&temp1.left==null&&temp2.right==null&&temp2.left==null){  

            }else{  
                return false;  
            }  
        }  
        return true;  
    }else{  
        return false;  
    }    
    }
}

discussion中提到的简单方法:

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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值