foldingFunction (x:y:ys) "-" = (y - x):ys
foldingFunction xs numberString = read numberString:xs
Аккумулятором свёртки был стек, который мы представили списком значений типа Double
. Если по мере того, как функция проходила по выражению в обратной польской записи, текущий элемент являлся оператором, она снимала два элемента с верхушки стека, применяла между ними оператор, а затем помещала результат обратно в стек. Если текущий элемент являлся строкой, представляющей число, она преобразовывала эту строку в фактическое число и возвращала новый стек, который был как прежний, только с этим числом, протолкнутым на верхушку.
Давайте сначала сделаем так, чтобы наша функция свёртки допускала мягкое окончание с неудачей. Её тип изменится с того, каким он является сейчас, на следующий:
foldingFunction :: [Double] –> String –> Maybe [Double]
Поэтому она либо вернёт новый стек в конструкторе Just
, либо потерпит неудачу, вернув значение Nothing
.
Функция reads
похожа на функцию read
, за исключением того, что она возвращает список с одним элементом в случае успешного чтения. Если ей не удалось что-либо прочитать, она возвращает пустой список. Помимо прочитанного ею значения она также возвращает ту часть строки, которую она не потребила. Мы сейчас скажем, что она должна потребить все входные данные для работы, и превратим её для удобства в функцию readMaybe
. Вот она:
readMaybe :: (Read a) => String –> Maybe a
readMaybe st = case reads st of [(x, "")] –> Just x
_ –> Nothing
Теперь протестируем её:
ghci> readMaybe "1" :: Maybe Int
Just 1
ghci> readMaybe "ИДИ К ЧЁРТУ" :: Maybe Int
Nothing
Хорошо, кажется, работает. Итак, давайте превратим нашу функцию свёртки в монадическую функцию, которая может завершаться неудачей:
foldingFunction :: [Double] –> String –> Maybe [Double]
foldingFunction (x:y:ys) "*" = return ((y * x):ys)
foldingFunction (x:y:ys) "+" = return ((y + x):ys)
foldingFunction (x:y:ys) "-" = return ((y - x):ys)
foldingFunction xs numberString = liftM (:xs) (readMaybe numberString)
Первые три случая – такие же, как и прежние, только новый стек обёрнут в конструктор Just
(для этого мы использовали здесь функцию return
, но могли и просто написать Just
). В последнем случае мы используем вызов readMaybe numberString
, а затем отображаем это с помощью (:xs)
. Поэтому если стек равен [1.0,2.0]
, а выражение readMaybe numberString
даёт в результате Just
3.0
, то результатом будет [3.0,1.0,2.0]
. Если же readMaybe numberString
даёт в результате значение Nothing
, результатом будет Nothing
.
Давайте проверим функцию свёртки отдельно:
ghci> foldingFunction [3,2] "*"
Just [6.0]
ghci> foldingFunction [3,2] "-"
Just [-1.0]
ghci> foldingFunction [] "*"
Nothing
ghci> foldingFunction [] "1"
Just [1.0]
ghci> foldingFunction [] "1 уа-уа-уа-уа"
Nothing
Похоже, она работает! А теперь пришла пора для новой и улучшенной функции solveRPN
. Вот она перед вами, дамы и господа!
import Data.List
solveRPN :: String –> Maybe Double
solveRPN st = do
[result] <– foldM foldingFunction [] (words st)
return result