progs/lecture2.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Thu, 27 Nov 2025 18:24:07 +0000
changeset 504 7653dd662db3
parent 491 2a30c7dfe3ed
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
51
0e60e6c24b99 updated
Christian Urban <urbanc@in.tum.de>
parents: 39
diff changeset
     1
// Scala Lecture 2
0e60e6c24b99 updated
Christian Urban <urbanc@in.tum.de>
parents: 39
diff changeset
     2
//=================
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
     3
 
468
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
     4
// - Options
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
     5
// - Higher-Order Functions (short-hand notation)
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
     6
// - maps (behind for-comprehensions)
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
     7
// - Pattern-Matching
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
     8
// - Recursion
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
     9
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    10
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    11
// Function Definitions
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    12
//======================
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    13
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    14
// The general scheme for a function: you have to give a 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    15
// type to each argument and a return type of the function
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    16
//
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    17
//  def fname(arg1: ty1, arg2: ty2,..., argn: tyn): rty = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    18
//     ....
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    19
//  }
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    20
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    21
def incr(x: Int) : Int = x + 1
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    22
incr(42)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    23
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    24
def add(x: Int, y: Int) : Int = x + y
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    25
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    26
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    27
def average(ls: List[Int]) : Option[Int] = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    28
  val s = ls.sum
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    29
  val l = ls.length
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    30
  if (l == 0) None else Some(s / l)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    31
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    32
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    33
average(List(1,2,3,4))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    34
average(List())
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    35
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    36
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    37
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    38
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    39
// The Option Type
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    40
//=================
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    41
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    42
// in Java, if something unusually happens, you return null 
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    43
// or raise an exception
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    44
//
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    45
//in Scala you use Options instead
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    46
//   - if the value is present, you use Some(value)
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    47
//   - if no value is present, you use None
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
    48
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
    49
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    50
List(7,2,3,4,5,6).find(_ < 4)
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    51
List(5,6,7,8,9).find(_ < 4)
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
    52
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    53
// Int:      ..., 0, 1, 2,...
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    54
// Boolean:  true false
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    55
//
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    56
// List[Int]: Nil, List(_) 
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    57
//
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    58
// Option[Int]: None, Some(0), Some(1), ...
444
e6df3c7ff132 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 384
diff changeset
    59
// Option[Boolean]: None, Some(true), Some(false)
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    60
// Option[...]: None, Some(_)
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    61
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    62
def div(x: Int, y: Int) : Int = 
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    63
  x / y
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    64
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    65
def safe_div(x: Int, y: Int) : Option[Int] = 
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    66
  if (y == 0) None else Some(x / y)
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    67
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    68
def foo_calculation(x: Int) = 
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    69
  if (safe_div(x, x) == None) 24 else safe_div(x, x).get
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    70
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    71
foo_calculation(3)
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    72
foo_calculation(0)
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    73
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
    74
safe_div(10 + 5, 3)  
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    75
444
e6df3c7ff132 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 384
diff changeset
    76
List(1,2,3,4,5,6).indexOf(7)
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    77
List[Int]().min
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    78
List[Int](1,2,3).minOption
444
e6df3c7ff132 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 384
diff changeset
    79
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
    80
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
    81
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    82
// better error handling with Options (no exceptions)
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    83
//
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    84
//  Try(something).getOrElse(what_to_do_in_case_of_an_exception)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
    85
//
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    86
468
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
    87
import scala.util._      // Try,...
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
    88
import io.Source         // fromURL
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    89
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    90
//val my_url = "https://nms.kcl.ac.uk/christian.urban/"
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    91
val my_url = "https://urbanchr.github.io/"
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    92
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    93
println(Try(Source.fromURL(my_url)(using "ISO-8859-1").mkString).toOption)
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    94
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    95
.mkString  
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    96
Source.fromURL(my_url)(using "ISO-8859-1").getLines().toList
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    97
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
    98
Try(Source.fromURL(my_url)(using "ISO-8859-1").mkString).getOrElse("")
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
    99
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   100
Try(Some(Source.fromURL(my_url)(using "ISO-8859-1").mkString)).getOrElse(None)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   101
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   102
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   103
// the same for files
444
e6df3c7ff132 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 384
diff changeset
   104
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   105
Try(Some(Source.fromFile("test.txt")(using "ISO-8859-1").mkString)).getOrElse(None)
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   106
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   107
Try(Source.fromFile("test.txt")(using "ISO-8859-1").mkString).toOption
444
e6df3c7ff132 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 384
diff changeset
   108
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   109
Using(Source.fromFile("test.txt")(using "ISO-8859-1"))(_.mkString).toOption
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   110
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   111
// how to implement a function for reading 
444
e6df3c7ff132 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 384
diff changeset
   112
// (lines) from files...
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   113
//
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   114
def get_contents(name: String) : List[String] = 
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   115
  Source.fromFile(name)(using "ISO-8859-1").getLines().toList
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   116
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   117
get_contents("text.txt")
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   118
get_contents("test.txt")
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   119
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   120
// slightly better - return Nil
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   121
def get_contents(name: String) : List[String] = 
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   122
  Try(Source.fromFile(name)(using "ISO-8859-1").getLines.toList).getOrElse(List())
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   123
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   124
get_contents("text.txt")
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   125
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   126
// much better - you record in the type that things can go wrong 
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   127
def get_contents(name: String) : Option[List[String]] = 
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   128
  Try(Some(Source.fromFile(name)(using "ISO-8859-1").getLines().toList)).getOrElse(None)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   129
316
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   130
get_contents("text.txt")
03d55eb6a0b7 updated
Christian Urban <urbanc@in.tum.de>
parents: 310
diff changeset
   131
get_contents("test.txt")
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   132
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   133
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   134
// operations on options
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   135
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   136
val lst = List(None, Some(1), Some(2), None, Some(3))
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   137
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   138
for (x <- lst) yield {
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   139
  if (x == None) None
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   140
  else Some(x.get + 1)
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   141
}
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   142
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   143
lst.flatten
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   144
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   145
Some(1).get
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   146
None.get
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   147
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   148
Some(1).isDefined
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   149
None.isDefined
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   150
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   151
for (x <- lst) yield x.getOrElse(0)
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   152
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   153
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   154
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   155
val ps = List((3, 0), (4, 2), (6, 2), 
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   156
              (2, 0), (1, 0), (1, 1))
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   157
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   158
// division where possible
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   159
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   160
for ((x, y) <- ps) yield {
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   161
  if (y == 0) None else Some(x / y)
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   162
}
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   163
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   164
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   165
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   166
// getOrElse is for setting a default value
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   167
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   168
val lst = List(None, Some(1), Some(2), None, Some(3))
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   169
361
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   170
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   171
// a function that turns strings into numbers 
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   172
// (similar to .toInt)
736a60afedbc updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 334
diff changeset
   173
Integer.parseInt("1234")
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   174
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   175
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   176
def get_me_an_int(s: String) : Option[Int] = 
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   177
 Try(Some(Integer.parseInt(s))).getOrElse(None)
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   178
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   179
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   180
// This may not look any better than working with null in Java, but to
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   181
// see the value, you have to put yourself in the shoes of the
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   182
// consumer of the get_me_an_int function, and imagine you didn't
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   183
// write that function.
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   184
//
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   185
// In Java, if you didn't write this function, you'd have to depend on
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   186
// the Javadoc of the get_me_an_int. If you didn't look at the Javadoc, 
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   187
// you might not know that get_me_an_int could return null, and your 
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   188
// code could potentially throw a NullPointerException.
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   189
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   190
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   191
// even Scala is not immune to problems like this:
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   192
317
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   193
List(5,6,7,8,9).indexOf(7)
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   194
List(5,6,7,8,9).indexOf(10)
07583fbe4f95 updated
Christian Urban <urbanc@in.tum.de>
parents: 316
diff changeset
   195
List(5,6,7,8,9)(-1)
310
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   196
996279af8952 updated
Christian Urban <urbanc@in.tum.de>
parents: 309
diff changeset
   197
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   198
Try({
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   199
  val x = 3
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   200
  val y = 0
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   201
  Some(x / y)
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   202
}).getOrElse(None)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   203
323
93b6c16dded8 updated
Christian Urban <urbanc@in.tum.de>
parents: 320
diff changeset
   204
93b6c16dded8 updated
Christian Urban <urbanc@in.tum.de>
parents: 320
diff changeset
   205
// minOption 
93b6c16dded8 updated
Christian Urban <urbanc@in.tum.de>
parents: 320
diff changeset
   206
// maxOption 
93b6c16dded8 updated
Christian Urban <urbanc@in.tum.de>
parents: 320
diff changeset
   207
// minByOption 
93b6c16dded8 updated
Christian Urban <urbanc@in.tum.de>
parents: 320
diff changeset
   208
// maxByOption
93b6c16dded8 updated
Christian Urban <urbanc@in.tum.de>
parents: 320
diff changeset
   209
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   210
def altproduct(xs: List[Int]) : List[Int] = xs match {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   211
  case Nil => Nil
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   212
  case (x::y::xs) => x * y :: altproduct(y::xs)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   213
  case (x::Nil) => List(x)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   214
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   215
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   216
altproduct(List(1,2,3,4,5))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   217
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   218
def powerset(xs: Set[Int]) : Set[Set[Int]] = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   219
  if (xs == Set()) Set(Set())
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   220
  else {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   221
    powerset(xs.tail) ++ powerset(xs.tail).map(_ + xs.head)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   222
  }
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   223
}     
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   224
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   225
powerset(Set(1,2,3)).mkString("\n")
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   226
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   227
// Higher-Order Functions
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   228
//========================
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   229
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   230
// functions can take functions as arguments
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   231
// and produce functions as result
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   232
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   233
val foo = for (n <- (1 to 10).toList) yield n * n
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   234
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   235
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   236
def even(x: Int) : Boolean = x % 2 == 0
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   237
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   238
even(2)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   239
even(3)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   240
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   241
def even(x: Int) : Boolean = x % 2 == 1
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   242
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   243
val foo_fun = even
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   244
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   245
foo_fun(2)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   246
foo_fun(3)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   247
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   248
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   249
def odd(x: Int) : Boolean = x % 2 == 1
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   250
478
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   251
def inc(x: Int) : Int = x + 1
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   252
val lst = (1 to 10).toList
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   253
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   254
lst.filter(odd)
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   255
lst.find(even)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   256
lst.find(_ % 2 == 0)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   257
lst.sortWith(_ < _)  // (x,y) => x < y
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   258
lst.find(_ > 4)
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   259
lst.count(odd)
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   260
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   261
lst.filter(_ % 2 == 0)
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   262
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   263
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   264
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   265
lst.find(_ < 4)
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   266
lst.filter(_ < 4) 
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   267
478
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   268
val x = 3 < 4
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   269
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   270
def less4(x: Int) : Boolean = x < 4
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   271
lst.find(less4)
478
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   272
lst.find(x => !(x < 4))
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   273
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   274
lst.filter(x => x % 2 == 0)
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   275
lst.filter(_ % 2 == 0)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   276
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   277
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   278
lst.sortWith((x, y) => x > y)
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   279
lst.sortWith(_ < _)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   280
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   281
// but this only works when the arguments are clear, but 
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   282
// not with multiple occurences
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   283
lst.find(n => odd(n) && n > 2)
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   284
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   285
444
e6df3c7ff132 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 384
diff changeset
   286
// lexicographic ordering
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   287
val ps = List((3, 0), (3, 2), (4, 2), (2, 2), 
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   288
              (2, 0), (1, 1), (1, 0))
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   289
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   290
def lex(x: (Int, Int), y: (Int, Int)) : Boolean = 
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   291
  if (x._1 == y._1) x._2 < y._2 else x._1 < y._1
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   292
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   293
ps.sortWith(lex)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   294
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   295
ps.sortBy(x => x._1)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   296
ps.sortBy(_._2)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   297
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   298
ps.maxBy(x => x._2)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   299
ps.maxBy(_._2)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   300
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   301
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   302
// maps (lower-case)
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   303
//===================
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   304
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   305
def double(x: Int): Int = x + x
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   306
def square(x: Int): Int = x * x
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   307
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   308
val lst = (1 to 10).toList
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   309
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   310
lst.map(square)
491
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   311
lst.map(n => square(n))
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   312
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   313
for (n <- lst) yield square(n)
2a30c7dfe3ed updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 478
diff changeset
   314
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   315
lst.map(x => (double(x), square(x)))
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   316
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   317
// works also for strings
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   318
def tweet(c: Char) = c.toUpper
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   319
478
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   320
"Hello\nWorld".map(tweet)
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   321
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   322
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   323
// this can be iterated
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   324
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   325
lst.map(square).filter(_ > 4)
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   326
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   327
lst.map(square).find(_ > 4)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   328
lst.map(square).find(_ > 4).map(double)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   329
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   330
lst.map(square)
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   331
   .find(_ > 4)
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   332
   .map(double)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   333
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   334
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   335
// Option Type and maps
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   336
//======================
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   337
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   338
// a function that turns strings into numbers (similar to .toInt)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   339
Integer.parseInt("12u34")
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   340
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   341
// maps on Options
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   342
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   343
import scala.util._
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   344
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   345
def get_me_an_int(s: String) : Option[Int] = 
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   346
 Try(Some(Integer.parseInt(s))).getOrElse(None)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   347
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   348
get_me_an_int("12345").map(_ % 2 == 0)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   349
get_me_an_int("12u34").map(_ % 2 == 0)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   350
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   351
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   352
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   353
val lst = List("12345", "foo", "5432", "bar", "x21", "456")
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   354
for (x <- lst) yield get_me_an_int(x)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   355
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   356
// summing up all the numbers
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   357
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   358
lst.map(get_me_an_int).flatten.sum
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   359
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   360
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   361
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   362
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   363
// this is actually how for-comprehensions are
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   364
// defined in Scala
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   365
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   366
lst.map(n => square(n))
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   367
for (n <- lst) yield square(n)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   368
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   369
// lets define our own higher-order functions
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   370
// type of functions is for example Int => Int
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   371
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   372
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   373
def my_map_int(lst: List[Int], f: Int => Int) : List[Int] = 
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   374
{
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   375
  if (lst == Nil) Nil
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   376
  else f(lst.head) :: my_map_int(lst.tail, f)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   377
}
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   378
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   379
my_map_int(lst, square)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   380
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   381
// same function using pattern matching: a kind
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   382
// of switch statement on steroids (see more later on)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   383
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   384
def my_map_int(lst: List[Int], f: Int => Int) : List[Int] = 
362
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   385
  lst match {
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   386
    case Nil => Nil
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   387
    case x::xs => f(x)::my_map_int(xs, f)
fc9394f4f0ea updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 361
diff changeset
   388
  }
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   389
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   390
363
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   391
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   392
val biglst = (1 to 10000).toList
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   393
my_map_int(biglst, double)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   394
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   395
(1 to 10000000).toList.map(double)
9f481fd7c613 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 362
diff changeset
   396
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   397
// other function types
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   398
//
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   399
// f1: (Int, Int) => Int
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   400
// f2: List[String] => Option[Int]
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   401
// ... 
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   402
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   403
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   404
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   405
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   406
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   407
// Map type (upper-case)
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   408
//=======================
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   409
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   410
// Note the difference between map and Map
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   411
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   412
val m = Map(1 -> "one", 2 -> "two", 10 -> "many")
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   413
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   414
List((1, "one"), (2, "two"), (10, "many")).toMap
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   415
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   416
m.get(1)
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   417
m.get(4)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   418
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   419
m.getOrElse(1, "")
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   420
m.getOrElse(4, "")
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   421
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   422
val new_m = m + (10 -> "ten")
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   423
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   424
new_m.get(10)
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   425
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   426
val m2 = for ((k, v) <- m) yield (k, v.toUpperCase)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   427
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   428
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   429
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   430
// groupBy function on Maps
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   431
val lst = List("one", "two", "three", "four", "five")
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   432
lst.groupBy(_.head)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   433
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   434
lst.groupBy(_.length)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   435
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   436
lst.groupBy(_.length).get(3)
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   437
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   438
val grps = lst.groupBy(_.length)
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   439
grps.keySet
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   440
478
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   441
// naive quicksort with "On" function
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   442
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   443
def sortOn(f: Int => Int, xs: List[Int]) : List[Int] = {
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   444
  if (xs.size < 2) xs
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   445
  else {
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   446
   val pivot = xs.head
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   447
   val (left, right) = xs.partition(f(_) < f(pivot))
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   448
   sortOn(f, left) ::: pivot :: sortOn(f, right.tail)
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   449
  }
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   450
} 
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   451
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   452
sortOn(identity, List(99,99,99,98,10,-3,2)) 
0e6ca70496c1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 468
diff changeset
   453
sortOn(n => - n, List(99,99,99,98,10,-3,2))
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   454
51
0e60e6c24b99 updated
Christian Urban <urbanc@in.tum.de>
parents: 39
diff changeset
   455
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   456
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   457
// Pattern Matching
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   458
//==================
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   459
468
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
   460
// A powerful tool which has even landed in Java during 
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
   461
// the last few years (https://inside.java/2021/06/13/podcast-017/).
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
   462
// ...Scala already has it for many years and the concept is
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
   463
// older than your friendly lecturer, that is stone old  ;o)
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   464
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   465
// The general schema:
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   466
//
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   467
//    expression match {
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   468
//       case pattern1 => expression1
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   469
//       case pattern2 => expression2
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   470
//       ...
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   471
//       case patternN => expressionN
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   472
//    }
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   473
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   474
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   475
// recall
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   476
def my_map_int(lst: List[Int], f: Int => Int) : List[Int] = 
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   477
  lst match {
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   478
    case Nil => Nil
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   479
    case x::xs => f(x)::my_map_int(xs, f)
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   480
  }
58
93a2b6e4b84c updated
Christian Urban <urbanc@in.tum.de>
parents: 57
diff changeset
   481
468
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
   482
def my_map_option(opt: Option[Int], f: Int => Int) : Option[Int] = 
c71ae4477e55 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 444
diff changeset
   483
  opt match {
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   484
    case None => None
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   485
    case Some(x) => Some(f(x))
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   486
  }
58
93a2b6e4b84c updated
Christian Urban <urbanc@in.tum.de>
parents: 57
diff changeset
   487
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   488
my_map_option(None, x => x * x)
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   489
my_map_option(Some(8), x => x * x)
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   490
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   491
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   492
// you can also have cases combined
266
31e5218f43de updated to 2.13
Christian Urban <urbanc@in.tum.de>
parents: 212
diff changeset
   493
def season(month: String) : String = month match {
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   494
  case "March" | "April" | "May" => "It's spring"
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   495
  case "June" | "July" | "August" => "It's summer"
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   496
  case "September" | "October" | "November" => "It's autumn"
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   497
  case "December" => "It's winter"
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   498
  case "January" | "February" => "It's unfortunately winter"
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   499
  case _ => "Wrong month"
266
31e5218f43de updated to 2.13
Christian Urban <urbanc@in.tum.de>
parents: 212
diff changeset
   500
}
31e5218f43de updated to 2.13
Christian Urban <urbanc@in.tum.de>
parents: 212
diff changeset
   501
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   502
// pattern-match on integers
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   503
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   504
def fib(n: Int) : Int = n match { 
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   505
  case 0 | 1 => 1
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   506
  case n => fib(n - 1) + fib(n - 2)
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   507
}
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   508
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   509
fib(10)
266
31e5218f43de updated to 2.13
Christian Urban <urbanc@in.tum.de>
parents: 212
diff changeset
   510
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   511
// Silly: fizz buzz
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   512
def fizz_buzz(n: Int) : String = (n % 3, n % 5) match {
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   513
  case (0, 0) => "fizz buzz"
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   514
  case (0, _) => "fizz"
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   515
  case (_, 0) => "buzz"
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   516
  case _ => n.toString  
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   517
}
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   518
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   519
for (n <- 1 to 20) 
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   520
 println(fizz_buzz(n))
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   521
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   522
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   523
val lst = List(None, Some(1), Some(2), None, Some(3)).flatten
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   524
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   525
def my_flatten(xs: List[Option[Int]]): List[Int] = 
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   526
 xs match {
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   527
   case Nil => Nil 
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   528
   case None::rest => my_flatten(rest)
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   529
   case Some(v)::rest => v :: my_flatten(rest)
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   530
 }
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   531
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   532
my_flatten(List(None, Some(1), Some(2), None, Some(3)))
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   533
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   534
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   535
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   536
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   537
// Recursion
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   538
//===========
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   539
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   540
// my_length
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   541
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   542
def my_length(xs: List[Int]) : Int = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   543
  if (xs == Nil) 0 else 1 + my_length(xs.tail)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   544
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   545
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   546
def my_sum(xs: List[Int]) : Int = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   547
  if (xs == Nil) 0 else xs.head + my_sum(xs.tail)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   548
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   549
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   550
my_sum((1 to 100).toList)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   551
my_length(List())
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   552
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   553
/* my_map */
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   554
for (n <- List[Int](1,2,3)) yield n * n
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   555
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   556
List(1,2,3) ::: List(3,4,5)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   557
3 :: List(3,4,5)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   558
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   559
def my_map(xs: List[Int], f : Int => Int ) : List[Int] = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   560
  if (xs == Nil) Nil 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   561
  else f(xs.head) :: my_map(xs.tail, f)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   562
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   563
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   564
def square(n: Int) : Int = n * n
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   565
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   566
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   567
my_map(List(1,2,3), square)
365
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   568
08241d957be4 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 364
diff changeset
   569
278
57b5bba67467 updated
Christian Urban <urbanc@in.tum.de>
parents: 268
diff changeset
   570
57b5bba67467 updated
Christian Urban <urbanc@in.tum.de>
parents: 268
diff changeset
   571
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   572
/* powerset */
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   573
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   574
def powerset(xs: Set[Int]) : Set[Set[Int]] = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   575
  if (xs == Set()) Set(Set())
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   576
  else powerset(xs.tail) ++ powerset(xs.tail).map(_ + xs.head)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   577
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   578
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   579
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   580
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   581
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   582
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   583
/* on lists */
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   584
def powerset(xs: List[Int]) : List[List[Int]] = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   585
  if (xs == Nil) List(Nil)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   586
  else powerset(xs.tail) ::: powerset(xs.tail).map(xs.head :: _)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   587
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   588
309
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   589
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   590
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   591
/* Say you have characters a, b, c.
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   592
   What are all the combinations of a certain length?
309
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   593
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   594
   All combinations of length 2:
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   595
  
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   596
     aa, ab, ac, ba, bb, bc, ca, cb, cc
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   597
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   598
   Combinations of length 3:
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   599
   
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   600
     aaa, baa, caa, and so on......
309
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   601
*/
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   602
320
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   603
def combs(cs: List[Char], n: Int) : List[String] = {
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   604
  if (n == 0) List("")
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   605
  else for (c <- cs; s <- combs(cs, n - 1)) yield s"$c$s"
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   606
}
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   607
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   608
combs(List('a', 'b', 'c'), 3)
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   609
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   610
90aed247c8cf updated
Christian Urban <urbanc@in.tum.de>
parents: 319
diff changeset
   611
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   612
def combs(cs: List[Char], l: Int) : List[String] = {
309
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   613
  if (l == 0) List("")
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   614
  else for (c <- cs; s <- combs(cs, l - 1)) yield s"$c$s"
309
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   615
}
c5f16a86ae92 updated
Christian Urban <urbanc@in.tum.de>
parents: 278
diff changeset
   616
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   617
combs("abc".toList, 2)
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   618
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   619
329
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   620
// When writing recursive functions you have to
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   621
// think about three points
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   622
// 
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   623
// - How to start with a recursive function
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   624
// - How to communicate between recursive calls
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   625
// - Exit conditions
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   626
828326d1b3b2 updated
Christian Urban <urbanc@in.tum.de>
parents: 323
diff changeset
   627
147
3e5d8657302f updated
Christian Urban <urbanc@in.tum.de>
parents: 95
diff changeset
   628
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   629
// A Recursive Web Crawler / Email Harvester
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   630
//===========================================
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   631
//
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   632
// the idea is to look for links using the
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   633
// regular expression "https?://[^"]*" and for
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   634
// email addresses using another regex.
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   635
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   636
import io.Source
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   637
import scala.util._
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   638
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   639
// gets the first 10K of a web-page
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   640
def get_page(url: String) : String = {
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   641
  Try(Source.fromURL(url)(using "ISO-8859-1").take(10000).mkString).
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   642
    getOrElse { println(s"  Problem with: $url"); ""}
147
3e5d8657302f updated
Christian Urban <urbanc@in.tum.de>
parents: 95
diff changeset
   643
}
3e5d8657302f updated
Christian Urban <urbanc@in.tum.de>
parents: 95
diff changeset
   644
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   645
// regex for URLs and emails
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   646
val http_pattern = """"https?://[^"]*"""".r
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   647
val email_pattern = """([a-z0-9_\.-]+)@([\da-z\.-]+)\.([a-z\.]{2,6})""".r
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   648
268
d20583497c5b updated
Christian Urban <urbanc@in.tum.de>
parents: 266
diff changeset
   649
//test case:
212
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   650
//email_pattern.findAllIn
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   651
//  ("foo bla christian@kcl.ac.uk 1234567").toList
c86e40fb3b21 updated
Christian Urban <urbanc@in.tum.de>
parents: 204
diff changeset
   652
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   653
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   654
// drops the first and last character from a string
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   655
def unquote(s: String) = s.drop(1).dropRight(1)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   656
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   657
def get_all_URLs(page: String): Set[String] = 
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   658
  http_pattern.findAllIn(page).map(unquote).toSet
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   659
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   660
// naive version of crawl - searches until a given depth,
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   661
// visits pages potentially more than once
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   662
def crawl(url: String, n: Int) : Unit = {
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   663
  if (n == 0) ()
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   664
  else {
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   665
    println(s"  Visiting: $n $url")
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   666
    for (u <- get_all_URLs(get_page(url))) crawl(u, n - 1)
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   667
  }
147
3e5d8657302f updated
Christian Urban <urbanc@in.tum.de>
parents: 95
diff changeset
   668
}
3e5d8657302f updated
Christian Urban <urbanc@in.tum.de>
parents: 95
diff changeset
   669
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   670
// some starting URLs for the crawler
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   671
//val startURL = """https://nms.kcl.ac.uk/christian.urban/"""
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   672
val startURL = """https://urbanchr.github.io/"""
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   673
147
3e5d8657302f updated
Christian Urban <urbanc@in.tum.de>
parents: 95
diff changeset
   674
204
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   675
crawl(startURL, 2)
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   676
1b04ea68dca6 updated
Christian Urban <urbanc@in.tum.de>
parents: 192
diff changeset
   677
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   678
// a primitive email harvester
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   679
def emails(url: String, n: Int) : Set[String] = {
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   680
  if (n == 0) Set()
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   681
  else {
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   682
    println(s"  Visiting: $n $url")
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   683
    val page = get_page(url)
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   684
    val new_emails = email_pattern.findAllIn(page).toSet
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   685
    new_emails ++ (for (u <- get_all_URLs(page)) yield emails(u, n - 1)).flatten
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   686
  }
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   687
}
55
6610c1dfa8a9 updated
Christian Urban <urbanc@in.tum.de>
parents: 53
diff changeset
   688
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   689
emails(startURL, 3)
55
6610c1dfa8a9 updated
Christian Urban <urbanc@in.tum.de>
parents: 53
diff changeset
   690
6610c1dfa8a9 updated
Christian Urban <urbanc@in.tum.de>
parents: 53
diff changeset
   691
318
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   692
// if we want to explore the internet "deeper", then we
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   693
// first have to parallelise the request of webpages:
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   694
//
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   695
// scala -cp scala-parallel-collections_2.13-0.2.0.jar 
f1215a72cd88 updated
Christian Urban <urbanc@in.tum.de>
parents: 317
diff changeset
   696
// import scala.collection.parallel.CollectionConverters._
55
6610c1dfa8a9 updated
Christian Urban <urbanc@in.tum.de>
parents: 53
diff changeset
   697
53
9f8751912560 updated
Christian Urban <urbanc@in.tum.de>
parents: 51
diff changeset
   698
9f8751912560 updated
Christian Urban <urbanc@in.tum.de>
parents: 51
diff changeset
   699
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   700
def powerset(xs: Set[Int]) : Set[Set[Int]] = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   701
  if (xs == Set()) Set(Set())
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   702
  else {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   703
    val ps = powerset(xs.tail)  
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   704
    ps ++ ps.map(_ + xs.head)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   705
  }
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   706
}  
53
9f8751912560 updated
Christian Urban <urbanc@in.tum.de>
parents: 51
diff changeset
   707
504
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   708
def psubsets(xs: Set[Int]) = 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   709
  powerset(xs) -- Set(Set(), xs) 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   710
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   711
//def psubsets(xs: Set[Int]) = 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   712
//  xs.subsets.toList -- Set(Set(), xs)  
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   713
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   714
def splits(xs: Set[Int]) :  Set[(Set[Int], Set[Int])] =
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   715
  psubsets(xs).map(s => (s, xs -- s))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   716
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   717
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   718
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   719
enum Tree {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   720
  case Num(i: Int)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   721
  case Add(l: Tree, r: Tree)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   722
  case Sub(l: Tree, r: Tree)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   723
  case Mul(l: Tree, r: Tree)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   724
  case Div(l: Tree, r: Tree)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   725
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   726
import Tree._
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   727
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   728
def pp(tr: Tree) : String = tr match {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   729
  case Num(n) => s"$n"
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   730
  case Add(l, r) => s"(${pp(l)} + ${pp(r)})"
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   731
  case Sub(l, r) => s"(${pp(l)} - ${pp(r)})"
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   732
  case Mul(l, r) => s"(${pp(l)} * ${pp(r)})"
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   733
  case Div(l, r) => s"(${pp(l)} / ${pp(r)})"
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   734
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   735
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   736
def search(nums: Set[Int]) : Set[Tree] =  nums.size match {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   737
  case 0 => Set()
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   738
  case 1 => Set(Num(nums.head))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   739
  case 2 => {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   740
    val l = nums.head
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   741
    val r = nums.tail.head
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   742
    Set(Add(Num(l), Num(r)), 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   743
        Mul(Num(l), Num(r)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   744
        ++ Option.when(l <= r)(Sub(Num(r), Num(l)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   745
        ++ Option.when(l > r)(Sub(Num(l), Num(r)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   746
        ++ Option.when(r > 0 && l % r == 0)(Div(Num(l), Num(r)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   747
        ++ Option.when(l > 0 && r % l == 0)(Div(Num(r), Num(l)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   748
  }
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   749
  case xs => {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   750
    val spls = splits(nums)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   751
    val subtrs = 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   752
      for ((lspls, rspls) <- spls;
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   753
           lt <- search(lspls); 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   754
          rt <- search(rspls)) yield {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   755
        Set(Add(lt, rt), Sub(lt, rt),
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   756
            Mul(lt, rt), Div(lt, rt))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   757
    } 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   758
    subtrs.flatten
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   759
  }
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   760
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   761
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   762
println(search(Set(1,2,3,4)).mkString("\n"))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   763
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   764
def eval(tr: Tree) : Option[Int] = tr match {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   765
  case Num(n) => Some(n)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   766
  case Add(l, r) => 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   767
    for (rl <- eval(l); rr <- eval(r)) yield rl + rr
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   768
  case Mul(l, r) => 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   769
    for (rl <- eval(l); rr <- eval(r)) yield rl * rr  
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   770
  case Sub(l, r) => 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   771
    for (rl <- eval(l); rr <- eval(r);
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   772
         if 0 <= rl - rr) yield rl - rr   
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   773
  case Div(l, r) => 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   774
    for (rl <- eval(l); rr <- eval(r);
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   775
         if rr > 0 && rl % rr == 0) yield rl / rr          
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   776
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   777
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   778
eval(Add(Num(1), Num(2)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   779
eval(Mul(Add(Num(1), Num(2)), Num(4)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   780
eval(Sub(Num(3), Num(2)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   781
eval(Sub(Num(3), Num(6)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   782
eval(Div(Num(6), Num(2)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   783
eval(Div(Num(6), Num(4)))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   784
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   785
def time_needed[T](n: Int, code: => T) = {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   786
  val start = System.nanoTime()
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   787
  for (i <- (0 to n)) code
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   788
  val end = System.nanoTime()
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   789
  (end - start) / 1.0e9
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   790
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   791
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   792
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   793
import scala.collection.parallel.CollectionConverters._
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   794
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   795
def check(xs: Set[Int], target: Int) =
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   796
  search(xs).find(eval(_) == Some(target))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   797
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   798
for (sol <- check(Set(50, 5, 4, 9, 10, 8), 560)) {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   799
  println(s"${pp(sol)} => ${eval(sol)}")
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   800
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   801
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   802
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   803
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   804
time_needed(1, check(Set(50, 5, 4, 9, 10, 8), 560))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   805
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   806
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   807
println(check(Set(25, 5, 2, 10, 7, 1), 986).mkString("\n"))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   808
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   809
for (sol <- check(Set(25, 5, 2, 10, 7, 1), 986)) {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   810
  println(s"${pp(sol)} => ${eval(sol)}")
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   811
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   812
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   813
for (sol <- check(Set(25, 5, 2, 10, 7, 1), -1)) {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   814
  println(s"${pp(sol)} => ${eval(sol)}")
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   815
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   816
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   817
for (sol <- check(Set(100, 25, 75, 50, 7, 10), 360)) {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   818
  println(s"${pp(sol)} => ${eval(sol)}")
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   819
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   820
time_needed(1, check(Set(100, 25, 75, 50, 7, 10), 360))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   821
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   822
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   823
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   824
time_needed(1, check(Set(25, 5, 2, 10, 7, 1), 986))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   825
time_needed(1, check(Set(25, 5, 2, 10, 7, 1), -1))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   826
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   827
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   828
def generate(nums: Set[Int]) : Set[(Tree, Int)] =  nums.size match {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   829
  case 0 => Set()
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   830
  case 1 => Set((Num(nums.head), nums.head))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   831
  case xs => {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   832
    val spls = splits(nums)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   833
    val subtrs =
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   834
      for ((lspls, rspls) <- spls;
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   835
           (lt, ln) <- generate(lspls); 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   836
           (rt, rn) <- generate(rspls)) yield {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   837
        Set((Add(lt, rt), ln + rn),
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   838
            (Mul(lt, rt), ln * rn))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   839
        ++ Option.when(ln <= rn)((Sub(rt, lt), rn - ln)) 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   840
        ++ Option.when(ln > rn)((Sub(lt, rt), ln - rn))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   841
        ++ Option.when(rn > 0 && ln % rn == 0)((Div(lt, rt), ln / rn))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   842
        ++ Option.when(ln > 0 && rn % ln == 0)((Div(rt, lt), rn / ln))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   843
    } 
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   844
    subtrs.flatten
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   845
  }
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   846
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   847
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   848
def check2(xs: Set[Int], target: Int) =
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   849
  generate(xs).find(_._2 == target)
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   850
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   851
for ((sol, ev) <- check2(Set(50, 5, 4, 9, 10, 8), 560)) {
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   852
  println(s"${pp(sol)} => ${eval(sol)} / $ev")
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   853
}
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   854
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   855
time_needed(1, check(Set(50, 5, 4, 9, 10, 8), 560))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   856
time_needed(1, check2(Set(50, 5, 4, 9, 10, 8), 560))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   857
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   858
time_needed(1, check(Set(50, 5, 4, 9, 10, 8), -1))
7653dd662db3 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 491
diff changeset
   859
time_needed(1, check2(Set(50, 5, 4, 9, 10, 8), -1))
192
cd2a9c969ef2 updated
Christian Urban <urbanc@in.tum.de>
parents: 174
diff changeset
   860
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   861
// Jumping Towers
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   862
//================
278
57b5bba67467 updated
Christian Urban <urbanc@in.tum.de>
parents: 268
diff changeset
   863
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   864
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   865
def moves(xs: List[Int], n: Int) : List[List[Int]] = 
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   866
 (xs, n) match {
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   867
   case (Nil, _) => Nil
366
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   868
   case (_, 0) => Nil
364
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   869
   case (x::xs, n) => (x::xs) :: moves(xs, n - 1)
18942af74fa1 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 363
diff changeset
   870
 }
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   871
366
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   872
// List(5,5,1,0) -> moves(List(5,1,0), 5)
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   873
moves(List(5,1,0), 1)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   874
moves(List(5,1,0), 2)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   875
moves(List(5,1,0), 5)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   876
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   877
// checks whether a jump tour exists at all
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   878
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   879
def search(xs: List[Int]) : Boolean = xs match {
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   880
  case Nil => true
366
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   881
  case x::xs =>
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   882
    if (xs.length < x) true 
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   883
    else moves(xs, x).exists(search(_))
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   884
}
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   885
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   886
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   887
search(List(5,3,2,5,1,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   888
search(List(3,5,1,0,0,0,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   889
search(List(3,5,1,0,0,0,0,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   890
search(List(3,5,1,0,0,0,1,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   891
search(List(3,5,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   892
search(List(5,1,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   893
search(Nil)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   894
search(List(1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   895
search(List(5,1,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   896
search(List(3,5,1,0,0,0,0,0,0,0,0,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   897
366
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   898
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   899
import scala.util._
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   900
List.fill(100)(Random.nextInt(2))
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   901
search(List.fill(100)(Random.nextInt(10)))
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   902
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   903
// generate *all* jump tours
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   904
//    if we are only interested in the shortes one, we could
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   905
//    shortcircut the calculation and only return List(x) in
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   906
//    case where xs.length < x, because no tour can be shorter
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   907
//    than 1
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   908
// 
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   909
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   910
def jumps(xs: List[Int]) : List[List[Int]] = xs match {
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   911
  case Nil => Nil
366
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   912
  case x::xs => {
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   913
    val children = moves(xs, x)
366
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   914
    val results = 
d2f895c1dba6 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 365
diff changeset
   915
      children.map(cs => jumps(cs).map(x :: _)).flatten
319
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   916
    if (xs.length < x) List(x) :: results else results
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   917
  }
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   918
}
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   919
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   920
jumps(List(3,5,1,2,1,2,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   921
jumps(List(3,5,1,2,3,4,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   922
jumps(List(3,5,1,0,0,0,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   923
jumps(List(3,5,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   924
jumps(List(5,1,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   925
jumps(Nil)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   926
jumps(List(1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   927
jumps(List(5,1,2))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   928
moves(List(1,2), 5)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   929
jumps(List(1,5,1,2))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   930
jumps(List(3,5,1,0,0,0,0,0,0,0,0,1))
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   931
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   932
jumps(List(5,3,2,5,1,1)).minBy(_.length)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   933
jumps(List(1,3,5,8,9,2,6,7,6,8,9)).minBy(_.length)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   934
jumps(List(1,3,6,1,0,9)).minBy(_.length)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   935
jumps(List(2,3,1,1,2,4,2,0,1,1)).minBy(_.length)
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   936
ed7543c5d317 updated
Christian Urban <urbanc@in.tum.de>
parents: 318
diff changeset
   937