Question

假设我有一个简单的函数，它构建了两个正整数（x，y）的所有列表的迭代器，其中＆lt; 1000并且x＆lt; = y

def twoIntsIterator(): Iterator[List[Int]] = {
  for {
    x <- Iterator.range(1, 1000)
    y <- Iterator.range(x, 1000)
  } yield List(x, y)
}

如何实现将列表创建广泛化为可变长度列表的函数intsListIterator(n: Int, limit: Int)？对于n = 2和limit = 1000，这样的函数将产生与上述相同的输出。如果使用n = 3和limit = 4调用它将返回产生以下内容的迭代器：

List(1,1,1)
List(1,1,2)
List(1,1,3)
List(1,2,2)
List(1,2,3)
List(1,3,3)
List(2,2,2)
List(2,2,3)
List(2,3,3)
List(3,3,3)

N.B。：我使用了迭代器，但它们可能是视图，重点是变量列表长度

Answer 1

这是我的解决方案：

scala> def gen(n: Int, limit: Int): Iterator[List[Int]] = n match {
     |   case 0 => Iterator(Nil)
     |   case _ => for(t <- 1 to limit iterator;s <- gen(n-1, t)) yield s:+t
     | }

修改以下方法生成大小为List的所有n且其元素满足start <= x < end，您可以通过

定义intsListIterator
def intsListIterator(n: Int, limit: Int) = gen(n, 1, limit) scala> def gen(n: Int, start: Int, end: Int): Iterator[List[Int]] = n match { | case 0 => Iterator(Nil) | case _ => for(i <- Iterator.range(start, end);s <- gen(n-1,i,end)) yield i::s | } gen: (n: Int, start: Int, end: Int)Iterator[List[Int]] scala> gen(3, 1, 4) foreach println List(1, 1, 1) List(1, 1, 2) List(1, 1, 3) List(1, 2, 2) List(1, 2, 3) List(1, 3, 3) List(2, 2, 2) List(2, 2, 3) List(2, 3, 3) List(3, 3, 3) scala> gen(7, -3, 4) take 10 foreach println List(-3, -3, -3, -3, -3, -3, -3) List(-3, -3, -3, -3, -3, -3, -2) List(-3, -3, -3, -3, -3, -3, -1) List(-3, -3, -3, -3, -3, -3, 0) List(-3, -3, -3, -3, -3, -3, 1) List(-3, -3, -3, -3, -3, -3, 2) List(-3, -3, -3, -3, -3, -3, 3) List(-3, -3, -3, -3, -3, -2, -2) List(-3, -3, -3, -3, -3, -2, -1) List(-3, -3, -3, -3, -3, -2, 0)

Answer 2

只需使用递归：

def produce(n: Int, limit: Int, k: Int = 1): Iterator[List[Int]] = {
  Iterator.range(k, limit) flatMap {
    case x if n > 1 => produce(n - 1, limit, x).map(x :: _)
    case x => Iterator(List(x))
  }
}

或者为了理解：

def produce(n: Int, limit: Int, k: Int = 1): Iterator[List[Int]] = for {
   x <- k to limit - 1 iterator;
   y <- if (n > 1) produce(n - 1, limit, x) else Iterator(Nil)
} yield x :: y

Answer 3

这很有效：

def intsIterator(n: Int, limit: Int) = (1 to n).map(List.fill(limit)(_)).flatten.combinations(limit).filter(l => (l, l.tail).zipped.forall(_ <= _))

scala> intsIterator(5,3) mkString "\n"
res16: String =
Vector(1, 2, 3)
Vector(1, 2, 4)
Vector(1, 2, 5)
Vector(1, 3, 4)
Vector(1, 3, 5)
Vector(1, 4, 5)
Vector(2, 3, 4)
Vector(2, 3, 5)
Vector(2, 4, 5)
Vector(3, 4, 5)

基本上你得到一个组合，即n C limit，然后根据列表是否排序进行过滤。

或者更易阅读的版本：

def intsIterator(n: Int, limit: Int) = (1 to n).map(List.fill(limit)(_)).flatten.combinations(limit).filter(l => l.sorted == l)

Answer 4

如果效率或可扩展性在Vector s上有效，我就不会使用递归并创建Iterator而不是List

new Iterator() {
  val max = limit - 1 // makes logic simpler
  var cur = Vector.fill(n - 1)(1) :+ 0
  var (i, v) = (n - 1, 1)

  def hasNext(): Boolean = cur.head != max

  def next(): List[Int] = {
    if (v <= max) 
      cur = cur.updated(i, v)
    else {
      i -= 1
      if (cur(i) == max - 1) 
        cur = cur.update(i, max)
      else {
        v = cur(i) + 1
        cur = cur.take(i) ++ Vector.fill(n - i)(v)
        i = n - 1
      }
    }
    v += 1
    cur.toList // you could leave as a Vector
  }
}

当然，您可以随时将其转换为List toList

（未经测试;用手机写过）

创建一个scala函数，生成长度为N的整数的有序列表

4 个答案: