2

我正在学习 Scala,我想将一些旧算法代码转换为 Scala。

我有非常简单的 Java 代码,它打印 superSet(所有可能的集合的组合,包括空集和集合本身)。

public static Set<Set<Integer>> createSuperSet(Set<Integer> originalSet){
    Set<Set<Integer>> superSet = new HashSet<Set<Integer>>();

    if (originalSet.size() == 0){
        Set<Integer> empty = new HashSet<Integer>();
        superSet.add(empty);
        return superSet;
    }

    List<Integer> list = new ArrayList<Integer>(originalSet);
    Integer head = list.get(0);
    Set<Integer> rest = new HashSet<Integer>(list.subList(1, list.size()));
    for (Set<Integer> set  : createSuperSet(rest)){
        Set<Integer> newSet = new HashSet<Integer>();
        newSet.add(head);
        newSet.addAll(set);
        superSet.add(newSet);
        superSet.add(set);
    }       
    return superSet;
}

现在我正在尝试在 Scala 中实现相同的功能:

  def createSuperSet(originalSet: Set[Int]): Set[Set[Int]] ={
     val superSet = Set[Set[Int]]()

     originalSet.toList match {   

       case List() => {superSet + Set[Int]()} 

       case head::restAsList => {
         val rest = restAsList.toSet[Int]
         val result = createSuperSet(rest)
         result.foreach(f=>{
           val newSet:Set[Int] =  f + head
           superSet + f +newSet
         }) 
         superSet
       }
     }   
  }

但不幸的是,此代码返回空集。我怀疑这个问题是由于不可变集合的使用而发生的。我试图在调试器中运行它,我看到对函数的递归调用总是返回空集,我的代码永远不会进入 foreach 函数。

请帮忙。欢迎任何想法。

4

2 回答 2

4

在惯用的 scala 中,应用于不可变集合的+(and -, , etc) 运算符创建一个新集合 - 否则它们将不是不可变的。++相反,您必须将修改与另一块语法糖结合起来,如果您附加=到运算符,它会将运算符的结果分配给左侧变量:superSet += f + newSet

于 2013-09-08T22:48:53.393 回答
0

我这样解决了这个问题:

 def createSuperSet(originalSet: Set[Int]): Set[Set[Int]] ={
     var superSet = Set[Set[Int]]()

     originalSet.toList match {   

       case List() => {superSet + Set[Int]()} 

       case head::restAsList => {
         val rest = restAsList.toSet[Int]
         val result = createSuperSet(rest)
         result.map(f=>{
           superSet = superSet  + f+ (f+head)
         }) 
         superSet
       }
     }   
  }

跑步 println(createSuperSet(Set[Int](1,2,3))

印刷

Set(Set(), Set(3, 1), Set(2), Set(2, 1), Set(3, 2), Set(3), Set(3, 2, 1), Set(1))

但我很高兴知道是否有更优雅的解决方案

于 2013-09-09T15:11:32.307 回答