程序员

剑指offer----树的子结构

2018-02-04  本文已影响0人  qming_c

题目:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/

代码:

public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root1 == null || root2 == null){
            return false;
        }
        return isSubTree(root1, root2) ||  HasSubtree(root1.left, root2)|| HasSubtree(root1.right, root2);
    }
    public boolean isSubTree(TreeNode root1,TreeNode root2){
        if(root2 == null){
            return true;
        }
        if(root1 == null){
            return false;
        }
        if(root1.val == root2.val){
            return isSubTree(root1.left, root2.left) && isSubTree(root1.right, root2.right);
        }else{
            return false;
        }
    }
}

isSubTree用来比较两个节点两个根节点相同的情况下2是不是1的一部分
return isSubTree(root1, root2) || HasSubtree(root1.left, root2)|| HasSubtree(root1.right, root2);
上面这个代码的意思大概是先比较root1,和root2是不是直接能构成子结构,如果不能,进行左节点递归,如果左子树中不包含root2,在递归右节点,(深度优先遍历)

上一篇 下一篇

猜你喜欢

热点阅读