Haskell 函数定义：let 与 where 子句

在 Haskell 中，let 和 where 都是用来在函数内部定义局部变量或辅助函数的语法结构。它们功能相似，但作用域、使用位置和语法风格不同。正确选择能让你的代码更清晰、更易维护。

1. 理解 let 表达式

let 是一个表达式，可以在任何需要值的地方使用。它的基本形式是：

let variable = expression in body

记住：let 必须搭配 in，in 后面的部分才能使用前面定义的变量。

示例：计算圆柱体表面积

cylinderSurfaceArea r h =
  let sideArea = 2 * pi * r * h
      topArea  = pi * r ^ 2
  in sideArea + 2 * topArea

这里：

• 定义了两个局部变量 sideArea 和 topArea
• 在 in 后面的表达式中使用它们
• 整个 let...in 结构作为一个表达式返回结果

你甚至可以在 GHCi 中直接使用：

ghci> let x = 5 in x * 2
10

2. 理解 where 子句

where 是一个声明，只能用在函数定义的末尾。它的基本形式是：

function args = body
  where
    variable = expression

注意：where 不需要 in，它绑定的是上方整个函数体中的变量。

同样示例：用 where 实现圆柱体表面积

cylinderSurfaceArea r h = sideArea + 2 * topArea
  where
    sideArea = 2 * pi * r * h
    topArea  = pi * r ^ 2

这里：

• 先写主表达式 sideArea + 2 * topArea
• 再用 where 块定义这些变量
• 变量只在当前函数体内可见

3. 关键区别对比

以下表格总结了 let 与 where 的核心差异：

4. 实战场景选择指南

场景一：你需要在列表推导中定义临时变量

使用 let：

-- 计算每个数的平方及其两倍
squaresAndDoubles xs = [ (sq, dbl) | x <- xs, let sq = x^2, let dbl = 2*x ]

where 无法在这里使用，因为列表推导不是函数定义。

场景二：你在守卫（guards）中需要共享变量

使用 where：

describeNumber n
  | score > 7 = "Great"
  | score > 4 = "Okay"
  | otherwise = "Poor"
  where
    score = abs n - 3

这里的 score 在多个守卫中被复用。如果用 let，你得在每个守卫里重复定义，或者把整个守卫逻辑包进一个 let...in，反而更啰嗦。

场景三：你想在 lambda 或其他表达式中嵌入逻辑

使用 let：

applyWithOffset f offset x = 
  (\y -> let adjusted = y + offset in f adjusted) x

where 不能出现在 lambda 内部。

5. 常见错误与避坑指南

1. 忘记 in

   -- ❌ 错误！
   badExample x = let y = x + 1

   修正：加上 in y 或其他使用 y 的表达式。

2. 在 where 中试图返回值

   -- ❌ 错误！where 不是表达式
   nonsense x = where y = x + 1 in y

   修正：改用 let...in。

3. 混淆作用域
   let 的作用域严格限制在 in 之后；where 的作用域覆盖整个函数体（包括所有守卫和模式）。

4. 在顶层定义中使用 let

   -- ❌ 无效！模块顶层不能用 let...in
   let piApprox = 3.14 in ...

   修正：用普通等式定义，或用 where/let 在函数内部。

6. 性能与编译器优化

Haskell 编译器（如 GHC）对 let 和 where 的处理几乎完全相同。两者在生成的代码中没有性能差异。选择应基于可读性和上下文适配度，而非效率。

7. 高级技巧：嵌套与混合使用

你可以在 let 中使用 where，反之亦然，但通常不推荐，会降低可读性。不过在某些复杂逻辑中可能有用：

complexFunc x = 
  let helper y = y * factor
        where factor = x + 1
  in helper (x * 2)

这里 factor 只在 helper 内部需要，用 where 封装很合理。

8. 最佳实践建议

1. 优先考虑可读性：让读者一眼看懂数据流向。

   • 如果辅助变量用于“准备输入”，用 let
   • 如果用于“解释输出”或“共享中间结果”，用 where

2. 保持一致性：在一个项目或模块中，对同类逻辑采用相同风格。

3. 避免深层嵌套：超过两层的 let 或 where 嵌套时，考虑拆分为独立函数。

4. 在守卫中一律用 where：这是 Haskell 社区的普遍习惯。

-- 推荐风格：守卫 + where
taxAmount income
  | bracket == 1 = income * 0.1
  | bracket == 2 = income * 0.2
  | otherwise    = income * 0.3
  where
    bracket = if income < 50000 then 1 else if income < 100000 then 2 else 3

5. 在表达式上下文中用 let：如列表推导、lambda、do 表达式等。

-- 推荐风格：列表推导 + let
normalizePoints points = 
  [ (x', y') 
  | (x, y) <- points
  , let center = average points
        dx = x - fst center
        dy = y - snd center
        dist = sqrt (dx^2 + dy^2)
        x' = dx / dist
        y' = dy / dist
  ]