@@ -58,6 +58,7 @@ package main
58
58
59
59
import (
60
60
" fmt"
61
+ " github.com/johnfercher/go-tree/node"
61
62
62
63
" github.com/johnfercher/go-tree/tree"
63
64
)
@@ -66,24 +67,24 @@ import (
66
67
func main () {
67
68
tr := tree.New [string ]()
68
69
69
- tr.AddRoot (tree. NewNode ( 0 , " 0.0" ))
70
+ tr.AddRoot (node. New ( " 0.0" ). WithID ( 0 ))
70
71
71
- tr.Add (0 , tree. NewNode ( 1 , " 0.1" ))
72
- tr.Add (0 , tree. NewNode ( 2 , " 0.2" ))
72
+ tr.Add (0 , node. New ( " 0.1" ). WithID ( 1 ))
73
+ tr.Add (0 , node. New ( " 0.2" ). WithID ( 2 ))
73
74
74
- tr.Add (1 , tree. NewNode ( 3 , " 1.3" ))
75
- tr.Add (1 , tree. NewNode ( 4 , " 1.4" ))
75
+ tr.Add (1 , node. New ( " 1.3" ). WithID ( 3 ))
76
+ tr.Add (1 , node. New ( " 1.4" ). WithID ( 4 ))
76
77
77
- tr.Add (2 , tree. NewNode ( 5 , " 2.5" ))
78
- tr.Add (2 , tree. NewNode ( 6 , " 2.6" ))
78
+ tr.Add (2 , node. New ( " 2.5" ). WithID ( 5 ))
79
+ tr.Add (2 , node. New ( " 2.6" ). WithID ( 6 ))
79
80
80
81
root , ok := tr.GetRoot ()
81
- fmt.Println (ok) // true
82
- fmt.Println (root.Get ()) // 0, 0.0
82
+ fmt.Println (ok) // true
83
+ fmt.Println (root.GetData ()) // 0.0
83
84
84
85
node , ok := tr.Get (3 )
85
- fmt.Println (ok) // true
86
- fmt.Println (node.Get ()) // 3, 1.3
86
+ fmt.Println (ok) // true
87
+ fmt.Println (node.GetData ()) // 1.3
87
88
88
89
structure , ok := tr.GetStructure ()
89
90
fmt.Println (ok) // true
@@ -92,7 +93,7 @@ func main() {
92
93
nodes , ok := tr.Backtrack (6 )
93
94
fmt.Println (ok) // true
94
95
for _ , node := range nodes {
95
- fmt.Println (node.Get ()) // 6, 2.6; 2, 0.2; 0, 0.0
96
+ fmt.Println (node.GetData ()) // 2.6; 0.2; 0.0
96
97
}
97
98
}
98
99
```
0 commit comments