scala - Scala列表理解

Question

我正在尝试根据公式在scala中生成一个列表:

for n > 1 f(n) = 4*n^2 - 6*n + 6 and for n == 1 f(n) = 1

目前我有:

def lGen(end: Int): List[Int] = {
    for { n <- List.range(3 , end + 1 , 2) } yields { 4*n*n - 6*n - 6 }
}

对于end = 5，将给出以下列表:

List(24 , 76)

现在，我一直在努力寻找一种使该函数具有宽容性的方法

List(1 , 24 , 74)

任何建议将不胜感激。

-李

Answer 1

这个怎么样:

scala> def lGen(end: Int): List[Int] =
         1 :: List.range(3, end+1, 2).map(n => 4*n*n - 6*n + 6)

scala> lGen(5)
res0: List[Int] = List(1, 24, 76)