mirror of https://github.com/emirpasic/gods
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
1.9 KiB
Go
83 lines
1.9 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"
|
|
"strings"
|
|
|
|
"github.com/emirpasic/gods/v2/maps/treemap"
|
|
)
|
|
|
|
// IteratorWithKeyExample to demonstrate basic usage of IteratorWithKey
|
|
func main() {
|
|
m := treemap.New[int, string]()
|
|
m.Put(0, "a")
|
|
m.Put(1, "b")
|
|
m.Put(2, "c")
|
|
it := m.Iterator()
|
|
|
|
fmt.Print("\nForward iteration\n")
|
|
for it.Next() {
|
|
key, value := it.Key(), it.Value()
|
|
fmt.Print("[", key, ":", value, "]") // [0:a][1:b][2:c]
|
|
}
|
|
|
|
fmt.Print("\nForward iteration (again)\n")
|
|
for it.Begin(); it.Next(); {
|
|
key, value := it.Key(), it.Value()
|
|
fmt.Print("[", key, ":", value, "]") // [0:a][1:b][2:c]
|
|
}
|
|
|
|
fmt.Print("\nBackward iteration\n")
|
|
for it.Prev() {
|
|
key, value := it.Key(), it.Value()
|
|
fmt.Print("[", key, ":", value, "]") // [2:c][1:b][0:a]
|
|
}
|
|
|
|
fmt.Print("\nBackward iteration (again)\n")
|
|
for it.End(); it.Prev(); {
|
|
key, value := it.Key(), it.Value()
|
|
fmt.Print("[", key, ":", value, "]") // [2:c][1:b][0:a]
|
|
}
|
|
|
|
if it.First() {
|
|
fmt.Print("\nFirst key: ", it.Key()) // First key: 0
|
|
fmt.Print("\nFirst value: ", it.Value()) // First value: a
|
|
}
|
|
|
|
if it.Last() {
|
|
fmt.Print("\nLast key: ", it.Key()) // Last key: 2
|
|
fmt.Print("\nLast value: ", it.Value()) // Last value: c
|
|
}
|
|
|
|
// Seek key-value pair whose value starts with "b"
|
|
seek := func(key int, value string) bool {
|
|
return strings.HasSuffix(value, "b")
|
|
}
|
|
|
|
it.Begin()
|
|
for found := it.NextTo(seek); found; found = it.Next() {
|
|
fmt.Print("\nNextTo key: ", it.Key())
|
|
fmt.Print("\nNextTo value: ", it.Value())
|
|
} /*
|
|
NextTo key: 1
|
|
NextTo value: "b"
|
|
NextTo key: 2
|
|
NextTo value: "c"
|
|
*/
|
|
|
|
it.End()
|
|
for found := it.PrevTo(seek); found; found = it.Prev() {
|
|
fmt.Print("\nNextTo key: ", it.Key())
|
|
fmt.Print("\nNextTo value: ", it.Value())
|
|
} /*
|
|
NextTo key: 1
|
|
NextTo value: "b"
|
|
NextTo key: 0
|
|
NextTo value: "a"
|
|
*/
|
|
}
|