From bb6a8806c956aeadcf3b21d0090f140fb5b0db4d Mon Sep 17 00:00:00 2001 From: terminaldweller Date: Mon, 8 Jan 2024 00:28:38 -0500 Subject: 872 --- 872/go.mod | 3 +++ 872/main.go | 74 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 77 insertions(+) create mode 100644 872/go.mod create mode 100644 872/main.go diff --git a/872/go.mod b/872/go.mod new file mode 100644 index 0000000..10fbec9 --- /dev/null +++ b/872/go.mod @@ -0,0 +1,3 @@ +module 872 + +go 1.18 diff --git a/872/main.go b/872/main.go new file mode 100644 index 0000000..24ef0ab --- /dev/null +++ b/872/main.go @@ -0,0 +1,74 @@ +package main + +import "fmt" + +type TreeNode struct { + Val int + Left *TreeNode + Right *TreeNode +} + +func preorderVisit(node *TreeNode, leafSequence *[]int) { + if node == nil { + return + } + if node.Left == nil && node.Right == nil { + *leafSequence = append(*leafSequence, node.Val) + return + } + + preorderVisit(node.Left, leafSequence) + preorderVisit(node.Right, leafSequence) +} + +func leafSimilar(root1, root2 *TreeNode) bool { + var leafSequence1 []int + var leafSequence2 []int + + preorderVisit(root1, &leafSequence1) + preorderVisit(root2, &leafSequence2) + + // for i := 0; i < len(leafSequence1); i++ { + // fmt.Println((leafSequence1)[i]) + // } + // for i := 0; i < len(leafSequence2); i++ { + // fmt.Println((leafSequence2)[i]) + // } + + if len(leafSequence1) != len(leafSequence2) { + return false + } + + for i := 0; i < len(leafSequence1); i++ { + if (leafSequence1)[i] != (leafSequence2)[i] { + return false + } + } + + return true +} + +func main() { + var node1 TreeNode + var node2 TreeNode + var node3 TreeNode + node1.Val = 1 + node2.Val = 2 + node3.Val = 3 + + node1.Left = &node2 + node1.Right = &node3 + + var node4 TreeNode + var node5 TreeNode + var node6 TreeNode + node4.Val = 1 + node5.Val = 2 + node6.Val = 3 + + node4.Left = &node6 + node4.Right = &node5 + + fmt.Println(leafSimilar(&node1, &node4)) + +} -- cgit v1.2.3