题目
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
| 输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构， 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B：
4
/
1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。
示例 1：
输入：A = [1,2,3], B = [3,1]
输出：false
示例 2：
输入：A = [3,4,5,1,2], B = [4,1]
输出：true
限制：
0 <= 节点个数 <= 10000
|
思路
深度遍历每一个节点，对每一个节点都对比一次TreeB，所以需要两个函数，第一个就是dfs，第二个就是对比TreeA和TreeB
实现
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
| private static class Solution {
public boolean isSubStructure(TreeNode A, TreeNode B) {
if (B == null || A == null) {
return false;
}
return comparison(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
}
private boolean comparison(TreeNode a, TreeNode b) {
if (b == null) {
return true;
}
if (a == null || a.val != b.val) {
return false;
}
return comparison(a.left, b.left) && comparison(a.right, b.right);
}
}
|
