问题 State Monad具有多个州的价值观


考虑以下:

do
  x1 <- new 2
  set x1 3
  x2 <- get x1
  y1 <- new 10
  set y1 20
  y2 <- get y1
  return (x2 + y2)

我想要这个结果 23。有没有办法在纯Haskell中实现这样的东西,如果是这样的话怎么样?我明白 STRef 做这样的事情,但我只是想在普通的Haskell中做到这一点(现在不担心效率)。我认为我必须创建一个数据类型并使其成为一个实例 Monad,但我不确定细节,所以一个有用的例子会有所帮助。


3617
2017-10-07 14:27


起源

您的问题是关于实现可变引用还是您还需要学习如何实现纯状态monad? - hugomg
纯净的状态。不需要实际的可变实现。如果这已经存在,那么hackage页面的链接就可以了。这个问题: hackage.haskell.org/package/mtl-2.1.2/docs/... 是一个似乎能够 put 整个州,不是特定的“变量” - Clinton
在你的代码中, x1 和 y1 似乎是变量,而 x2 和 y2 似乎是价值观;这些必须有不同的类型,所以我发现命名令人惊讶。你知道吗? - Joachim Breitner
我不知道,但是,我明白了你的观点。命名是普通的,但我现在不会改变它。 - Clinton
不确定你的意思是什么。在Haskell中,类型的变量 Integer某种状态下与命名引用完全不同的东西;我真的建议不要在那里给出令人困惑的名字。 - Joachim Breitner


答案:


这允许多个值,但它更毛茸茸:)这很好地简化了丹尼尔的建议 Dynamic

import Data.Dynamic
import Data.Maybe
import Control.Monad.State
import Data.Map as M

newtype Ref a = Ref {ref :: Int}

type MutState = State (Int, Map Int Dynamic)

val :: Typeable a => Ref a -> MutState a
val r = snd `fmap` get >>= 
        return . fromJust . (>>= fromDynamic) .  M.lookup (ref r)

new :: Typeable a => a -> MutState (Ref a)
new a = do
  (curr, binds) <- get
  put (curr + 1, M.insert (curr + 1) (toDyn a) binds)
  return . Ref $ curr + 1

set :: Typeable a => Ref a -> a -> MutState ()
set (Ref i) a = do
  (c, m) <- get
  put (c, M.insert i (toDyn a) m)

runMut :: MutState a -> a
runMut = flip evalState (0, M.fromList [])

然后使用它

default (Int) -- too lazy for signatures :)
test :: Int
test = runMut $ do
  x1 <- new 2
  set x1 3
  x2 <- val x1
  y1 <- new 10
  set y1 20
  y2 <- val y1
  return (x2 + y2)

Ref基本上是 Ints附上一些类型信息和 val 会查找合适的 Dynamic 并尝试强制它进入正确的类型。

如果这是真正的代码,你应该隐藏实现 Ref 和 MutState。为方便起见,我 fromJust回归 val 如果你想要一个安全的实现,我想你可以分层 State 和 Maybe monads处理未绑定的变量。

如果您担心类型限制,如上所示,它们可以简单地推导出来。


6
2017-10-07 15:02



为什么重新实现 Dynamic? - Daniel Wagner
@DanielWagner Simplified谢谢!你知道有一天我会发布一个答案而你没有发表评论,显示它是多么不必要的过于复杂;) - jozefg


已经有一个实现 Control.Monad.State但是一般来说这很麻烦:一个复杂来自MonadState类,另一个复杂来自于plain State 以更一般的方式实施 StateT

以下是使用该实现的任务示例。没有使用可变性。请注意,您的示例按原样粘贴,只是添加 x 字首:

import Control.Monad.State
import qualified Data.Map as M

type MyMap a = M.Map Int a
type MyState a b = State (MyMap a) b
type MyRef = Int

xrun :: MyState a b -> b
xrun x = evalState x (M.empty)

mget :: MyState a (MyMap a)
mget = get

mput :: MyMap a -> MyState a ()
mput = put

mmodify :: (MyMap a -> MyMap a) -> MyState a ()
mmodify x = modify x

xnew :: s -> MyState s MyRef
xnew val = do
    s <- mget
    let newRef = if M.null s then 0 else fst (M.findMax s) + 1
    mput $ M.insert newRef val s
    return newRef

xset :: MyRef -> a -> MyState a () 
xset ref val = modify $ M.insert ref val

xget :: MyRef -> MyState a a
xget ref = fmap (\s -> case M.lookup ref s of Just v -> v) get

test :: MyState Int Int
test = do
  x1 <- xnew 2
  xset x1 3
  x2 <- xget x1
  y1 <- xnew 10
  xset y1 20
  y2 <- xget y1
  return (x2 + y2)

main = print $ xrun test

可以实现模块中的所有功能 >>=/return 没有使用库存实现 Control.Monad 保留签名。

这里是:

module MyState (State, get, put, modify, evalState) where

newtype State s a = State (s -> (a, s))

evalState :: State s a -> s -> a
evalState (State f) = fst . f

instance Monad (State s) where
    return a = State $ \s -> (a, s)
    State f >>= g = State $ \s -> 
        case f s of 
            (a', s') -> case g a' of 
                State h -> h s'

instance Functor (State s) where
    fmap f (State g) = State $ 
        \s -> case g s of (a, s) -> (f a, s) 

get :: State s s
get = State (\s -> (s, s))

put :: s -> State s ()
put s = State $ \_ -> ((), s)

modify :: (s -> s) -> State s ()
modify f = get >>= put . f

保存到 MyState.hs 并替换 import Control.Monad.State 同 import MyState


5
2017-10-07 14:51



我首先得到了这个答案(得到了投票然后被我删除),但如果你仔细观察,你会发现克林顿想要跟踪状态中的变量,而不仅仅是一个状态,所以类似于 ST 单子。 - Joachim Breitner
我觉得有点像 type MyState a b = State (M.Map Int a, Int) b 会为他做这个工作。我现在正在详细解释答案 - nponeccop
这就是我在下面写的。请注意,您不需要第二个Int,您可以轻松(并且有效)获取Map中的最大索引。 - Joachim Breitner


State 要么 StateT 你可以效仿它(State 只允许1个值)。最简单的方法是使用 Map

 do
  put empty
  set "x1" 3  
  x2 <-  getKey "x1"
  set "y1" 20
  y2 <-  getKey "y1"
  return (x2 + y2)
    where
      getKey k = fromJust . (lookup k) `fmap` get
      set = modify replace
      replace d k m = if k `member` m then update (\_ -> Just d) k m
                      else insert k d m

2
2017-10-07 14:56



该 let x1 = 2 是死代码;你为什么把它放在那里? - Joachim Breitner
我宁愿不在地图中查找字符串。我知道我说我并不担心效率,但这似乎有点像黑客。 - Clinton
@Clinton如果你有200 ..值,它效率不高。这只是 O(log n) - wit


StateT for a tuple怎么样?

flip evalState (2, 10) $ do
  modify $ \(_, y) -> (3, y)
  x2 <- fst <$> get
  modify $ \(x, _) -> (x, 20)
  y2 <- snd <$> get
  return (x2 + y2)

如果你真的想要可变的单元格,我建议使用ST,STM或IO而不是StateT。使用StateT在异构映射中实现从增加自然到对象的实现似乎是可能的,但可能有点尴尬。


2
2017-10-17 15:58