File: Permutation.fs

package info (click to toggle)
fsharp 3.1.1.26%2Bdfsg2-3
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 59,244 kB
  • ctags: 4,190
  • sloc: cs: 13,398; ml: 1,098; sh: 399; makefile: 293; xml: 82
file content (56 lines) | stat: -rwxr-xr-x 1,876 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// (c) Microsoft Corporation 2005-2009. 

namespace Microsoft.FSharp.Math

type Permutation = int -> int

type permutation = int -> int

[<RequireQualifiedAccess>]
[<CompilationRepresentation(CompilationRepresentationFlags.ModuleSuffix)>]
module Permutation =

    let invalidArg arg msg = raise (new System.ArgumentException((msg:string),(arg:string)))        

    let ofFreshArray (arr:_[]) = 
        let arr2 = Array.zeroCreate arr.Length
        for i = 0 to arr.Length - 1 do 
            let x = arr.[i] 
            if x < 0 || x >= arr.Length then invalidArg "arr" "invalid permutation" 
            arr2.[x] <- 1
        for i = 0 to arr.Length - 1 do 
            if arr2.[i] <> 1 then invalidArg "arr" "invalid permutation"
        (fun k -> arr.[k])

    let ofArray (arr:_[]) = arr |> Array.copy |> ofFreshArray

    let of_array (arr:_[]) = ofArray arr

    let ofPairs  (mappings: seq<int * int>) = 
      let p = dict mappings 
      (fun k -> if p.ContainsKey k then p.[k] else k)

    let of_pairs  (mappings: seq<int * int>) =  ofPairs mappings

    let swap (n:int) (m:int) = 
      (fun k -> if k = n then m elif k = m then n else k)

    let reversal size = 
      if size <= 0 then invalidArg "size" "a permutation size must be positive";
      (fun k -> (size - 1 - k))

    let rotation size distance = 
      if size <= 0 then invalidArg "size" "a permutation size must be positive";
      if abs distance >= size then invalidArg "distance" "the absolute value of the distance must be less than the size of the permutation";
      (fun k -> (k + size + distance) % size)

    let identity (k:int) = k
    
    let inverse size p =
        if size <= 0 then invalidArg "size" "a permutation size must be positive";
        let arr2 = Array.zeroCreate size
        for i = 0 to size - 1 do
             arr2.[p i] <- i
        ofFreshArray arr2