MVectorのmaxIndex

Question

Data.Vectorには、そのベクトルのmaximum値を返すタイプmaxIndex :: (Ord a) => Vector a -> Intの関数maxIndexが含まれています。私は可変ベクトルで作業していますが、MVectorにはmaxIndexが定義されていません。

私は持っているデータを取得する最良の方法は何ですかMVector私は持っていますか？私のコードは、現在、次のとおりです。

i

import qualified Data.Vector.Unboxed.Mutable as MV 
import   Control.Monad.ST 
import   Control.Monad    (mapM_) 

type MaxIndex = Int 

step :: forall s. MV.MVector s Int -> MaxIndex -> ST s() 
step vec i = do 
    n <- MV.unsafeRead vec i 
    MV.write vec i 0 
    let l  = MV.length vec 
     (k, x) = n `divMod` l 
    mapM_ (\j -> MV.modify vec (+k) j) [0..l-1] -- side note, this is just 
               -- fmap (+k) vec, but MVector is not 
               -- a functor. Is there a better way? 
    mapM_ (\j -> MV.modify vec (+1) (j `mod` l)) [i+1..i+x] 

は私がstepの内側に誘導するために探していたインデックスです。私はこれをやっています。なぜなら、ここでの行動は最終的にuntilの中にラップされ、述語が満たされるまで繰り返され、毎回の凍結と解凍は馬鹿げたほど高価に聞こえるからです。

Answer 1

このメモリを後で変更しようとしているので、安全でない凍結について多くの話題があります。したがって、unsafeFreezeを呼び出すときに暗黙のうちに保証していることに違反します。

私の提案は、命令型のmaxIndex関数を書くことです。以下は入力されていますが、テストされていません：

import qualified Data.Vector.Unboxed.Mutable as MV 
import   Control.Monad.ST 
import   Control.Monad    (mapM_) 

maxIndex :: (Ord a, MV.Unbox a) => MV.MVector s a -> ST s (Maybe Int) 
maxIndex mv | len == 0 = pure Nothing 
      | otherwise = Just <$> go 0 0 
where 
len = MV.length mv 
go n i | i >=len = pure n 
     | otherwise = do 
    nVal <- MV.unsafeRead mv n 
    iVal <- MV.unsafeRead mv i 
    if nVal < iVal then go i (i+1) 
        else go n (i+1) 

Answer 2

unsafeFreezeと速いと思われるベクトルをフリーズすることを検討しましたか（つまりΘ(1)）？たとえば、次のように変更可能なベクターについてmaxIndexを定義することができます

maxIndex = fmap V.maxIndex . V.unsafeFreeze 

これは、あなたが次のことをインポートしていることを前提としています

import qualified Data.Vector.Unboxed   as V 

unsafeFreezeは、実際にデータをコピーしないと高速である必要がありますが、それこのアプローチが明示的なループに比べて実際に速いかどうかを見るために基準ベンチマークを実行することは面白いでしょう。