core_testing1/collatz.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Sat, 11 Mar 2023 22:01:53 +0000
changeset 460 f5c0749858fd
parent 430 4029552de5fc
child 469 ed7a9ea24f71
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
460
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
     1
import scala.annotation.tailrec
401
317e49b9cab6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 399
diff changeset
     2
// Core Part 1 about the 3n+1 conjecture
430
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
     3
//============================================
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
     4
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
     5
object C1 {
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
     6
460
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
     7
@tailrec
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
     8
private def collatz(n: Long, steps: Long = 0): Long = {
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
     9
  if (n == 1) steps
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    10
  else if (n % 2 == 0) collatz(n / 2, steps + 1)
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    11
  else collatz(n * 3 + 1, steps + 1)
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    12
}
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    13
460
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    14
def collatz_max(upper: Long): (Long, Long) = {
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    15
  (1L to upper).map(n => (collatz(n), n)).maxBy(_._1)
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
    16
}
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    17
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    18
460
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    19
private def is_pow_of_two(n: Long) : Boolean = {
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    20
  (n & (n - 1)) == 0
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    21
}
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
    22
460
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    23
private def is_hard(n: Long) : Boolean = {
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    24
  is_pow_of_two(3 * n + 1)
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    25
}
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
    26
460
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    27
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    28
private def last_odd(n: Long): Long = {
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    29
  (1L to n).filter(is_hard).max
f5c0749858fd updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 430
diff changeset
    30
}
335
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    31
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
    32
}
335
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    33
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    34
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    35
430
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
    36
// This template code is subject to copyright 
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
    37
// by King's College London, 2022. Do not 
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
    38
// make the template code public in any shape 
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
    39
// or form, and do not exchange it with other 
4029552de5fc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 401
diff changeset
    40
// students under any circumstance.