更快的置换发生器

时间:2011-01-04 17:54:47

标签: list optimization scala functional-programming permutation

我为Scala列表编写了一个置换生成器,用于生成给定列表的所有排列。到目前为止,我已经基于this Haskell implementation获得了以下内容(我认为它比我尝试的其他几个选项更有效)。有没有办法让这个更有效率,或者我是否涵盖了我的所有基础?

   /** For each element x in List xss, returns (x, xss - x) */
   def selections[A](xss:List[A]):List[(A,List[A])] = xss match {
      case Nil => Nil
      case x :: xs =>
         (x, xs) :: (for( (y, ys) <- selections (xs) )
            yield (y, x :: ys))
   }

   /** Returns a list containing all permutations of the input list */
   def permute[A](xs:List[A]):List[List[A]] = xs match {
      case Nil => List(Nil)

      //special case lists of length 1 and 2 for better performance
      case t :: Nil => List(xs)
      case t :: u :: Nil => List(xs,List(u,t))

      case _ => 
         for ( (y,ys) <- selections(xs); ps <- permute(ys))
            yield y :: ps
   }

1 个答案:

答案 0 :(得分:3)

在Scala 2.9中,extempore为scala集合类添加了一些有用的方法,包括一个Seq.permutations,它生成了这个seq的所有排列。见link text。我有一个非递归实现,我认为会有更好的性能。见A non-recursive implementation of SeqLike.permutations