25
\$\begingroup\$

The challenge is to write codegolf for the Hafnian of a matrix. The Hafnian of an \$2n \times 2n\$ symmetric matrix \$A\$ is defined as:

$$\operatorname{haf}(A) = \frac 1 {n!2^n} \sum_{\sigma \in S_{2n}} \prod^n_{j=1} A_{\sigma(2j-1),\sigma(2j)}$$

Here \$S_{2n}\$ represents the set of all permutations of the integers from \$1\$ to \$2n\$, that is \$[1, 2n]\$.

The Wikipedia link talks about adjacency matrices but your code should work for any real valued symmetric input matrices.

For those interested in applications of the Hafnian, the mathoverflow link discusses some more.

Your code can take input however it wishes and give output in any sensible format but please include in your answer a full worked example including clear instructions for how to supply input to your code.

The input matrix is always square and will be at most 16 by 16. There is no need to be able to handle the empty matrix or matrices of odd dimension.

Reference implementation

Here is some example python code from Mr. Xcoder.

from itertools import permutations
from math import factorial

def hafnian(matrix):
    my_sum = 0
    n = len(matrix) // 2
    for sigma in permutations(range(n*2)):
        prod = 1
        for j in range(n):
            prod *= matrix[sigma[2*j]][sigma[2*j+1]]
        my_sum += prod
    return my_sum / (factorial(n) * 2 ** n)


print(hafnian([[0, 4.5], [4.5, 0]]))
4.5
print(hafnian([[0, 4.7, 4.6, 4.5], [4.7, 0, 2.1, 0.4], [4.6, 2.1, 0, 1.2], [4.5, 0.4, 1.2, 0]])
16.93
print(hafnian([[1.3, 4.1, 1.2, 0.0, 0.9, 4.4], [4.1, 4.2, 2.7, 1.2, 0.4, 1.7], [1.2, 2.7, 4.9, 4.7, 4.0, 3.7], [0.0, 1.2, 4.7, 2.2, 3.3, 1.8], [0.9, 0.4, 4.0, 3.3, 0.5, 4.4], [4.4, 1.7, 3.7, 1.8, 4.4, 3.2]])
262.458

The wiki page has now (March 2 2018) been updated by ShreevatsaR to include a different way of calculating the Hafnian. It would be very interesting to see this golfed.

\$\endgroup\$
1
  • 9
    \$\begingroup\$ I think this would be easier to digest with an informal explanation of the hafnian. Something like, take all subsets of n matrix entries where their n row indices and n column indices form a partition of 1..2n, take the product of each, add them, and scale the sum. \$\endgroup\$
    – xnor
    Commented Mar 1, 2018 at 17:11

17 Answers 17

9
\$\begingroup\$

R, 150 142 127 119 bytes

function(A,N=nrow(A),k=1:(N/2)*2)sum(apply(gtools::permutations(N,N),1,function(r)prod(A[cbind(r[k-1],r[k])])))/prod(k)

Try it online!

Uses the same trick I discovered golfing down this answer to index the matrix P, and @Vlo suggested an approach to entirely remove the for loop for -6 bytes!

To create a new test case, you can do matrix(c(values,separated,by,commas,going,across,rows),nrow=2n,ncol=2n,byrow=T).

Explanation: (the code is the same; it uses an apply rather than a for loop but the logic is otherwise identical).

function(A){
N <- nrow(A)                   #N = 2*n
k <- 1:(N/2) * 2               #k = c(2,4,...,N) -- aka 2*j in the formula
P <- gtools::permutations(N,N) #all N-length permutations of 1:N
for(i in 1:nrow(P))
 F <- F + prod(A[cbind(P[i,k-1],P[i,k])]) # takes the product of all A_sigma(2j-1)sigma(2j) for fixed i and adds it to F (initialized to 0)
F / prod(k)                    #return value; prod(k) == n! * 2^n
}

\$\endgroup\$
3
  • \$\begingroup\$ Apply is cheaper by 2 bytes, which allows for additional 4 bytes of saving by cramming the other lines together. tio.run/##PY6xDoIwEIZ3nsLxzpxiS4ymkYEXYHIjDFDEEKBtSokS47PX4sDw5/… It is also quite interesting how base R lacks a permutation function for a statistical programming language \$\endgroup\$
    – Vlo
    Commented Mar 1, 2018 at 21:36
  • \$\begingroup\$ @Vlo very nice! we can move N and k into function arguments to get it to one statement, removing the {} and saving another two bytes. \$\endgroup\$
    – Giuseppe
    Commented Mar 1, 2018 at 21:41
  • \$\begingroup\$ @Giuseppe Darn keeps on forgetting that you can define them in the function args. Spent a few minutes trying to ember those variables... \$\endgroup\$
    – Vlo
    Commented Mar 1, 2018 at 21:42
8
\$\begingroup\$

Pyth, 24 bytes

sm*Fmc@@Qhkek2d{mScd2.pU

Try it here!


Old version, 35 bytes

*c1**FK/lQ2^2Ksm*Fm@@Q@[email protected]

Try it here!

\$\endgroup\$
6
  • 3
    \$\begingroup\$ Currently the lead but you have to fear the Jelly answers to come.... :) \$\endgroup\$
    – user9206
    Commented Mar 1, 2018 at 17:25
  • \$\begingroup\$ Eh Jelly will definitely beat my by about 10 bytes. Pyth ain’t the best tool for the job \$\endgroup\$
    – Mr. Xcoder
    Commented Mar 1, 2018 at 17:26
  • \$\begingroup\$ 05AB1E is looking like it may even tie Pyth (believe it or not, finally a matrix challenge where a[b] is enough to compete). \$\endgroup\$ Commented Mar 1, 2018 at 17:31
  • \$\begingroup\$ @MagicOctopusUrn I already have an 05AB1E solution that beats Pyth :-) Not going to post it (for now, at least) \$\endgroup\$
    – Mr. Xcoder
    Commented Mar 1, 2018 at 17:32
  • \$\begingroup\$ Is it something along the lines of xÍysè<¹sès·<ysè<è lmao? P.S. Mine is 40 bytes and isn't working so well hah, so feel free to post it, unsure I'm able to finish before I have to go home. \$\endgroup\$ Commented Mar 1, 2018 at 17:38
7
\$\begingroup\$

Stax, 23 22 19 17 bytes

ü;Y╙◘▌Φq↓ê²╧▐å↑┌C

Run and debug it online

The corresponding ascii representation of the same program is this.

%r|TF2/{xsE@i^H/m:*+

The program suffers from some floating point rounding error. In particular, it reports 33673.5000000011 instead of 33673.5. But I think the accuracy is acceptable given that this program operates on floating point values. It's also very slow, taking almost a minute for the example inputs on this machine.

%                             get size of matrix
 r|T                          get all permutations of [0 ... size-1]
    F                         for each, execute the rest of the program
     2/                       get consecutive pairs
       {        m             map each pair... 
        xsE@                      the matrix element at that location
            i^H/                  divided by 2*(i+1) where i=iteration index
                 :*           product of array
                   +          add to running total
\$\endgroup\$
5
  • 1
    \$\begingroup\$ Very impressive! \$\endgroup\$
    – user9206
    Commented Mar 1, 2018 at 20:11
  • \$\begingroup\$ Finally, I actually see someone using Stax! I thought I was the only person that knew about it. \$\endgroup\$ Commented May 28, 2020 at 13:15
  • \$\begingroup\$ I know a lot about it, but I'm equally surprised to find someone else who knows about it. :) \$\endgroup\$
    – recursive
    Commented May 28, 2020 at 15:01
  • \$\begingroup\$ you created Stax?! No way! ... also could I create a chat room for Stax? I don't know, I want to chat with the legendary tomtheisen (at least legendary to me) \$\endgroup\$ Commented May 29, 2020 at 22:05
  • \$\begingroup\$ I used to have a chat room some months ago, but it's fallen into disuse. I have no problem with you creating a chat room. The idea of being legendary over is pretty amusing to me. :) \$\endgroup\$
    – recursive
    Commented May 31, 2020 at 15:59
6
\$\begingroup\$

05AB1E, 21 bytes

ā<œε2ô{}Ùεε`Isèsè;]PO

Try it online!


Old version, 32 bytes

āœvIg;©Lε·UIyX<èèyXèè}P}Oθ®!/®o/

Try it online!

How it works?

āœvIg;©Lε·UIyX<èèyXèè}P}Oθ®!/®o/ – Full program. Argument: A matrix M.
ā                                – The range [1 ... len(M)].
 œ                               – Permutations.
  v                    }         – Iterate over the above with a variable y.
   Ig;©                          – Push len(M) / 2 and also store it in register c.
       Lε            }           – For each integer in the range [1 ... ^]:
         ·U                      – Double it and store it in a variable X.
            yX<                  – Push the element of y at index X-1.
           I   è                 – And index with the result into M.
                yXè              – Push the element of y at index X.
                   è             – And index with the result into ^^.
                      P          – Take the product of the resulting list.
                        O        – Sum the result of the mapping.
                         θ       – And take the last element*.
                          ®!     – Take the factorial of the last item in register c.
                             ®o  – Raise 2 to the power of the last item in register c.
                            /  / – And divide the sum of the mapping accordingly.

* – Yeah, this is needed because I mess up the stack when pushing so many values in the loop and not popping correctly ;P
\$\endgroup\$
1
  • 2
    \$\begingroup\$ No kidding èsè, hah... haha... I'm punny. \$\endgroup\$ Commented Mar 2, 2018 at 17:13
6
\$\begingroup\$

R, 84 78 bytes

h=function(m)"if"(n<-nrow(m),{for(j in 2:n)F=F+m[1,j]*h(m[v<--c(1,j),v]);F},1)

Try it online!

Edit: Thanks to Vlo for -6 bytes.

It seems that everybody here is implementing the standard reference algorithm with permutations, but I tried to take advantage of the community knowledge gained in the related challenge, which is basically the same task targeted for fastest code instead of golf.

It turns out that for a language that is good at slicing matrices (like R), the recursive algorithm: hafnian(m) = sum(m[i,j] * hafnian(m[-rows and columns at i,j]) is not only faster, but also quite golfy. Here is the ungolfed code:

hafnian<-function(m)
{
    n=nrow(m)
    #Exits one step earlier than golfed version
    if(n == 2) return(m[1,2])
    h = 0
    for(j in 2:n) {
        if(m[1,j] == 0) next
        h = h + m[1,j] * hafnian(m[c(-1,-j),c(-1,-j)])
    }
    h
}
\$\endgroup\$
4
  • \$\begingroup\$ Very nice answer. -1 for calling If with parenthesis, -4 for using F as initialized variable, -1 for assigning n within the if. tio.run/##XU/LCsIwELz7FcFTVtOQl1pf1/… \$\endgroup\$
    – Vlo
    Commented Mar 27, 2018 at 21:35
  • \$\begingroup\$ neat! I'd say post it in the speed challenge, but there are probably some more optimizations (like threading) that can be made, and while R isn't exactly known for its speed, it'd be good to have it there for reference. \$\endgroup\$
    – Giuseppe
    Commented Mar 28, 2018 at 12:44
  • \$\begingroup\$ Do it for benchmark purposes! \$\endgroup\$
    – Vlo
    Commented Mar 28, 2018 at 15:32
  • \$\begingroup\$ I actually tried testing this for speed, but got quickly discouraged by the results. The slowest Python submission in the speed challenge using the same exact algorithm crunches 24x24 matrix in a few seconds on TIO, but R times out. On my local machine it also didn't respond in a reasonable time, even when aided with memoization from package 'memo'... \$\endgroup\$
    – Kirill L.
    Commented Mar 29, 2018 at 8:48
5
\$\begingroup\$

Jelly, 19 bytes

LŒ!s€2Ṣ€QḅL_LịFHP€S

Try it online!

Alternate version, 15 bytes, postdates challenge

LŒ!s€2Ṣ€QœịHP€S

Jelly finally got n-dimensional array indexing.

Try it online!

How it works

LŒ!s€2Ṣ€QœiHP€S  Main link. Argument: M (matrix / 2D array)

L                Take the length, yielding 2n.
 Œ!              Generate all permutations of [1, ..., 2n].
   s€2           Split each permutation into pairs.
      Ṣ€         Sort the pair arrays.
        Q        Unique; deduplicate the array of pair arrays.
                 This avoids dividing by n! at the end.
           H     Halve; yield M, with all of its elements divided by 2.
                 This avoids dividing by 2**n at the end.
         œị      At-index (n-dimensional); take each pair of indices [i, j] and
                 yield M[i][j].
            P€   Take the product the results corresponding the same permutation.
              S  Take the sum of the products.

The 19-byte version works in a similar fashion; it just has to implement œị itself.

...ḅL_LịFH...    Return value: Array of arrays of index pairs. Argument: M

    L            Length; yield 2n.
   ḅ             Convert each pair of indices [i, j] from base 2n to integer,
                 yielding ((2n)i + j).
     _L          Subtract 2n, yielding ((2n)(i - 1) + j).
                 This is necessary because indexing is 1-based in Jelly, so the
                 index pair [1, 1] must map to index 1.
        F        Yield M, flattened.
       ị         Take the indices to the left and get the element at these indices
                 from the array to the right.
         H       Halve; divide all retrieved elements by 2.
\$\endgroup\$
0
3
\$\begingroup\$

C (gcc), 288 285 282 293 292 272 271 263 bytes

  • Saved three bytes by fiddling with two post-increments and for loop placement.
  • Saved three bytes by fiddling with anothter post-increment, moving both variable initializations before the branch -- golfed if(...)...k=0...else...,j=0... to if(k=j=0,...)...else... -- and performed an index shift.
  • Required eleven bytes by supporting float matrices.
  • Saved a byte thanks to Mr. Xcoder; golfing 2*j+++1 to j-~j++.
  • Saved twenty bytes by removing a superfluous int variable type declaration and not using a factorial function but instead calculating the factorial value by using an already existing for loop.
  • Saved a byte by golfing S=S/F/(1<<n); to S/=F*(1<<n);.
  • Saved eight bytes thanks to ceilingcat.
float S,p,F;j,i;s(A,n,P,l,o,k)float*A;int*P;{if(o-l)for(k=0;k<l;s(A,n,P,l,o+1))P[o]=k++;else{for(p=j=l;j--;)for(i=l;i--;)p-=P[j]==P[i];if(!p){for(F=p=1,j=0;j<n;F*=j)p*=A[P[2*j]*2*n+P[j+++j]];S+=p;}}}float h(A,n)float*A;{int P[j=2*n];S=0;s(A,n,P,j,0);S/=F*(1<<n);}

Try it online!

Explanation (271 bytes version)

float S,p,F;                    // global float variables: total sum, temporary, factorial
j,i;                            // global integer variables: indices
s(A,n,P,l,o,k)float*A;int*P;{   // recursively look at every permutation in S_n
 if(k=j=0,o-l)                  // initialize k and j, check if o != l (possible  permutation not yet fully generated)
  for(;k<l;s(A,n,P,l,o+1))      // loop through possible values for current possible  permuation position
   P[o]=k++;                    // set possible  permutation, recursively call (golfed into the for loop)
 else{for(p=-l;j<l;j++)         // there exists a possible permutation fully generated
  for(i=0;i<l;)                 // test if the possible permutation is a bijection
   p+=P[j]==P[i++];             // check for unique elements
  if(!p){                       // indeed, it is a permutation
   for(F=p=1,j=0;j<n;F*=j)      // Hafnian product loop and calculate the factorial (over and over to save bytes)
    p*=A[P[2*j]*2*n+P[j-~j++]]; // Hafnian product
   S+=p;}}}                     // add to sum
float h(A,n)float*A;{           // Hafnian function
 int P[j=2*n];S=0;              // allocate permutation memory, initialize sum
 s(A,n,P,j,0);                  // calculate Hafnian sum
 S/=F*(1<<n);}                  // calculate Hafnian

Try it online!

At the program's core is the following permutation generator which loops through S_n. All Hafnian computation is simply build upon it -- and further golfed.

j,i,p;Sn(A,l,o,k)int*A;{          // compute every element in S_n
 if(o-l)                          // o!=l, the permutation has not fully been generated
  for(k=0;k<l;k++)                // loop through the integers [0, n)
   A[o]=k,Sn(A,l,o+1);            // modify permutation, call recursively
 else{                            // possible permutation has been generated
  for(p=-l,j=0;j<l;j++)           // look at the entire possible permutation
   for(i=0;i<l;i++)p+=A[j]==A[i]; // check that all elements appear uniquely
  if(!p)                          // no duplicat elements, it is indeed a permutation
   for(printf("["),j=0;j<l        // print
   ||printf("]\n")*0;)            //  the
    printf("%d, ",A[j++]);}}      //   permutation
main(){int l=4,A[l];Sn(A,l,0);}   // all permutations in S_4

Try it online!

\$\endgroup\$
1
  • \$\begingroup\$ 2*j+++1 is equivalent to j+j+++1, which is the same as j-(-j++-1), so we can use the bitwise complement efficiently to save a byte: j-~j++ (Try it online) \$\endgroup\$
    – Mr. Xcoder
    Commented Mar 2, 2018 at 18:51
2
\$\begingroup\$

Jelly, 29 bytes

LHµ2*×!
LŒ!s€2;@€€Wị@/€€P€S÷Ç

Try it online!

I think the ;@€€Wị@/€€P€ part can likely be golfed down. Need to come back later to check and add an explanation.

\$\endgroup\$
4
  • \$\begingroup\$ Identical to my solution (except the J) before golfed down. Jelly minds think alike? source \$\endgroup\$
    – DELETE_ME
    Commented Mar 2, 2018 at 7:17
  • \$\begingroup\$ I was able to reduce it a bit more by refactoring the part that you mentioned as well as the division by 2 and the factorial. LḶŒ!s€2ḅL‘ịFZµPS÷JḤ$P$ TIO \$\endgroup\$
    – miles
    Commented Mar 2, 2018 at 8:59
  • \$\begingroup\$ @user202729 haha nice \$\endgroup\$
    – dylnan
    Commented Mar 2, 2018 at 15:39
  • \$\begingroup\$ @miles Wow that's a lot of savings. I'll edit it into my answer but it's pretty different so feel free to submit your own answer if you want \$\endgroup\$
    – dylnan
    Commented Mar 2, 2018 at 15:40
2
\$\begingroup\$

Haskell, 136 bytes

-14 bytes thanks to ovs.

import Data.List
f m|n<-length m`div`2=sum[product[m!!(s!!(2*j-2)-1)!!(s!!(2*j-1)-1)/realToFrac(2*j)|j<-[1..n]]|s<-permutations[1..n*2]]

Try it online!

Ugh...

\$\endgroup\$
0
2
\$\begingroup\$

MATL, 29 24 22 bytes

Zy:Y@!"G@2eZ{)tn:E/pvs

Try it online! Or verify all test cases: 1, 2, 3.

How it works

Zy       % Size of (implicit) input: pushes [2*n 2*n], where the
         % input is a 2*n × 2*n matrix. 
:        % Range: gives row vector [1 2 ... 2*n]
Y@       % All permutation of that vector as rows of a matrix
!"       % For each permutation 
  G      %   Push input matrix
  @      %   Push current permutation
  2e     %   Reshape as a 2-row array
  Z{     %   Split rows into a cell array of size 2
  )      %   Reference indexing. With a cell array as index this
         %   applies element-wise indexing (similar to sub2ind).
         %   Gives a row vector with the n matrix entries selected
         %   by the current permutation
  t      %   Duplicate
  n:     %   Number of elements, range: this gives [1 2 ... n]
  E      %   Double, element-wise: gives [2 4 ... 2*n]
  /      %   Divide, element-wise
  p      %   Product
  vs     %   Vertically concatenate and sum
         % End (implicit). Display (implicit)
\$\endgroup\$
2
\$\begingroup\$

JavaScript (Node.js), 121 119 117 115 101 bytes

A=>(F=(a,t,r)=>{a--?A[t].map((x,i)=>F[i]=F[i]||F(F[i]=a,i,a%2?r:r*x/(a+2))):s+=r})(A.length,s=0,1)||s

Try it online!

Explanation & Ungolfed

function hafnian(A) {            // Main function taking an array
 return (
  F = function(                  //  Helper function, and also a temporary storage
   a,                            //   how many numbers left
   t,                            //   previous number
   r                             //   the product
  ) {                            //   The braces are to avoid returning value
   a-- ?                         //   If a > 0:
    A[t].map((x, i) =>           //    Loop from i = 2n - 1 downto 0:
     F[i] = F[i] || F(           //     Skip if F[i] is truthy, i.e. i used in the permutation    
      F[i] = a,                  //     Otherwise: Set F[i] to truthy* & start a new recursion
      i,                         //      with i as the previous number
      a % 2                      //      If a - 1 is odd:
       ? r                       //       r does not change
       : r * x / (a + 2)         //      Otherwise: multiply A[t][i] to r, divide by a + 1
     )                           //      Set back F[i] to undefined after the recursion
    )
   : s += r;                     //   Otherwise: add r to s. The factor 1/(n!*2^n) = 1/(2n)!!
  }                              //   has been included in the recursion
 )(
  A.length,                      //  2n = size of the array
  s = 0,                         //  Initial sum s = 0
  1                              //  Initial product r = 1
 ) || s;                         //  Return s
}

/* Note that when a = 1 F[i] will be falsy, but the next recursion will just 
   skip this part anyway. */ 
\$\endgroup\$
1
\$\begingroup\$

Coconut, 165 145 128 127 bytes

-1 byte thanks to Mr. Xcoder

m->sum(reduce((o,j)->o*m[p[2*j]][p[j-~j]]/-~j/2,range(len(m)//2),1)for p in permutations(range(len(m))))
from itertools import*

Try it online!

\$\endgroup\$
0
1
\$\begingroup\$

Perl 6, 86 bytes

{my \n=$^m/2;^$m .permutations.map({[*] .map(->\a,\b{$m[a][b]})}).sum/(2**n*[*] 1..n)}
\$\endgroup\$
1
\$\begingroup\$

JavaScript (Node.js), 87 86 bytes

f=A=>A[0]?A[0].reduce((y,x,i)=>i&&y+x*f((g=B=>B.filter((_,j)=>j&&i-j))(A).map(g)),0):1

Try it online!

This is a port of Kirill's algorithm, and this turns out to be shorter than the naive approach! However I'm retaining my original answer and posting this as a separate one.

Explanation

f = A =>                               // A function receiving an array as input
 A[0]                                  //  If A is not empty:
  ? A[0].reduce((y, x, i) =>           //   Calculate the sum
   i                                   //   for i = 1 to 2n - 1:
   && y + x * f(                       //    A[0][i] * the hafnian of
    (g = B =>
     B.filter((_, j) => j && i - j)    //    array A removing rows 0 and i
    )(A).map(g)                        //    as well as columns 0 and i
   ),
   0
  )
  : 1                                  //  If A is empty: return 1
\$\endgroup\$
1
\$\begingroup\$

Wolfram Language (Mathematica), 84 bytes

aTr[1##&@@(a[[##]]&@@@#~Partition~2)&/@Permutations@Range[n=Tr[1^a]]]/(n/=2)!/2^n

Try it online!


Wolfram Language (Mathematica), 68 bytes

If[#=={},1,Sum[d=Rest@*Delete[i];#[[1,i]]#0@d[d/@#],{i,2,Tr[1^#]}]]&

Try it online!

\$\endgroup\$
1
  • 1
    \$\begingroup\$ Very slightly disappointed Mathematica doesn't have a function for this :) \$\endgroup\$
    – user9206
    Commented Mar 2, 2018 at 8:27
1
\$\begingroup\$

Pari/GP, 51 bytes

h(m)=if(#m,sum(n=2,#m,m[1,n]*h(m[^n,^n][^1,^1])),1)

Try it online!

\$\endgroup\$
1
\$\begingroup\$

Python 3.8 (pre-release), 185 181 bytes

def f(m):
 a=0;n=len(m)//2
 for x in itertools.permutations(range(n*2)):p=1;[p:=p*m[x[2*j]][x[2*j+1]]for j in range(n)];a+=p
 return a/(math.factorial(n)*2**n)
import math,itertools

Try it online!

Golfed reference implementation provided in the question.

\$\endgroup\$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.