mirror of
https://github.com/emirpasic/gods.git
synced 2026-06-14 16:06:42 +03:00
Emir Pasic
81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
// Copyright (c) 2015, Emir Pasic. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/emirpasic/gods/sets/treeset"
|
|
"strings"
|
|
)
|
|
|
|
// IteratorWithIndexExample to demonstrate basic usage of IteratorWithIndex
|
|
func main() {
|
|
set := treeset.NewWithStringComparator()
|
|
set.Add("a", "b", "c")
|
|
it := set.Iterator()
|
|
|
|
fmt.Print("\nForward iteration\n")
|
|
for it.Next() {
|
|
index, value := it.Index(), it.Value()
|
|
fmt.Print("[", index, ":", value, "]") // [0:a][1:b][2:c]
|
|
}
|
|
|
|
fmt.Print("\nForward iteration (again)\n")
|
|
for it.Begin(); it.Next(); {
|
|
index, value := it.Index(), it.Value()
|
|
fmt.Print("[", index, ":", value, "]") // [0:a][1:b][2:c]
|
|
}
|
|
|
|
fmt.Print("\nBackward iteration\n")
|
|
for it.Prev() {
|
|
index, value := it.Index(), it.Value()
|
|
fmt.Print("[", index, ":", value, "]") // [2:c][1:b][0:a]
|
|
}
|
|
|
|
fmt.Print("\nBackward iteration (again)\n")
|
|
for it.End(); it.Prev(); {
|
|
index, value := it.Index(), it.Value()
|
|
fmt.Print("[", index, ":", value, "]") // [2:c][1:b][0:a]
|
|
}
|
|
|
|
if it.First() {
|
|
fmt.Print("\nFirst index: ", it.Index()) // First index: 0
|
|
fmt.Print("\nFirst value: ", it.Value()) // First value: a
|
|
}
|
|
|
|
if it.Last() {
|
|
fmt.Print("\nLast index: ", it.Index()) // Last index: 3
|
|
fmt.Print("\nLast value: ", it.Value()) // Last value: c
|
|
}
|
|
|
|
// Seek element starting with "b"
|
|
seek := func(index int, value interface{}) bool {
|
|
return strings.HasSuffix(value.(string), "b")
|
|
}
|
|
|
|
it.Begin()
|
|
for found := it.NextTo(seek); found; found = it.Next() {
|
|
fmt.Print("\nNextTo index: ", it.Index())
|
|
fmt.Print("\nNextTo value: ", it.Value())
|
|
} /*
|
|
NextTo index: 1
|
|
NextTo value: "b"
|
|
NextTo index: 2
|
|
NextTo value: "c"
|
|
*/
|
|
|
|
it.End()
|
|
for found := it.PrevTo(seek); found; found = it.Prev() {
|
|
fmt.Print("\nNextTo index: ", it.Index())
|
|
fmt.Print("\nNextTo value: ", it.Value())
|
|
} /*
|
|
NextTo index: 1
|
|
NextTo value: "b"
|
|
NextTo index: 0
|
|
NextTo value: "a"
|
|
*/
|
|
|
|
}
|