0095-Go-泛型

环境

• Time 2022-08-24
• Go 1.19

前言

说明

参考：https://gobyexample.com/generics

目标

使用 Go 语言的泛型。

泛型函数

package main

import "fmt"

func MapKeys[K comparable, V any](m map[K]V) []K {
    r := make([]K, 0, len(m))
    for k := range m {
        r = append(r, k)
    }
    return r
}

func main() {
    var m = map[int]string{1: "2", 2: "4", 4: "8"}

    fmt.Println("keys m:", MapKeys(m))
}

定义链表

type List[T any] struct {
    head, tail *element[T]
}

链表元素

type element[T any] struct {
    next *element[T]
    val  T
}

新增

func (lst *List[T]) Push(v T) {
    if lst.tail == nil {
        lst.head = &element[T]{val: v}
        lst.tail = lst.head
    } else {
        lst.tail.next = &element[T]{val: v}
        lst.tail = lst.tail.next
    }
}

获取

func (lst *List[T]) GetAll() []T {
    var elems []T
    for e := lst.head; e != nil; e = e.next {
        elems = append(elems, e.val)
    }
    return elems
}

使用

func main() {

    lst := List[int]{}
    lst.Push(10)
    lst.Push(13)
    lst.Push(23)
    fmt.Println("list:", lst.GetAll())
}

总结

使用 Go 语言的泛型。

附录