feat: ✨ add support to return the values in the kd-tree

Author: rishitc

kdtree.go

@@ -96,6 +96,12 @@ func (t *KDTree[T]) Query(getRelativePosition func(T, int) RelativePosition) []T
 	return res
 }
 
+func (t *KDTree[T]) Values() []T {
+	var res []T
+	valuesImpl(t.root, &res)
+	return res
+}
+
 func (t *KDTree[T]) Add(value T) bool {
 	if t.root == nil {
 		t.root = NewKDNode(value)
@@ -117,6 +123,16 @@ func (t *KDTree[T]) Delete(value T) bool {
 	return ok
 }
 
+func valuesImpl[T Comparable[T]](r *kdNode[T], res *[]T) {
+	if r == nil {
+		return
+	}
+
+	*res = append(*res, r.value)
+	valuesImpl(r.left, res)
+	valuesImpl(r.right, res)
+}
+
 func (t *KDTree[T]) String() string {
 	b := strings.Builder{}
 	var q queue.Queue[*kdNode[T]] = queue.NewLLQueue[*kdNode[T]]()