pre_testing1/collatz.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Tue, 17 Nov 2020 00:34:55 +0000
changeset 362 fc9394f4f0ea
parent 360 1bd1e9cd8c72
child 363 9f481fd7c613
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
     1
object CW6a {
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     2
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
     3
//(1) Complete the collatz function below. It should
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
     4
//    recursively calculate the number of steps needed
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
     5
//    until the collatz series reaches the number 1.
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
     6
//    If needed, you can use an auxiliary function that
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
     7
//    performs the recursion. The function should expect
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
     8
//    arguments in the range of 1 to 1 Million.
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     9
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    10
def collatz(n: Long) : Long =
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    11
    if ( n == 1) 1;
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    12
    else if (n % 2 == 0) 1 + collatz( n / 2);
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    13
    else 1 + collatz( n * 3 + 1);
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    14
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    15
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    16
//(2) Complete the collatz_max function below. It should
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    17
//    calculate how many steps are needed for each number
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    18
//    from 1 up to a bound and then calculate the maximum number of
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    19
//    steps and the corresponding number that needs that many
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    20
//    steps. Again, you should expect bounds in the range of 1
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    21
//    up to 1 Million. The first component of the pair is
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    22
//    the maximum number of steps and the second is the
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    23
//    corresponding number.
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    24
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    25
def collatz_max(bnd: Long) : (Long, Long) =
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    26
     ((1.toLong to bnd).toList.map
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    27
        (n => collatz(n)).max ,
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    28
            (1.toLong to bnd).toList.map
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    29
                (n => collatz(n)).indexOf((1.toLong to bnd).toList.map
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    30
                    (n => collatz(n)).max) + 1);
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    31
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    32
//(3) Implement a function that calculates the last_odd
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    33
//    number in a collatz series.  For this implement an
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    34
//    is_pow_of_two function which tests whether a number
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    35
//    is a power of two. The function is_hard calculates
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    36
//    whether 3n + 1 is a power of two. Again you can
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    37
//    assume the input ranges between 1 and 1 Million,
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    38
//    and also assume that the input of last_odd will not
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    39
//    be a power of 2.
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    40
//idk
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    41
 def is_pow_of_two(n: Long) : Boolean =
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    42
    if ( n & ( n - 1) == 0) true;
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    43
    else false;
208
ab0ccbc8271f updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    44
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    45
def is_hard(n: Long) : Boolean =
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    46
    if ( (3*n + 1) & 3*n == 0) true;
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    47
    else false;
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
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    50
def last_odd(n: Long) : Long = ???
335
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    51
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    52
9476aee44eed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 320
diff changeset
    53
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 360
diff changeset
    54
}