1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
import qualified Data.Vector.Unboxed.Mutable as MV
import Data.Vector.Unboxed ((!))
import qualified Data.Vector.Unboxed as V
import Control.Monad (forM_)
similar :: V.Vector Char -> Int
similar v = l + sum (map (similar' 1 1) checks)
where
(l,checks) = let h = V.head v in V.foldl'
(\(i,is) c -> if c == h then (i+1,i:is) else (i+1,is)) (1,[]) (V.tail v)
similar' !r !n !i = if i < l1 && v!(n) == v!(i+1) then similar' (r+1) (n+1) (i+1)
else r
main :: IO ()
main = do
n <- getLine
v <- MV.replicate (read n) 0
forM_ [1..read n] $ \n' -> do
v' <- getLine
MV.unsafeWrite v (n'1) (similar . V.fromList $ v')
V.unsafeFreeze v >>= V.mapM_ print
|