Составление списка Scala

Я пытаюсь создать список в 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)

Мы будем очень благодарны за любые предложения.

-Ли

6
задан LeeG 26 January 2012 в 05:59
поделиться