Fix references (#5)

Author: johnfercher

README.md

@@ -15,39 +15,41 @@ package main
 import (
 	"fmt"
 
-	"github.com/johnfercher/tree/pkg/node"
 	"github.com/johnfercher/tree/pkg/tree"
 )
 
 // nolint:gomnd,gocritic
 func main() {
 	tr := tree.New[string]()
 
-	tr.AddRoot(node.New(0, "0.0"))
+	tr.AddRoot(tree.NewNode(0, "0.0"))
 
-	tr.Add(0, node.New(1, "0.1"))
-	tr.Add(0, node.New(2, "0.2"))
+	tr.Add(0, tree.NewNode(1, "0.1"))
+	tr.Add(0, tree.NewNode(2, "0.2"))
 
-	tr.Add(1, node.New(3, "1.3"))
-	tr.Add(1, node.New(4, "1.4"))
+	tr.Add(1, tree.NewNode(3, "1.3"))
+	tr.Add(1, tree.NewNode(4, "1.4"))
 
-	tr.Add(2, node.New(5, "2.5"))
-	tr.Add(2, node.New(6, "2.6"))
+	tr.Add(2, tree.NewNode(5, "2.5"))
+	tr.Add(2, tree.NewNode(6, "2.6"))
 
 	root, ok := tr.GetRoot()
-	fmt.Println(ok)
-	fmt.Println(root)
+	fmt.Println(ok)         // true
+	fmt.Println(root.Get()) // 0.0
 
 	node, ok := tr.Get(3)
-	fmt.Println(ok)
-	fmt.Println(node)
+	fmt.Println(ok)         // true
+	fmt.Println(node.Get()) // 1.3
 
 	structure, ok := tr.GetStructure()
-	fmt.Println(ok)
-	fmt.Println(structure)
+	fmt.Println(ok)        // true
+	fmt.Println(structure) // (NULL) -> (0), (0) -> (1),  (1) -> (3),  (1) -> (4),  (0) -> (2),  (2) -> (5),  (2) -> (6)
 
 	nodes, ok := tr.Backtrack(6)
-	fmt.Println(ok)
-	fmt.Println(nodes)
+	fmt.Println(ok) // true
+	for _, node := range nodes {
+		fmt.Println(node.Get()) // 2.6, 0.2, 0.0
+	}
 }
+
 ```

cmd/main.go

@@ -3,38 +3,39 @@ package main
 import (
 	"fmt"
 
-	"github.com/johnfercher/tree/pkg/node"
 	"github.com/johnfercher/tree/pkg/tree"
 )
 
 // nolint:gomnd,gocritic
 func main() {
 	tr := tree.New[string]()
 
-	tr.AddRoot(node.New(0, "0.0"))
+	tr.AddRoot(tree.NewNode(0, "0.0"))
 
-	tr.Add(0, node.New(1, "0.1"))
-	tr.Add(0, node.New(2, "0.2"))
+	tr.Add(0, tree.NewNode(1, "0.1"))
+	tr.Add(0, tree.NewNode(2, "0.2"))
 
-	tr.Add(1, node.New(3, "1.3"))
-	tr.Add(1, node.New(4, "1.4"))
+	tr.Add(1, tree.NewNode(3, "1.3"))
+	tr.Add(1, tree.NewNode(4, "1.4"))
 
-	tr.Add(2, node.New(5, "2.5"))
-	tr.Add(2, node.New(6, "2.6"))
+	tr.Add(2, tree.NewNode(5, "2.5"))
+	tr.Add(2, tree.NewNode(6, "2.6"))
 
 	root, ok := tr.GetRoot()
-	fmt.Println(ok)
-	fmt.Println(root)
+	fmt.Println(ok)         // true
+	fmt.Println(root.Get()) // 0.0
 
 	node, ok := tr.Get(3)
-	fmt.Println(ok)
-	fmt.Println(node)
+	fmt.Println(ok)         // true
+	fmt.Println(node.Get()) // 1.3
 
 	structure, ok := tr.GetStructure()
-	fmt.Println(ok)
-	fmt.Println(structure)
+	fmt.Println(ok)        // true
+	fmt.Println(structure) // (NULL) -> (0), (0) -> (1),  (1) -> (3),  (1) -> (4),  (0) -> (2),  (2) -> (5),  (2) -> (6)
 
 	nodes, ok := tr.Backtrack(6)
-	fmt.Println(ok)
-	fmt.Println(nodes)
+	fmt.Println(ok) // true
+	for _, node := range nodes {
+		fmt.Println(node.Get()) // 2.6, 0.2, 0.0
+	}
 }