【LeetCode】第572题——另一个树的子树(难度:简单)
题目描述
给定两个非空二叉树 s 和 t,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。
-
示例 1:
给定的树 s:3 / \ 4 5 / \ 1 2
给定的树 t:
4 / \ 1 2
返回 true,因为 t 与 s 的一个子树拥有相同的结构和节点值。
-
示例 2:
给定的树 s:3 / \ 4 5 / \ 1 2 / 0
给定的树 t:
4 / \ 1 2
返回 false。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subtree-of-another-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
利用之前做过的相同的树的思路,先遍历s树,对s树的每一个子树都用 “相同的树” 的方法去检验,得出结果。本人遍历方法选用BFS。
也算是暴力法了。
代码详解
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean isSubtree(TreeNode s, TreeNode t) {
if(t == null) { // t是null肯定返回true,空树是任意树的子树
return true;
} else if(s == null) { // 这里t已经不是空树了,这时s如果是空树则返回false
return false;
}
// 典型BFS
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(s);
while(!queue.isEmpty()) {
TreeNode node = queue.poll();
if(func(node, t)) { // 判断树是否相同
return true;
}
if(node != null) {
queue.offer(node.left);
queue.offer(node.right);
}
}
return false;
}
public boolean func(TreeNode p, TreeNode q) {
// 相同的树题目的BFS思路
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(p);
queue.offer(q);
while(!queue.isEmpty()) {
TreeNode nodep = queue.poll();
TreeNode nodeq = queue.poll();
if(nodep == null && nodeq == null) {
continue;
} else if(nodep == null || nodeq == null) {
return false;
} else if(nodep.val != nodeq.val) {
return false;
}
queue.offer(nodep.left);
queue.offer(nodeq.left);
queue.offer(nodep.right);
queue.offer(nodeq.right);
}
return true;
}
}
注意点
- 可以用DFS、KMP、树哈希等方法解答,不过本人能力有限,这些方法详见官方题解吧
- 友情链接:第100题——相同的树。这里既有BFS方法详解又有DFS方法详解。