Я определил следующую функцию в haskell:
step :: [Int] -> [Char] -> [Int]
step stack str
| str == "*" = remaining ++ [x*y]
| str == "+" = remaining ++ [x+y]
| str == "-" = remaining ++ [x-y]
| str == "/" = remaining ++ [x `div` y]
| otherwise = stack ++ [read str :: Int]
where x = (last . init) stack
y = (last stack)
remaining = (init . init) stack
Эта функция принимает целочисленный массив [10, 4, 3]
и строковый оператор *
и применяет оператор к последним двум элементам в массиве ивозвращает следующий массив [10, 7]
.
. Это составляет часть промежуточной функции, конечный результат - функция оценки обратной польской записи.
Как я могу использовать step
функция, которую я определил, и foldl
для выполнения следующих действий: :
Взять строку примеров: "10 4 3 + 2 * -"
.
Добавлять каждый элемент в строку до первого оператора.встречается так:
10, 4, 3
Затем примените оператор к двум элементам сверху стека и поместите результат в стек:
10, 7
.
Продолжайте так до тех пор, пока не будет оценен окончательный ответ (-4
)
Ответ:
Ради полноты этой функции я пришел с помощью@ talex
rpn :: String -> Int
rpn input = head result
where arr = words input
result = foldl step [] arr