二叉樹 8 (在每個樹行中尋找最大值 leetcode 515)

思想

二叉樹的核心思想是分治和遞歸,特點(diǎn)是遍歷方式裕偿。
解題方式常見兩類思路:

  1. 遍歷一遍二叉樹尋找答案尊剔;
  2. 通過分治分解問題尋求答案爪幻;

遍歷分為前中后序,本質(zhì)上是遍歷二叉樹過程中處理每個節(jié)點(diǎn)的三個特殊時間點(diǎn):

  1. 前序是在剛剛進(jìn)入二叉樹節(jié)點(diǎn)時執(zhí)行须误;
  2. 后序是在將要離開二叉樹節(jié)點(diǎn)時執(zhí)行挨稿;
  3. 中序是左子樹遍歷完進(jìn)入右子樹前執(zhí)行;
# 前序
     1 node
    /      \
 2 left   3 right
中左右
 
# 中序
     2 node
    /      \
 1 left    3 right
左中右
 
# 后序
     3 node
    /      \
 1 left    2 right     
左右中       

多叉樹只有前后序列遍歷京痢,因?yàn)橹挥卸鏄溆形ㄒ灰淮沃虚g節(jié)點(diǎn)的遍歷

題目的關(guān)鍵就是找到遍歷過程中的位置奶甘,插入對應(yīng)代碼邏輯實(shí)現(xiàn)場景的目的。

實(shí)例

在每個樹行中尋找最大值 leetcode 515

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

輸入:
root: TreeNode祭椰,二叉樹的根節(jié)點(diǎn)

輸出:
List[int]臭家,返回每層中的最大值。

舉例:
給定二叉樹 [1,3,2,5,3,null,9]
總共 3 層方淤,分層遍歷記錄最大值钉赁,[1, 3, 9]

    1
   / \
 3    2
/ \    \
5 3     9

二叉樹的數(shù)據(jù)存儲可以使用鏈表,也可以使用數(shù)組携茂,往往數(shù)組更容易表達(dá)你踩,根節(jié)點(diǎn)從 index=1 處開始存儲,浪費(fèi) index=0 的位置
left_child = 2 * parent
right_child = 2 * parent + 1
parent = child // 2

遍歷解

層序遍歷在每一層是從左到右的讳苦,解題關(guān)鍵是識別當(dāng)前節(jié)點(diǎn)在哪一層带膜,這決定了將當(dāng)前的節(jié)點(diǎn)值寫入返回?cái)?shù)組的哪一級;
另一是實(shí)現(xiàn)從左到右的遍歷医吊,這個過程符合隊(duì)列的數(shù)據(jù)結(jié)構(gòu)钱慢。
使用隊(duì)列來存儲待遍歷的節(jié)點(diǎn)逮京,上例卿堂,使用 result = [] 記錄遍歷的結(jié)果:

  • 首先從 [(0, TreeNode(1))] 開始,高度 0 == len(result)懒棉,此時標(biāo)志著新的一層開始草描,建立當(dāng)層的列表放入 result,result = [1]策严,同時高度設(shè)置為
    1穗慕。再將左右子樹的節(jié)點(diǎn)放入待遍歷列表,[(1, TreeNode(2), (1, TreeNode(3)))]妻导;
  • 繼續(xù)從待遍歷列表取出隊(duì)列尾部元素 (1, TreeNode(3))逛绵,高度 1 == len(result)怀各,此時標(biāo)志著新的一層開始,建立當(dāng)層的列表放入 result术浪,result = [1, 3]瓢对,同時高度設(shè)置為
    2,再將左右子樹的節(jié)點(diǎn)放入待遍歷列表胰苏,[(2, TreeNode(3)), (2, TreeNode(5)),(1, TreeNode(3)))]硕蛹;
  • 繼續(xù)從待遍歷列表取出隊(duì)列尾部元素 (1, TreeNode(3)),高度 1 < len(result)硕并,標(biāo)志當(dāng)前層的繼續(xù)遍歷法焰,在對應(yīng)層高 1 的下標(biāo)找最大值,result = [1倔毙,3]
    埃仪。再將左右子樹的節(jié)點(diǎn)放入待遍歷列表,[(2, TreeNode(9)), (2, TreeNode(3)), (2, TreeNode(5))]普监;
  • 繼續(xù)從待遍歷列表取出隊(duì)列尾部元素 (2, TreeNode(5))贵试,高度 2 == len(result),此時標(biāo)志著新的一層開始凯正,建立當(dāng)層的列表放入 result毙玻,result = [1,3,5]
    ,同時高度設(shè)置為 3廊散,沒有左右子樹節(jié)點(diǎn)桑滩;
  • 繼續(xù)從待遍歷列表取出隊(duì)列尾部元素 (2, TreeNode(3)),高度 2 < len(result)允睹,標(biāo)志當(dāng)前層的繼續(xù)遍歷运准,在對應(yīng)層高 2 的下標(biāo)找最大值,result = [1,3,5]
    缭受,沒有左右子樹節(jié)點(diǎn)胁澳;
  • 繼續(xù)從待遍歷列表取出隊(duì)列尾部元素 (2, TreeNode(9)),高度 2 < len(result)米者,標(biāo)志當(dāng)前層的繼續(xù)遍歷韭畸,在對應(yīng)層高 2 的下標(biāo)找最大值,result = [1,3,9]
    蔓搞,沒有左右子樹節(jié)點(diǎn)胰丁;

編碼


from typing import Optional, List


class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


def find_largest_value_in_each_tree_row(root: Optional[TreeNode]) -> List[int]:
    # 初始化
    level_max = []
    node_queue = []
    height = 0
    if root is not None:
        node_queue.insert(0, (height, root))
    # 層序遍歷
    while node_queue:
        cur_height, cur_node = node_queue.pop()
        if cur_height == len(level_max):
            # 新的一層創(chuàng)建一個新的對象
            level_max.append(cur_node.val)
            height += 1
        else:
            # 遍歷已有層選取歷史上最大值
            level_max[cur_height] = max(level_max[cur_height], cur_node.val)
        if cur_node.left is not None:
            node_queue.insert(0, (height, cur_node.left))
        if cur_node.right is not None:
            node_queue.insert(0, (height, cur_node.right))
    return level_max

相關(guān)

二叉樹 0
二叉樹 1
二叉樹 2
二叉樹 3
二叉樹 4
二叉樹 5
二叉樹 6
二叉樹 7

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市喂分,隨后出現(xiàn)的幾起案子锦庸,更是在濱河造成了極大的恐慌,老刑警劉巖蒲祈,帶你破解...
    沈念sama閱讀 207,113評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件甘萧,死亡現(xiàn)場離奇詭異萝嘁,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)扬卷,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,644評論 2 381
  • 文/潘曉璐 我一進(jìn)店門酿愧,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人邀泉,你說我怎么就攤上這事嬉挡。” “怎么了汇恤?”我有些...
    開封第一講書人閱讀 153,340評論 0 344
  • 文/不壞的土叔 我叫張陵庞钢,是天一觀的道長。 經(jīng)常有香客問我因谎,道長基括,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 55,449評論 1 279
  • 正文 為了忘掉前任财岔,我火速辦了婚禮风皿,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘匠璧。我一直安慰自己桐款,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,445評論 5 374
  • 文/花漫 我一把揭開白布夷恍。 她就那樣靜靜地躺著魔眨,像睡著了一般。 火紅的嫁衣襯著肌膚如雪酿雪。 梳的紋絲不亂的頭發(fā)上遏暴,一...
    開封第一講書人閱讀 49,166評論 1 284
  • 那天,我揣著相機(jī)與錄音指黎,去河邊找鬼朋凉。 笑死,一個胖子當(dāng)著我的面吹牛醋安,可吹牛的內(nèi)容都是我干的杂彭。 我是一名探鬼主播,決...
    沈念sama閱讀 38,442評論 3 401
  • 文/蒼蘭香墨 我猛地睜開眼茬故,長吁一口氣:“原來是場噩夢啊……” “哼盖灸!你這毒婦竟也來了蚁鳖?” 一聲冷哼從身側(cè)響起磺芭,我...
    開封第一講書人閱讀 37,105評論 0 261
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎醉箕,沒想到半個月后钾腺,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體徙垫,經(jīng)...
    沈念sama閱讀 43,601評論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,066評論 2 325
  • 正文 我和宋清朗相戀三年放棒,在試婚紗的時候發(fā)現(xiàn)自己被綠了姻报。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,161評論 1 334
  • 序言:一個原本活蹦亂跳的男人離奇死亡间螟,死狀恐怖吴旋,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情厢破,我是刑警寧澤荣瑟,帶...
    沈念sama閱讀 33,792評論 4 323
  • 正文 年R本政府宣布,位于F島的核電站摩泪,受9級特大地震影響笆焰,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜见坑,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,351評論 3 307
  • 文/蒙蒙 一嚷掠、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧荞驴,春花似錦不皆、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,352評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至孙蒙,卻和暖如春项棠,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背挎峦。 一陣腳步聲響...
    開封第一講書人閱讀 31,584評論 1 261
  • 我被黑心中介騙來泰國打工香追, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人坦胶。 一個月前我還...
    沈念sama閱讀 45,618評論 2 355
  • 正文 我出身青樓透典,卻偏偏與公主長得像,于是被迫代替她去往敵國和親顿苇。 傳聞我的和親對象是個殘疾皇子峭咒,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,916評論 2 344

推薦閱讀更多精彩內(nèi)容