51 lines
773 B
Go
51 lines
773 B
Go
package lib
|
|
|
|
// Sets
|
|
|
|
type Set[T comparable] struct {
|
|
inner map[T]struct{}
|
|
}
|
|
|
|
func NewSet[T comparable](items ...T) Set[T] {
|
|
set := Set[T]{
|
|
inner: make(map[T]struct{}),
|
|
}
|
|
for _, i := range items {
|
|
set.Add(i)
|
|
}
|
|
return set
|
|
}
|
|
|
|
func (s Set[T]) Add(item T) {
|
|
s.inner[item] = struct{}{}
|
|
}
|
|
|
|
func (s Set[T]) Remove(item T) {
|
|
delete(s.inner, item)
|
|
}
|
|
|
|
func (s Set[T]) Contains(item T) bool {
|
|
_, exists := s.inner[item]
|
|
return exists
|
|
}
|
|
|
|
func (s Set[T]) ToArray() []T {
|
|
array := []T{}
|
|
for i := range s.inner {
|
|
array = append(array, i)
|
|
}
|
|
return array
|
|
}
|
|
|
|
// Operations
|
|
|
|
func Flatten[K comparable, V comparable](mapping map[K][]V) []V {
|
|
var values []V
|
|
for _, array := range mapping {
|
|
for _, v := range array {
|
|
values = append(values, v)
|
|
}
|
|
}
|
|
return values
|
|
}
|