VBA中代数数据类型开发：自定义vector类型及代数运算需求

Hey there! 针对你正在开发涵盖矩阵乘法、向量乘法、矩阵求逆等基础代数运算的函数，且想要定义一个能统一表示标量、向量、矩阵的vector数据类型的需求，结合你当前使用Variant多维数组的场景，我给你整理了一套实用的实现方案：

1. 用VBA自定义类型封装核心属性 #

既然你是在VBA环境下（从Variant数组返回赋值的特性可以推断），首先可以定义一个自定义类型，把类型标识、数据存储、维度信息打包在一起，这样能明确区分标量、向量、矩阵：

Type vector
    DataType As Integer ' 0=标量, 1=向量, 2=矩阵
    ScalarVal As Double ' 存储标量数值
    VectorData As Variant ' 一维数组存储向量元素
    MatrixData As Variant ' 二维数组存储矩阵元素
    Dimension As Integer ' 维度数：标量0，向量1，矩阵2
    Rows As Integer ' 矩阵行数/向量长度
    Cols As Integer ' 矩阵列数（标量/向量时设为1）
End Type

后续操作时，先通过DataType字段判断当前实例的类型，再针对性操作对应的数据存储字段即可。

2. 配套构造函数，简化类型创建 #

为了快速把标量、数组转换成vector类型，写几个辅助构造函数会很方便：

• 标量转vector：

Function ScalarToVector(scalar As Double) As vector
    Dim v As vector
    v.DataType = 0
    v.ScalarVal = scalar
    v.Dimension = 0
    v.Rows = 1
    v.Cols = 1
    ScalarToVector = v
End Function

• 自动识别数组类型转vector：

Function ArrayToVector(arr As Variant) As vector
    Dim v As vector
    Dim arrDim As Integer
    arrDim = GetArrayDimension(arr)
    
    Select Case arrDim
        Case 1
            v.DataType = 1
            v.VectorData = arr
            v.Dimension = 1
            v.Rows = UBound(arr) - LBound(arr) + 1
            v.Cols = 1
        Case 2
            v.DataType = 2
            v.MatrixData = arr
            v.Dimension = 2
            v.Rows = UBound(arr, 1) - LBound(arr, 1) + 1
            v.Cols = UBound(arr, 2) - LBound(arr, 2) + 1
        Case Else
            Err.Raise vbObjectError + 1001, , "不支持当前数组的维度"
    End Select
    ArrayToVector = v
End Function

' 辅助函数：获取数组的维度数
Function GetArrayDimension(arr As Variant) As Integer
    On Error Resume Next
    Dim dimCount As Integer
    Do
        dimCount = dimCount + 1
        UBound(arr, dimCount)
    Loop Until Err.Number <> 0
    GetArrayDimension = dimCount - 1
    On Error GoTo 0
End Function

3. 兼容原有Variant数组逻辑 #

考虑到你之前依赖Variant数组的赋值特性，可以加一个转换函数，把vector转回对应的Variant数组/标量，方便和旧代码兼容：

Function VectorToVariant(v As vector) As Variant
    Select Case v.DataType
        Case 0
            VectorToVariant = v.ScalarVal
        Case 1
            VectorToVariant = v.VectorData
        Case 2
            VectorToVariant = v.MatrixData
    End Select
End Function

这样你原有的函数如果需要返回Variant类型，就可以先处理vector实例，再通过这个函数转成Variant输出。

4. 运算函数的类型校验逻辑 #

在你的代数运算函数里，先通过DataType做类型校验，避免非法运算：
比如矩阵乘法函数：

Function MatrixMultiply(v1 As vector, v2 As vector) As vector
    Dim resultArr As Variant
    ' 先校验输入类型是否为矩阵
    If v1.DataType <> 2 Or v2.DataType <> 2 Then
        Err.Raise vbObjectError + 1002, , "该函数仅支持矩阵之间的乘法运算"
    End If
    ' 这里写你的矩阵乘法逻辑，操作v1.MatrixData和v2.MatrixData
    ' ... 计算得到resultArr二维数组 ...
    MatrixMultiply = ArrayToVector(resultArr)
End Function

这种设计既保留了你之前用Variant数组的灵活性，又通过自定义类型统一了标量、向量、矩阵的表示，避免了直接用Variant时类型模糊的问题，也方便后续维护和扩展运算逻辑。

内容的提问来源于stack exchange，提问作者Galo Mateo