File: ListRank.hs

package info (click to toggle)
haskell-vector 0.6.0.1-1
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 632 kB
  • ctags: 20
  • sloc: haskell: 7,341; ansic: 23; makefile: 2
file content (21 lines) | stat: -rw-r--r-- 495 bytes parent folder | download | duplicates (8) 
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)