請考慮一棵二叉樹上所有的葉子,這些葉子的值按從左到右的順序排列形成一個 葉值序列 。
舉個例子,如上圖所示,給定一棵葉值序列為 (6, 7, 4, 9, 8) 的樹。
如果有兩棵二叉樹的葉值序列是相同,那么我們就認為它們是 葉相似 的。
如果給定的兩個根結點分別為 root1 和 root2 的樹是葉相似的,則返回 true;否則返回 false 。
示例 1:
輸入:root1 = [3,5,1,6,2,9,8,null,null,7,4], root2 = [3,5,1,6,7,4,2,null,null,null,null,null,null,9,8]
輸出:true
示例 2:
輸入:root1 = [1], root2 = [1]
輸出:true
解題思路
定義兩個函數,一個是用來查找并且記錄第一顆二叉樹葉子節點的。第二個函數是用來查找第二顆二叉樹葉子節點,并且驗證這顆二叉樹的葉子節點是否與第一顆的葉子節點序列相同
代碼
/*** Definition for a binary tree node.* type TreeNode struct {* Val int* Left *TreeNode* Right *TreeNode* }*/
func leafSimilar(root1 *TreeNode, root2 *TreeNode) bool {root := make([]int,0)var leadfs func(root1 *TreeNode)leadfs= func(root1 *TreeNode) {if root1.Left==nil&&root1.Right==nil{root=append(root,root1.Val)return}if root1.Left!=nil{leadfs(root1.Left)}if root1.Right!=nil{leadfs(root1.Right)}}leadfs(root1)i:=0flag:=truevar check func(root1 *TreeNode)check= func(root1 *TreeNode) {if root1.Left==nil&&root1.Right==nil{if i<len(root)&&root[i]!=root1.Val{flag=false}i++return}if root1.Left!=nil{check(root1.Left)}if root1.Right!=nil{check(root1.Right)}}check(root2)if i!=len(root){return false}return flag}