2021-07-31

记go语言实现n维数组

package ndimarr

type Element int

// Array 任意维度的数组
type Array struct {
	Elements []Element // 存放数组的元素
	Dim []int // 用于存放数组的维度
	HelperConstants []int // 辅助常量，计算偏移量的时候需要这个值
}

// 给定维度计算偏移量
func calOffset(arr *Array, dim ...int) (int, bool) {
	offset := 0
	for i := 0; i < len(dim); i++ {
		if dim[i] < 0 {
			return -1, false
		}
		offset += dim[i] * arr.HelperConstants[i]
	}
	return offset, true
}

// Create 创建指定维度的数组
func (arr *Array)Create(dim ...int) bool {
	arr.Dim = make([]int, len(dim)) // 创建存放数组维度数据的切片
	totalElements := 1
	for i := 0; i < len(dim); i++ {
		if dim[i] < 0 {
			return false
		}
		totalElements *= dim[i]
		arr.Dim[i] = dim[i]
	}

	// 创建数组用于存放数据的切片
	arr.Elements = make([]Element, totalElements)

	// 计算辅助常量数组值，该值后面将用于计算偏移量
	arr.HelperConstants = make([]int, len(dim))
	// 辅助常量数组的最后一个数据元素的值是1
	arr.HelperConstants[len(arr.HelperConstants) - 1] = 1
	// 通过递推的方式计算后续的常量值
	for i := len(arr.HelperConstants) - 2; i >= 0; i-- {
		arr.HelperConstants[i] = arr.HelperConstants[i + 1] * arr.Dim[i + 1]
	}

	return true
}

// Get 返回数组指定位置上的数据元素
func (arr *Array)Get(dim ...int) (Element, bool) {
	offset, ok := calOffset(arr, dim...)
	if ok {
		return arr.Elements[offset], true
	} else {
		return 0, false
	}
}

// Set 修改数组上指定位置的数据元素，如果修改成功返回true，否则返回false
func (arr *Array)Set(e Element, dim ...int) bool {
	offset, ok := calOffset(arr, dim...)
	if ok {
		arr.Elements[offset] = e
		return true
	} else {
		return false
	}
}