Haskell固定大小数组选型及fixed-vector索引取值问题

需求背景 #

需要处理编译时已知长度（仅2或3）的同构结构，避免变长容器的性能开销，同时拒绝异构元组的类型不一致性，寻求类似C++ std::array的固定大小数组实现。

可选库分析与问题处理 #

已调研vector-sized和fixed-vector两个库，两者各有优劣：

1. fixed-vector的常见问题与修复 #

使用fixed-vector时遇到的模块导入、作用域及类型约束问题，本质是其激进的依赖类型设计与GHC类型推断能力的冲突：

• 模块导入问题：需明确导入核心模块及对应长度的构造器模块，示例：

  import qualified Data.Vector.Fixed as V
  import Data.Vector.Fixed.Containers (mk3) -- 导入3长度构造器
  

• mk3作用域错误：若未显式导入子模块构造器，需使用限定名调用，如V.mk3 1 2 3（需确认模块导出规则）；
• V.index类型约束不满足：因index操作依赖长度相关的类型约束，需显式标注数组类型帮助GHC推断：

  v :: V.Vector 3 Int
  v = V.mk3 1 2 3
  val = V.index v 1 -- 此时类型约束满足
  

2. 更适合场景的替代方案 #

由于仅需处理2或3长度的结构，以下两种方案更轻便：

方案一：手写极简固定大小类型

完全无需外部依赖，性能与元组一致，编译时可通过模式匹配保证索引安全：

newtype Vec2 a = Vec2 (a, a)
newtype Vec3 a = Vec3 (a, a, a)

-- 安全索引（非穷尽模式会触发GHC警告）
indexVec2 :: Vec2 a -> Int -> a
indexVec2 (Vec2 (x, _)) 0 = x
indexVec2 (Vec2 (_, y)) 1 = y
indexVec2 _ i = error $ "Vec2 index out of bounds: " ++ show i

indexVec3 :: Vec3 a -> Int -> a
indexVec3 (Vec3 (x, _, _)) 0 = x
indexVec3 (Vec3 (_, y, _)) 1 = y
indexVec3 (Vec3 (_, _, z)) 2 = z
indexVec3 _ i = error $ "Vec3 index out of bounds: " ++ show i

-- 构造器
mkVec2 :: a -> a -> Vec2 a
mkVec2 = Vec2

mkVec3 :: a -> a -> a -> Vec3 a
mkVec3 = Vec3

方案二：使用vector-sized库

基于成熟的vector库，用类型级自然数标记长度，类型推断更友好，编译时可检查索引越界：

import qualified Data.Vector.Sized as VS
import GHC.TypeLits (Nat)

-- 构造3长度数组
v3 :: VS.Vector 3 Int
v3 = VS.fromTuple (1, 2, 3)

-- 编译时安全索引（越界会触发编译错误）
val = VS.index v3 1 -- 返回2

总结 #

• 若仅需2/3长度的固定数组，手写类型是最轻便、无依赖的选择；
• 若需扩展到其他小长度场景，vector-sized比fixed-vector更省心，类型推断更顺畅；
• fixed-vector适合复杂依赖类型场景，但小长度下无需为其类型约束成本买单。

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