872

2 files changed, 77 insertions, 0 deletions

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))
+
+}