如何计算有序 [Char] 中 Char 的出现次数？

Question

我有一个输入 [String] 如下：

main = do
  let res = calcCounts ["IOIOOIOOI\r","OOIIIOOO\r",...]

我想使用非统一的基本系统按顺序计数I 和O 实例。

我创建了一个Type：

data OutputState = OutputState { totNumI :: Int,
                                 totNumO :: Int,
                                 numI :: Int,
                                 numO :: Int
                               } deriving (Show)

到目前为止，我创建的代码尝试单步执行输入并每次创建一个新的OutputState，同时考虑之前的OutputState，然后再达到每个[Char] 的完整条件'\r' 和[String] 的列表末尾但不完整：

calcCounts :: [String] -> [OutputState]
calcCounts input = map genOutputState input

genOutputState :: [Char] -> OutputState
genOutputState (x:xs)
  | x == '\r' = --return final OutputState 
  | otherwise do                 
      let currOutputState = incrementOutputState x
      genOutputState xs --also pass current OutputState back each recursion

incrementOutputState :: OutputState -> Char -> OutputState
incrementOutputState pos@prevOS x = do
  | if x == 'I' = do 
      let currOutputState = OutputState pos{totNumA} pos{totNumB} pos{numA}+1 pos{numB}
      | if currOutputState{numA} == 4 = OutputState currOutputState{totNumA}+1 currOutputState{totNumB} 0 0         
  | otherwise = do 
      let currentOutputState = OutputState pos{totNumA} pos{totNumB} pos{numA} pos{numB}+1
      | if currentOutputState{numB} == 4 = OutputState currOutputState{totNumA} currOutputState{totNumB}+1 0 0

最终输出应该是[OutputState,OutputState,..]

我知道我的代码目前不合法并且缺少逻辑。对此表示歉意。总的来说，我仍在回归 Haskell 和函数式编程。在命令式范例中，我会使用可变变量来跟踪必要的计数，但要理解这不是从功能意义上考虑问题的正确方法。

非常感谢任何帮助！

Answer 1

我认为这可以满足您的要求，尽管我仍然不完全确定我是否理解正确。

data OutputState = OutputState { totNumI :: Int,
                                 totNumO :: Int,
                                 numI :: Int,
                                 numO :: Int
                               } deriving (Show)

genOutputState :: String -> OutputState
genOutputState = foldl incrementOutputState (OutputState 0 0 0 0)

incrementOutputState :: OutputState -> Char -> OutputState
incrementOutputState (OutputState ti to 3 no)  'I' = OutputState (ti + 1) to 0 0
incrementOutputState (OutputState ti to ni no) 'I' = OutputState ti to (ni + 1) no
incrementOutputState (OutputState ti to ni 3)  'O' = OutputState ti (to + 1) 0 0
incrementOutputState (OutputState ti to ni no) 'O' = OutputState ti to ni (no + 1)
incrementOutputState os _                          = os -- Do not increment anything for characters other than 'I' and 'O'

我建议你多学习一些 Haskell 教程，因为我觉得你还不太了解语法和函数式思维方式。