版權聲明:本文為博主原創文章,未經博主允許不得轉載。
難度:容易
要求:
有兩個不同大小的二進制樹: T1有上百萬的節點; T2 有好幾百的節點。請設計一種算法,判定 T2 是否為 T1的子樹。
注意事項
若 T1 中存在從節點 n 開始的子樹與 T2 相同,我們稱 T2 是 T1 的子樹。也就是說,如果在 T1 節點 n 處將樹砍斷,砍斷的部分將與 T2 完全相同。
樣例
下面的例子中 T2 是 T1 的子樹:
1 3
/ \ /
T1 = 2 3 T2 = 4
/
4
下面的例子中 T2 不是 T1 的子樹:
1 3
/ \ \
T1 = 2 3 T2 = 4
/
4
思路:
/**
* @param T1, T2: The roots of binary tree.
* @return: True if T2 is a subtree of T1, or false.
*/
public boolean isSubtree(TreeNode T1, TreeNode T2) {
if (T2 == null) {
return true;
}
if (T1 == null) {
return false;
}
if (isEqual(T1, T2)) {
return true;
}
if (isSubtree(T1.left, T2) || isSubtree(T1.right, T2)) {
return true;
}
return false;
}
private boolean isEqual(TreeNode T1, TreeNode T2) {
if (T1 == null || T2 == null) {
return T1 == T2;
}
if (T1.val != T2.val) {
return false;
}
return isEqual(T1.left, T2.left) && isEqual(T1.right, T2.right);
}