Best way to handle while loop style program in Scala

Question

I have a simple program written in C++. It generates random numbers, and stop when the sum of those numbers equal or greater than 100. The code looks like:

vector<int> container;
while(container.sum() <100)
{
    int new_number = rand()%10 + 1 ;// generate a number in range 1 to 10
    container.push_back(new_number); // add new number to the container

}

What is the best way to handle the task in Scala? (without using while loop?)

It seems that FoldLeft or FoldRight function doesn't have ability to break at a condition?

what do you want to do with those numbers ? do you need to store those numbers in memory ? — Mzf
– Mzf, Commented Dec 21, 2017 at 13:16

jwvh · Accepted Answer · 2017-12-21 05:11:07Z

3

Create an infinite Stream of random numbers (requires very little CPU and memory), take only what you need, then turn the result Stream into the desired collection type.

val randoms = Stream.continually(util.Random.nextInt(10)+1)
val container = randoms.take(randoms.scan(0)(_+_).indexWhere(_>=100)).toVector

Added bonus is that the sums are calculated as you go, i.e. added to the previous sum, not summing from the beginning each time.

edited Dec 21, 2017 at 5:11

answered Dec 21, 2017 at 4:55

jwvh

51.3k7 gold badges42 silver badges70 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Leo C · Accepted Answer · 2017-12-21 02:24:33Z

2

Here's one approach to handle a while loop using a tail-recursive function as follows:

@scala.annotation.tailrec
def addToContainer(container: Vector[Int], max: Int): Vector[Int] = {
  val newContainer = container ++ Vector(scala.util.Random.nextInt(10) + 1)

  if (newContainer.sum >= max) container
  else addToContainer(newContainer, max)
}

addToContainer(Vector[Int](), 100)
// res1: Vector[Int] = Vector(9, 9, 5, 9, 3, 5, 2, 5, 10, 7, 6, 4, 5, 5, 9, 3)

res1.sum
// res2: Int = 96

answered Dec 21, 2017 at 2:24

Leo C

22.5k3 gold badges28 silver badges42 bronze badges

1 Comment

Levi Ramsey Over a year ago

In general, it should be noted that while loops and tail recursive functions are exactly equivalent. So if you've got a while loop in another language (even pseudocode), you can translate it into scala as a tail recursive function (and Scala will compile it as if you'd written a while loop).

abhi · Accepted Answer · 2017-12-21 01:53:44Z

Here's one way to do it:

  val randomNumberGenerator = new scala.util.Random
  def sumUntil(list: List[Int]): List[Int] = list match {
    case exceeds if list.filter(_ > 0).sum > 100 => list
    case _ => sumUntil(list :+ (randomNumberGenerator.nextInt(10) + 1))
  }

To explain the solution:

Create an instance of scala.util.Random which will help us generate random numbers
sumUntil will pattern match; if the sum of the list exceeds 100, return it.
In the event that the sum does not exceed 100, call sumUntil again, but with another random number generated between 10 and 1 (inclusive). Keep in mind that the _ means, "I don't care about the value, or even the type." _ is anything else but the case where the sum of all the integers in our list is greater than 100.

If you're new to Scala, I understand that it may be a bit rough on the eyes to read. Below is a refined version:

  val randomNumberGenerator = new scala.util.Random

  def sumUntil(list: List[Int]): List[Int] = list match {
    case exceeds if sumList(list) > 100 => list
    case _ => sumUntil(appendRandomNumberToList(list))
  }

  private def sumList(list: List[Int]): Int = {
    list.filter(_ > 0).sum
  }

  private def appendRandomNumberToList(list: List[Int]): List[Int] = {
    list :+ randomNumberGenerator.nextInt(10) + 1
  }

Duong Nguyen · Accepted Answer · 2017-12-21 05:19:12Z

0

If your loop just scans through the collection, use fold or reduce.
If it needs some custom terminate condition, recursion is favored.

answered Dec 21, 2017 at 5:19

Duong Nguyen

8506 silver badges11 bronze badges

Collectives™ on Stack Overflow

Best way to handle while loop style program in Scala

4 Answers 4

Comments

1 Comment

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related