Haskell中如何用类型类与数据类型实现递归运算？向量实例问题

你的代码核心问题是类型不匹配和递归返回值构造错误：

• MathObject类的add/sub/mul方法要求返回同类型a（即Vec），但你定义的addVecs/subVecs/mulVecs返回[Int]或[Vec]，类型完全不兼容。
• 递归计算时没有正确将结果包装成Vec类型，导致列表无法匹配预期返回值。

修正后的完整代码 #

class MathObject a where
    add :: a -> a -> a
    sub :: a -> a -> a
    mul :: a -> a -> a

type Vector = [Int]

data Vec = Vec Vector deriving (Show, Eq)

-- 辅助函数：取出Vec内部的列表
unVec :: Vec -> Vector
unVec (Vec v) = v

instance MathObject Vec where 
    add v1 v2 = addVecs v1 v2
    sub v1 v2 = subVecs v1 v2
    mul v1 v2 = mulVecs v1 v2

addVecs :: Vec -> Vec -> Vec
addVecs (Vec []) _ = Vec []
addVecs _ (Vec []) = Vec []
addVecs (Vec (x:xs)) (Vec (y:ys)) = 
    let rest = addVecs (Vec xs) (Vec ys)
    in Vec $ (x + y) : unVec rest

subVecs :: Vec -> Vec -> Vec
subVecs (Vec []) _ = Vec []
subVecs _ (Vec []) = Vec []
subVecs (Vec (x:xs)) (Vec (y:ys)) = 
    let rest = subVecs (Vec xs) (Vec ys)
    in Vec $ (x - y) : unVec rest

-- 实现向量逐元素乘积（Hadamard乘积），若需点积/外积需调整逻辑
mulVecs :: Vec -> Vec -> Vec
mulVecs (Vec []) _ = Vec []
mulVecs _ (Vec []) = Vec []
mulVecs (Vec (x:xs)) (Vec (y:ys)) = 
    let rest = mulVecs (Vec xs) (Vec ys)
    in Vec $ (x * y) : unVec rest

更简洁的实现方式 #

Haskell的zipWith函数已经封装了列表的递归元素操作，直接用它可以避免手动写递归：

instance MathObject Vec where 
    add (Vec v1) (Vec v2) = Vec $ zipWith (+) v1 v2
    sub (Vec v1) (Vec v2) = Vec $ zipWith (-) v1 v2
    mul (Vec v1) (Vec v2) = Vec $ zipWith (*) v1 v2

向量乘法的其他说明 #

如果你的需求不是逐元素乘积：

• 点积：返回标量，此时mul的返回类型无法匹配MathObject的a -> a -> a，需要调整类型类或新增方法。
• 外积：返回矩阵，需要定义Matrix类型并实现MathObject实例。

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