如何定义具有不同类型输入和输出的函数？

Question

我尝试了以下代码：

data InputType = NumType Int Int | StrType String String
data OutputType = Either Int String
add' :: InputType -> InputType -> OutputType
add' (NumType a b) = a + b
add' (StrType a b) = a ++ b

但失败了:(

Answer 1

您对OutputType的声明并未指明它可以是Int还是String;相反，您创建了一个新的对，需要Int 和一个String。您的数据构建器Either恰好与类型构造函数Either具有相同的名称。

我认为你的意思是

type OutputType = Either Int String

在这种情况下，如果使用正确的构造函数，则可以定义函数

add' :: InputType -> OutputType
add' (NumType a b) = Left (a + b)
add' (StrType a b) = Right (a ++ b)

Answer 2

您可以使用GADT在值级别表示您获得的输入类型：

{-# LANGUAGE GADTs #-}

module Add where

data SType a where
  SString :: SType String
  SInt    :: SType Int

add :: SType a -> a -> a -> a
add SString = (++)
add SInt    = (+)

我致电GADT SType，因为它是singleton type。