Initiatlize the map right

This commit is contained in:
Simon Martens
2025-02-11 22:06:58 +01:00
parent cda95c3958
commit 8ded984d01

View File

@@ -2,7 +2,7 @@ package datatypes
// INFO: use this if your key is unique
func MakeMap[T any, U comparable](data []T, f func(T) U) map[U]T {
m := make(map[U]T)
m := make(map[U]T, len(data))
for _, v := range data {
m[f(v)] = v
}
@@ -11,7 +11,7 @@ func MakeMap[T any, U comparable](data []T, f func(T) U) map[U]T {
// INFO: use this if your key is not unique
func MakeMultiMap[T any, U comparable](data []T, f func(T) U) map[U][]T {
m := make(map[U][]T)
m := make(map[U][]T, len(data))
for _, v := range data {
m[f(v)] = append(m[f(v)], v)
}