Stefan Pochmann 的上帝之手(8)

100. 相同的樹(shù)

給定兩個(gè)二叉樹(shù),編寫一個(gè)函數(shù)來(lái)檢驗(yàn)它們是否相同在岂。

如果兩個(gè)樹(shù)在結(jié)構(gòu)上相同奔则,并且節(jié)點(diǎn)具有相同的值,則認(rèn)為它們是相同的蔽午。
示例 1:

輸入:

           1         1
          / \       / \
         2   3     2   3

        [1,2,3],   [1,2,3]

輸出: true
示例 2:

輸入:

           1         1
          /           \
         2             2

        [1,2],     [1,null,2]

輸出: false
示例 3:

輸入:

           1         1
          / \       / \
         2   1     1   2

        [1,2,1],   [1,1,2]

輸出: false

來(lái)源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/same-tree

普通寫法1:深度優(yōu)先搜索(7行)

def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
    if not p and not q:
        return True
    if not p or not q:
        return False
    if p.val != q.val:
        return False
    return self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right)

思路:判斷當(dāng)下節(jié)點(diǎn)是否相同易茬,而后判斷左孩子和右孩子是否相同。如果pq都是null則當(dāng)下相同;pq有一個(gè)是null則不同抽莱;否則看二者value是否相同范抓。

普通寫法2:廣度優(yōu)先搜索(17行)

def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
    def check(p, q):
        if not p and not q:
            return True
        if not q or not p:
            return False
        if p.val != q.val:
            return False
        return True
    deq = deque([(p, q)])
    while deq:
        p, q = deq.popleft()
        if not check(p, q):
            return False
        if p:
            deq.append((p.left, q.left))
            deq.append((p.right, q.right))
    return True

思路:用一個(gè)雙邊隊(duì)列將當(dāng)下節(jié)點(diǎn)的孩子放在隊(duì)列里,每次都查看當(dāng)前隊(duì)首是否相同食铐。

大神寫法1:深度優(yōu)先搜索(3行)

def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
    if p and q:
        return p.val == q.val and self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right)
    return p is q

思路與普通寫法1一致匕垫。return p is q等價(jià)于return True if p == None and q == None else False

大神寫法2:廣度優(yōu)先搜索(3行)

def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
    def t(n):
        return n and (n.val, t(n.left), t(n.right))
    return t(p) == t(q)

將所有的節(jié)點(diǎn)展成元組的形式虐呻,(root, left, right)象泵。由于and只在前項(xiàng)為真時(shí)返回后項(xiàng)的邏輯值,所以遇到空節(jié)點(diǎn)會(huì)返回空斟叼。
舉例偶惠,如果二叉樹(shù)p的形狀是:

           1
          / \
         2   3
        /
       4

print(t(p))返回

(1, (2, (4, None, None), None), (3, None, None))

這里的(n.val, t(n.left), t(n.right))換成[n.val, t(n.left), t(n.right)]也可以。

上帝之手(1行)

def isSameTree(self, p, q):
    return p and q and p.val == q.val and all(map(self.isSameTree, (p.left, p.right), (q.left, q.right))) or p is q

思路與深度優(yōu)先搜索一致朗涩。
上述所有寫法的時(shí)間復(fù)雜度(除大神寫法2)都是O(min(m,n))其中m,n表示兩棵樹(shù)的葉子節(jié)點(diǎn)數(shù)忽孽。如果先遍歷兩棵樹(shù)再比較,也包括大神寫法2馋缅,復(fù)雜度是O(max(m,n))扒腕。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末绢淀,一起剝皮案震驚了整個(gè)濱河市萤悴,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌皆的,老刑警劉巖覆履,帶你破解...
    沈念sama閱讀 221,198評(píng)論 6 514
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異费薄,居然都是意外死亡硝全,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,334評(píng)論 3 398
  • 文/潘曉璐 我一進(jìn)店門楞抡,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)伟众,“玉大人,你說(shuō)我怎么就攤上這事召廷〉氏幔” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 167,643評(píng)論 0 360
  • 文/不壞的土叔 我叫張陵竞慢,是天一觀的道長(zhǎng)先紫。 經(jīng)常有香客問(wèn)我,道長(zhǎng)筹煮,這世上最難降的妖魔是什么遮精? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 59,495評(píng)論 1 296
  • 正文 為了忘掉前任,我火速辦了婚禮败潦,結(jié)果婚禮上本冲,老公的妹妹穿的比我還像新娘准脂。我一直安慰自己,他們只是感情好檬洞,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,502評(píng)論 6 397
  • 文/花漫 我一把揭開(kāi)白布岩饼。 她就那樣靜靜地躺著,像睡著了一般尔觉。 火紅的嫁衣襯著肌膚如雪设易。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 52,156評(píng)論 1 308
  • 那天澎灸,我揣著相機(jī)與錄音院塞,去河邊找鬼。 笑死性昭,一個(gè)胖子當(dāng)著我的面吹牛拦止,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播糜颠,決...
    沈念sama閱讀 40,743評(píng)論 3 421
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼汹族,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了其兴?” 一聲冷哼從身側(cè)響起顶瞒,我...
    開(kāi)封第一講書(shū)人閱讀 39,659評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎元旬,沒(méi)想到半個(gè)月后榴徐,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,200評(píng)論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡匀归,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,282評(píng)論 3 340
  • 正文 我和宋清朗相戀三年坑资,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片穆端。...
    茶點(diǎn)故事閱讀 40,424評(píng)論 1 352
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡袱贮,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出体啰,到底是詐尸還是另有隱情攒巍,我是刑警寧澤,帶...
    沈念sama閱讀 36,107評(píng)論 5 349
  • 正文 年R本政府宣布狡赐,位于F島的核電站窑业,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏枕屉。R本人自食惡果不足惜常柄,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,789評(píng)論 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧西潘,春花似錦卷玉、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 32,264評(píng)論 0 23
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至品姓,卻和暖如春寝并,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背腹备。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 33,390評(píng)論 1 271
  • 我被黑心中介騙來(lái)泰國(guó)打工衬潦, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人植酥。 一個(gè)月前我還...
    沈念sama閱讀 48,798評(píng)論 3 376
  • 正文 我出身青樓镀岛,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親友驮。 傳聞我的和親對(duì)象是個(gè)殘疾皇子漂羊,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,435評(píng)論 2 359