blob: 933bd8eb2ec94be2e72336de4e0eb77d2bf5e038 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
module Algo.ListRank
where
import Data.Vector.Unboxed as V
listRank :: Int -> Vector Int
{-# NOINLINE listRank #-}
listRank n = pointer_jump xs val
where
xs = 0 `V.cons` V.enumFromTo 0 (n-2)
val = V.zipWith (\i j -> if i == j then 0 else 1)
xs (V.enumFromTo 0 (n-1))
pointer_jump pt val
| npt == pt = val
| otherwise = pointer_jump npt nval
where
npt = V.backpermute pt pt
nval = V.zipWith (+) val (V.backpermute val pt)
|