pre_testing1/collatz.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Sat, 14 Nov 2020 00:40:47 +0000
changeset 360 1bd1e9cd8c72
parent 346 ed1b6b8c36e5
child 362 fc9394f4f0ea
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
282
2d290b79fc73 updated
Christian Urban <urbanc@in.tum.de>
parents: 266
diff changeset
     1
// Basic Part about the 3n+1 conjecture
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     2
//==================================
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     3
266
31e5218f43de updated to 2.13
Christian Urban <urbanc@in.tum.de>
parents: 208
diff changeset
     4
// generate jar with
31e5218f43de updated to 2.13
Christian Urban <urbanc@in.tum.de>
parents: 208
diff changeset
     5
//   > scala -d collatz.jar  collatz.scala
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     6
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     7
object CW6a { // for purposes of generating a jar
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     8
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     9
def collatz(n: Long): Long =
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    10
  if (n == 1) 0 else
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    11
    if (n % 2 == 0) 1 + collatz(n / 2) else 
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    12
      1 + collatz(3 * n + 1)
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    13
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    14
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    15
def collatz_max(bnd: Long): (Long, Long) = {
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    16
  val all = for (i <- (1L to bnd)) yield (collatz(i), i)
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    17
  all.maxBy(_._1)
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    18
}
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    19
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 282
diff changeset
    20
//collatz_max(1000000)
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 282
diff changeset
    21
//collatz_max(10000000)
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 282
diff changeset
    22
//collatz_max(100000000)
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    23
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    24
/* some test cases
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    25
val bnds = List(10, 100, 1000, 10000, 100000, 1000000)
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    26
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    27
for (bnd <- bnds) {
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    28
  val (steps, max) = collatz_max(bnd)
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    29
  println(s"In the range of 1 - ${bnd} the number ${max} needs the maximum steps of ${steps}")
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    30
}
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    31
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    32
*/
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    33
335
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    34
def is_pow(n: Long) : Boolean = (n & (n - 1)) == 0
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    35
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    36
def is_hard(n: Long) : Boolean = is_pow(3 * n + 1)
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    37
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    38
def last_odd(n: Long) : Long = 
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    39
  if (is_hard(n)) n else
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    40
    if (n % 2 == 0) last_odd(n / 2) else 
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    41
      last_odd(3 * n + 1)
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    42
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    43
336
cccdae0fccc7 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 335
diff changeset
    44
//for (i <- 130 to 10000) println(s"$i: ${last_odd(i)}")
360
1bd1e9cd8c72 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 346
diff changeset
    45
//for (i <- 1 to 100) println(s"$i: ${collatz(i)}")
335
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    46
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    47
}
335
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    48
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    49
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    50