LeetCode105. 從前序與中序遍歷序列構造二叉樹

描述

根據一棵樹的前序遍歷與中序遍歷構造二叉樹

注意:

你能夠假設樹中沒有重複的元素。

例如，給出

前序遍歷 preorder = [3,9,20,15,7]
中序遍歷 inorder = [9,3,15,20,7]
複製代碼

返回以下的二叉樹：

3
   / \
  9  20
    /  \
   15   7
複製代碼

思路

• 前序遍歷中，第一個節點即根節點
• 在中序遍歷中，找出第一個節點的位置，根節點前面的 L 個數據，即根節點左子樹的中序遍歷數據，前序遍歷中根節點後面的 L 個數據即左子樹的前序遍歷
• 右子樹同上
• 簡而言之，肯定了根節點，肯定了左子樹和右子樹的數據，遞歸對左子樹和右子樹進行重建

代碼實現

// 根據一棵樹的前序遍歷與中序遍歷構造二叉樹

// TreeNode Definition for a binary tree node.
type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

func buildTree(preorder []int, inorder []int) *TreeNode {
	if len(preorder) == 0 {
		return nil
	}

	//前序遍歷的第一個節點即根節點
	res := &TreeNode{
		Val: preorder[0],
	}

	if len(preorder) == 1 {
		return res
	}

	// 在中序遍歷中，根節點前面的即根節點的左子樹，後面的即右子樹

	//匿名函數
	idx := func(val int, nums []int) int {
		for i, v := range nums {
			if v == val {
				return i
			}
		}
		return -1
	}(res.Val, inorder)
	if idx == -1 {
		panic("data error")
	}

	//遞歸,構建根節點左子樹和右子樹
	res.Left = buildTree(preorder[1:idx+1], inorder[:idx])
	res.Right = buildTree(preorder[idx+1:], inorder[idx+1:])
	return res
}
複製代碼

題目來源

105. 從前序與中序遍歷序列構造二叉樹

GitHub

• 源碼傳送門
• 各類數據結構及算法實現, LeetCode解題思路及答案
• 大廠面試題彙總及答案解析

本文爲原創文章，轉載註明出處，歡迎掃碼關注公衆號 樓蘭 或者網站lovecoding.club,第一時間看後續精彩文章，以爲好的話，順手分享到朋友圈吧，感謝支持。