array 计算错误

sparse_array/array2/array2.go

@@ -0,0 +1,90 @@
+package array2
+
+type Array2 struct {
+	sizes []int
+	ysize int
+	xsize int
+	data  [][]interface{}
+}
+
+func New() *Array2 {
+	return NewWithCap(8, 8)
+}
+
+func NewWithCap(ysize, xsize int) *Array2 {
+	arr := &Array2{ysize: ysize, xsize: xsize}
+	arr.sizes = make([]int, arr.ysize, arr.ysize)
+	arr.data = make([][]interface{}, arr.ysize, arr.ysize)
+	return arr
+}
+
+func (arr *Array2) Values() []interface{} {
+	var result []interface{}
+	for _, y := range arr.data {
+		if y != nil {
+			for _, v := range y {
+				if v == nil {
+					result = append(result, struct{}{})
+				} else {
+					result = append(result, v)
+				}
+			}
+		} else {
+			for i := 0; i < arr.ysize; i++ {
+				result = append(result, nil)
+			}
+		}
+	}
+
+	return result
+}
+
+func (arr *Array2) Set(idx int, value interface{}) {
+	yindex := idx / arr.ysize
+	xindex := idx % arr.ysize
+
+	ydata := arr.data[yindex]
+	if ydata == nil {
+		ydata = make([]interface{}, arr.xsize, arr.xsize)
+		arr.data[yindex] = ydata
+	}
+
+	if ydata[xindex] == nil {
+		arr.sizes[yindex]++
+	}
+
+	ydata[xindex] = value
+}
+
+func (arr *Array2) Get(idx int) (interface{}, bool) {
+	yindex := idx / arr.ysize
+	xindex := idx % arr.ysize
+
+	ydata := arr.data[yindex]
+	if ydata == nil {
+		return nil, false
+	}
+	xdata := ydata[xindex]
+	return xdata, xdata != nil
+}
+
+func (arr *Array2) Del(idx int) (interface{}, bool) {
+	yindex := idx / arr.ysize
+	xindex := idx % arr.ysize
+
+	ydata := arr.data[yindex]
+	if ydata == nil {
+		return nil, false
+	}
+	xdata := ydata[xindex]
+	ydata[xindex] = nil
+
+	isnil := xdata != nil
+	if isnil {
+		arr.sizes[yindex]--
+		if arr.sizes[yindex] == 0 {
+			arr.data[yindex] = nil
+		}
+	}
+	return xdata, isnil
+}

sparse_array/array2/array2_test.go

@@ -0,0 +1,154 @@
+package array2
+
+import (
+	"testing"
+
+	"github.com/davecgh/go-spew/spew"
+
+	"github.com/Pallinder/go-randomdata"
+)
+
+func TestArray2Set(t *testing.T) {
+	arr := NewWithCap(4, 4)
+	l := []int{0, 6, 5, 15}
+	for _, v := range l {
+		arr.Set(v, v)
+	}
+
+	var result string
+	result = spew.Sprint(arr.Values())
+	if result != "[0 {} {} {} {} 5 6 {} <nil> <nil> <nil> <nil> {} {} {} 15]" {
+		t.Error(result)
+	}
+
+	defer func() {
+		if err := recover(); err == nil {
+			t.Error("err == nil, but array the set is out of range")
+		}
+	}()
+
+	arr.Set(16, 16)
+}
+
+func TestArray2Get(t *testing.T) {
+	arr := New()
+	for i := 0; i < 64; i++ {
+		arr.Set(i, i)
+	}
+
+	for i := 0; i < 64; i++ {
+		if v, ok := arr.Get(i); ok {
+			if v != i {
+				t.Error("v is equal i, but", v, i)
+			}
+		} else {
+			t.Error("not ok is error")
+		}
+	}
+
+	defer func() {
+		if err := recover(); err == nil {
+			t.Error("err == nil, but array the get is out of range")
+		}
+	}()
+
+	arr.Get(64)
+}
+
+func TestArray2Del(t *testing.T) {
+	arr := NewWithCap(4, 4)
+	l := []int{0, 6, 5, 15}
+	for _, v := range l {
+		arr.Set(v, v)
+	}
+	// default  [0 {} {} {} {} 5 6 {} <nil> <nil> <nil> <nil> {} {} {} 15]
+	var result string
+
+	arr.Del(0)
+	result = spew.Sprint(arr.Values())
+	if result != "[<nil> <nil> <nil> <nil> {} 5 6 {} <nil> <nil> <nil> <nil> {} {} {} 15]" {
+		t.Error(result)
+	}
+
+	arr.Del(5)
+	result = spew.Sprint(arr.Values())
+	if result != "[<nil> <nil> <nil> <nil> {} {} 6 {} <nil> <nil> <nil> <nil> {} {} {} 15]" {
+		t.Error(result)
+	}
+
+	arr.Del(6)
+	result = spew.Sprint(arr.Values())
+	if result != "[<nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> {} {} {} 15]" {
+		t.Error(result)
+	}
+
+	arr.Del(15)
+	result = spew.Sprint(arr.Values())
+	if result != "[<nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil> <nil>]" {
+		t.Error(result)
+	}
+
+	defer func() {
+		if err := recover(); err == nil {
+			t.Error("err == nil, but array the del is out of range")
+		}
+	}()
+
+	arr.Del(16)
+}
+
+func BenchmarkArray2Set(b *testing.B) {
+
+	arr := NewWithCap(1000, 100)
+	b.N = 500000000
+
+	b.StopTimer()
+	var l []int
+	for i := 0; i < b.N/10; i++ {
+		l = append(l, randomdata.Number(0, 65535))
+	}
+	b.StartTimer()
+	for c := 0; c < 10; c++ {
+		for i := 0; i < b.N/10; i++ {
+			arr.Set(l[i], i)
+		}
+	}
+
+}
+
+func BenchmarkArray2Get(b *testing.B) {
+
+	arr := NewWithCap(1000, 100)
+	b.N = 500000000
+
+	b.StopTimer()
+
+	for i := 0; i < 105535; i++ {
+		v := randomdata.Number(0, 65535)
+		arr.Set(v, v)
+	}
+	b.StartTimer()
+
+	for i := 0; i < b.N; i++ {
+		arr.Get(i % 65535)
+	}
+
+}
+
+func BenchmarkArray2Del(b *testing.B) {
+
+	arr := NewWithCap(1000, 100)
+	b.N = 500000000
+
+	b.StopTimer()
+	for i := 0; i < 105535; i++ {
+		v := randomdata.Number(0, 65535)
+		arr.Set(v, v)
+	}
+	b.StartTimer()
+
+	for i := 0; i < b.N; i++ {
+		arr.Del(i % 65535)
+	}
+
+}