Haskell中Integral类型类比较问题及calc函数适配咨询

问题背景 #

我正在通过Codewars练习Haskell，编写了如下代码：

import Data.Char(digitToInt)

calc :: String -> Int
calc s = sum $ map (\i -> (digitToInt i)^n) s
        where n = length s

narcissistic :: Integral n => n -> Bool
narcissistic n = calc s == n
                where s = show n

运行后出现错误：

Couldn't match expected type 'Int' with actual type 'n'
      'n' is a rigid type variable bound by
        the type signature for:
          narcissistic :: forall n. Integral n => n -> Bool

我明白这是因为narcissistic函数接收的输入n属于Integral类型类（如Int或Integer），但calc函数返回Int类型，导致无法比较两者结果。想知道：

• 处理这类类型类的理想方式是什么？
• 如何实现适配Integral类型类的calc函数（目前因digitToInt受限无法实现）？

解决方案 #

一、处理类型类不匹配的核心思路 #

核心是让计算结果的类型与输入类型保持一致，或通过类型转换在兼容类型间安全转换。对于Integral类型类，可利用它提供的fromIntegral转换函数，或让函数本身支持多类型输出，避免硬编码具体类型。

二、适配Integral类型类的calc实现 #

问题根源在于digitToInt返回Int、length返回Int，与narcissistic的通用Integral输入类型不兼容。我们可以通过fromIntegral统一类型，修改后的代码如下：

import Data.Char(digitToInt)

-- 让calc支持返回任意Integral类型
calc :: Integral a => String -> a
calc s = sum $ map (\c -> fromIntegral (digitToInt c) ^ fromIntegral n) s
        where n = length s

narcissistic :: Integral n => n -> Bool
narcissistic n = calc (show n) == n

更简洁的列表推导写法：

import Data.Char(digitToInt)

calc :: Integral a => String -> a
calc s = let digitCount = length s
         in sum [ fromIntegral (digitToInt c) ^ fromIntegral digitCount | c <- s ]

narcissistic :: Integral n => n -> Bool
narcissistic n = calc (show n) == n

三、关键细节解释 #

• fromIntegral的作用：它能将任意Integral类型的值转换为另一个Num类型（这里转为目标Integral类型），解决了digitToInt、length的返回类型与目标类型不匹配的问题。
• 多态类型签名：给calc加上Integral a => String -> a的签名，让它可以返回任意Integral类型，完美适配narcissistic的输入类型n。
• 幂运算类型一致性：确保幂运算的底数和指数都是同一Integral类型，避免类型不匹配错误。

四、简化思路：限定具体类型 #

如果不需要支持所有Integral类型，也可以直接把narcissistic的输入限定为Int，这样calc的返回类型直接匹配，代码更简单：

import Data.Char(digitToInt)

calc :: String -> Int
calc s = sum $ map (\i -> (digitToInt i)^n) s
        where n = length s

narcissistic :: Int -> Bool
narcissistic n = calc (show n) == n

但这种方式灵活性较差，无法处理Integer等其他Integral类型。

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