无序链表(顺序查找)符号表

参考书籍
算法(第四版)Robert Sedgewick Kevin Wayne

该代码是基于泛型实现
键不允许为零,必须是可比较的。
值不允许为零值,因为我们在获取值不到元素时,会返回零值

实现文件

sequentialSearchST

package symbolTable

import (
	"errors"
	"reflect"
)

var (
	nullError = errors.New("key is null")
)

type Node[KEY any, VALUE any] struct {
	key   KEY
	value VALUE
	next  *Node[KEY, VALUE]
}

func NewNode[KEY any, VALUE any](key KEY, value VALUE, node *Node[KEY, VALUE]) *Node[KEY, VALUE] {
	return &Node[KEY, VALUE]{key: key, value: value, next: node}
}

// 顺序查找,基于无序链表
type SequentialSearchST[KEY comparable, VALUE any] struct {
	first *Node[KEY, VALUE]
	key   KEY
	value VALUE
	//compare Comparer[KEY]
	n int
}

func (s *SequentialSearchST[KEY, VALUE]) equals(key1, key2 KEY) bool {
	return key1 == key2
}

func NewSequentialSearchST[KEY comparable, VALUE any]() *SequentialSearchST[KEY, VALUE] {
	return &SequentialSearchST[KEY, VALUE]{}
}

func (s *SequentialSearchST[KEY, VALUE]) Get(key KEY) (val VALUE, error error) {

	if reflect.ValueOf(key).IsZero() {
		return val, nullError
	}

	for x := s.first; x != nil; x = x.next {
		if s.equals(x.key, key) {
			return x.value, nil
		}
	}

	return val, nil
}

func (s *SequentialSearchST[KEY, VALUE]) Put(key KEY, value VALUE) error {
	if reflect.ValueOf(key).IsZero() {
		return nullError
	}
	if reflect.ValueOf(value).IsZero() {
		return s.Delete(key)
	}
	//查找给定的健,找到则更新其值,否则在表中新建节点
	for x := s.first; x != nil; x = x.next {
		if s.equals(x.key, key) {
			x.value = value
			return nil
		}
	}
	s.first = NewNode(key, value, s.first)
	s.n++
	return nil
}

func (s *SequentialSearchST[KEY, VALUE]) Delete(key KEY) (err error) {
	if reflect.ValueOf(key).IsZero() {
		return nullError
	}
	s.first = s.delete(s.first, key)
	return nil
}

func (s *SequentialSearchST[KEY, VALUE]) delete(node *Node[KEY, VALUE], key KEY) (res *Node[KEY, VALUE]) {

	if reflect.ValueOf(key).IsZero() {
		return
	}

	if reflect.ValueOf(node).IsZero() {
		return
	}

	if s.equals(node.key, key) {
		res = node.next
		s.n--
		return node.next
	}

	node.next = s.delete(node.next, key)
	return
}

func (s *SequentialSearchST[KEY, VALUE]) Keys() []KEY {
	keys := make([]KEY, s.Size())
	for x := s.first; x != nil; x = x.next {
		keys = append(keys, x.key)
	}
	return keys
}

func (s *SequentialSearchST[KEY, VALUE]) Size() int {
	return s.n
}

func (s *SequentialSearchST[KEY, VALUE]) IsEmpty() bool {
	return s.Size() == 0
}

func (s *SequentialSearchST[KEY, VALUE]) Contains(key KEY) bool {
	if reflect.ValueOf(key).IsZero() {
		return false
	}
	var v VALUE
	v, err := s.Get(key)
	if err != nil {
		return false
	}
	if reflect.ValueOf(v).IsZero() {
		return false
	}

	return true
}

测试文件

sequentialSearchST_test.go

package symbolTable

import (
	"fmt"
	"strconv"
	"testing"
)

func TestNewSequentialSearchST(t *testing.T) {
	st := NewSequentialSearchST[string, int]()

	if st.IsEmpty() != true {
		fmt.Println("Expected true, but got false")
	}

	for i := 0; i < 10; i++ {
		st.Put(strconv.Itoa(i+1), i+1)
	}

	if st.Size() != 10 {
		t.Errorf("Expected 10, but got %d", st.Size())
	}

	get, err := st.Get("1")
	if err != nil {
		t.Fatal(err)
	}
	if get != 1 {
		t.Errorf("Expected 1, but got %d", get)
	}

	get, err = st.Get("100")
	if err != nil {
		t.Fatal(err)
	}
	if get != 0 {
		t.Errorf("Expected 0, but got %d", get)
	}

	get, err = st.Get("")
	if err == nil {
		t.Fatal("Expected error, but got nil")
	}
	if get != 0 {
		t.Errorf("Expected 0, but got %d", get)
	}

	err = st.Delete("1")
	get, err = st.Get("100")
	if err != nil {
		t.Fatal(err)
	}
	if get != 0 {
		t.Errorf("Expected 0, but got %d", get)
	}

	if st.IsEmpty() == true {
		t.Errorf("Expected false, but got true")
	}

	st.Put("6", 66)
	if v, _ := st.Get("6"); v != 66 {
		t.Errorf("Expected 66, but got %d", v)
	}

	err = st.Delete("6")
	if err != nil {
		t.Fatal(err)
	}

	err = st.Delete("0")
	if err != nil {
		t.Fatal(err)
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值