используя scala Найти четыре элемента из списка, которые суммируют с заданным значением - PullRequest
Новая
       8

используя scala Найти четыре элемента из списка, которые суммируют с заданным значением

0 голосов
/ 10 сентября 2018

Я новичок в языке программирования Scala и хочу реализовать код, представленный ниже scenerio. Для данного примера выборки из n целых и целых трех выборок в элементе выборки есть элементы a, b, c и d, так что a + b + c + d = выборка трех. Найти все уникальные четверки в списке, который дает сумму выборки три

Пример: 

sampleone =[1,0,-1,0,-2,2] and samplethree = 0 
a solution set is 
[-1,0,0,1]
[-2,-1,1,2]
[-2,0,0,2]

код, который я использовал: 

scala> def findFourElements(A: List[Int], n: Int, x: Int) = {
     | {
     | for(a <- 0 to A.length-3)
     | {
     | for(b <- a+1 to A.length-2)
     | {
     | for(c <- b+1 to A.length-1)
     | {
     |  for(d <- c+1 to A.length)
     | {
     | if(A(a) + A(b) + A(c) + A(d) == x)
     | {
     | print(A(a)+A(b)+A(c)+A(d))
     | }}}}}}
     | }
findFourElements: (A: List[Int], n: Int, x: Int)Unit


scala> val sampleone = List(1,0,-1,0,-2,2)
sampleone: List[Int] = List(1, 0, -1, 0, -2, 2)

scala> val sampletwo = sampleone.length
sampletwo: Int = 6

scala> val samplethree = 0
samplethree: Int = 0

scala> findFourElements(sampleone,sampletwo,samplethree)
0java.lang.IndexOutOfBoundsException: 6
  at scala.collection.LinearSeqOptimized$class.apply(LinearSeqOptimized.scala:65)
  at scala.collection.immutable.List.apply(List.scala:84)
  at $anonfun$findFourElements$1$$anonfun$apply$mcVI$sp$1$$anonfun$apply$mcVI$sp$2$$anonfun$apply$mcVI$sp$3.apply$mcVI$sp(<console>:33)
  at scala.collection.immutable.Range.foreach$mVc$sp(Range.scala:160)
  at $anonfun$findFourElements$1$$anonfun$apply$mcVI$sp$1$$anonfun$apply$mcVI$sp$2.apply$mcVI$sp(<console>:31)
  at scala.collection.immutable.Range.foreach$mVc$sp(Range.scala:160)
  at $anonfun$findFourElements$1$$anonfun$apply$mcVI$sp$1.apply$mcVI$sp(<console>:29)
  at scala.collection.immutable.Range.foreach$mVc$sp(Range.scala:160)
  at $anonfun$findFourElements$1.apply$mcVI$sp(<console>:27)
  at scala.collection.immutable.Range.foreach$mVc$sp(Range.scala:160)
  at findFourElements(<console>:25)
  ... 48 elided

Но я получаю ошибку индекса из-за привязанного исключения. Также есть способ иметь более оптимизированный код в Scala.

Спасибо за помощь.

1 Ответ

0 голосов
/ 10 сентября 2018

Это может делать то, что вы хотите: 

sampleone.combinations(4).filter(_.sum == samplethree)

Метод combinations дает итератор, который доставляет каждую возможную комбинацию значений по очереди. Если существует несколько способов создать одну и ту же последовательность, будет возвращен только один.

Вызов filter удаляет все последовательности, которые не суммируются со значением samplethree.

...