树的子结构-数据结构与算法(下)

当前位置:　首页>> 技术小册>> 数据结构与算法(下)

### 题目描述
输入两棵二叉树`A`，`B`，判断`B`是不是`A`的子结构。（ps：我们约定空树不是任意一个树的子结构）

### 解法
递归方式遍历：

- 在树A中找到和树B的根结点值一样的结点R
- 判断树A以R为根结点的子树是否包含与树B一样的结构

这道题的time complexity应该为O(n * m)，其中n为root1的节点数，m为root2的节点数。

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

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

}
}
*/

/**
 * @author mcrwayfun
 * @version v1.0
 * @date Created in 2019/01/01
 * @description
 */
public class Solution {

public boolean HasSubtree(TreeNode root1, TreeNode root2) {

if (root1 == null || root2 == null) {
            return false;
        }

return isSame(root1, root2) || 
                HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2);
    }

private boolean isSame(TreeNode root1, TreeNode root2) {

if (root2 == null) {
            return true;
        }

// 在root2，root1遍历完成后，仍未找到符合的结构，返回false
        if (root1 == null) {
            return false;
        }

if (root1.val != root2.val) {
            return false;
        }

return isSame(root1.left, root2.left) && isSame(root1.right, root2.right);
    }

}
```

### 测试用例
1. 功能测试（树A和树B都是普通的二叉树；树B是/不是树A的子结构）；
2. 特殊输入测试（两棵二叉树的一个或者两个根结点为空指针，二叉树的所有结点都没有左/右子树）。