題目

輸入兩棵二叉樹(shù)A和B，判斷B是不是A的子結(jié)構(gòu)。(約定空樹(shù)不是任意一個(gè)樹(shù)的子結(jié)構(gòu))

B是A的子結(jié)構(gòu)， 即 A中有出現(xiàn)和B相同的結(jié)構(gòu)和節(jié)點(diǎn)值。

例如:

給定的樹(shù) A:

? ? ?3

? ? / \

? ?4 ? 5

? / \

?1 ? 2

給定的樹(shù) B：

? ?4?

? /

?1

返回 true，因?yàn)?B 與 A 的一個(gè)子樹(shù)擁有相同的結(jié)構(gòu)和節(jié)點(diǎn)值。

示例 1：

輸入：A = [1,2,3], B = [3,1]

輸出：false

示例 2：

輸入：A = [3,4,5,1,2], B = [4,1]

輸出：true

限制：0 <= 節(jié)點(diǎn)個(gè)數(shù) <= 10000

思路

如果A 和 B 當(dāng)前節(jié)點(diǎn)值相等，則判斷 B 節(jié)點(diǎn)的子樹(shù)是否是 A 節(jié)點(diǎn)子樹(shù)的子結(jié)構(gòu)，即各個(gè)節(jié)點(diǎn)值相等（注意： 此時(shí)B的子樹(shù)中的空節(jié)點(diǎn)為A的子結(jié)構(gòu)）

如果不是，則遞歸判斷 B 是否是 A 的左節(jié)點(diǎn)的子結(jié)構(gòu)；

如果不是，則遞歸判斷 B 是否是 A 的右節(jié)點(diǎn)的子結(jié)構(gòu)；

代碼

時(shí)間復(fù)雜度：O(n)

空間復(fù)雜度：O(1)

class Solution {

public:

    bool isSubStructure(TreeNode* A, TreeNode* B) {

        if (!A || !B) return false;

        bool res = false;

        if (A->val == B->val) res = helper(A, B);

        if (!res) res = isSubStructure(A->left, B);

        if (!res) res = isSubStructure(A->right, B);

        return res;

    }

    bool helper(TreeNode *A, TreeNode *B) {

        if (!B) return true;  // B為空為A的子結(jié)構(gòu)，直接返回true

        if (!A || A->val != B->val) return false;  // B不為空A為空 或 兩者值不相等

        return helper(A->left, B->left) && helper(A->right, B->right);

    }

};

總結(jié)

以上是生活随笔為你收集整理的【剑指Offer】面试题26. 树的子结构的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。