summaryrefslogtreecommitdiffstats
path: root/net/ricketyspace/sicp/two/fortyone.scm
blob: 9ab708cb4f808201afae9c7a229e46ad0d543b84 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
;;;; License: CC0-1.0
;;;;
;;;;  ___________________________________
;;;; < let's make some distinct bastards >
;;;;  -----------------------------------
;;;;      \
;;;;       \
;;;;              ,;;;;;;;,
;;;;             ;;;;;;;;;;;,
;;;;            ;;;;;'_____;'
;;;;            ;;;(/))))|((\
;;;;            _;;((((((|))))
;;;;           / |_\\\\\\\\\\\\
;;;;      .--~(  \ ~))))))))))))
;;;;     /     \  `\-(((((((((((\\
;;;;     |    | `\   ) |\       /|)
;;;;      |    |  `. _/  \_____/ |
;;;;       |    , `\~            /
;;;;        |    \  \           /
;;;;       | `.   `\|          /
;;;;       |   ~-   `\        /
;;;;        \____~._/~ -_,   (\
;;;;         |-----|\   \    ';;
;;;;        |      | :;;;'     \
;;;;       |  /    |            |
;;;;       |       |            |
;;;;


(define-module (net ricketyspace sicp two fortyone)
  #:export (bastard-triplets))


(define (accumulate op initial sequence)
  (if (null? sequence)
      initial
      (op (car sequence)
          (accumulate op initial (cdr sequence)))))


(define (enumerate-interval low high)
  (if (> low high)
      '()
      (cons low (enumerate-interval (+ low 1) high))))


(define (flatmap proc seq)
  (accumulate append '() (map proc seq)))


(define (bastard-triplets n s)
  (define (spawn-triplet i n)
    (flatmap (lambda (j)
               (map (lambda (k) (list i j k))
                    (enumerate-interval 1 n)))
             (enumerate-interval 1 n)))

  (define (spawn-triplets n)
    (flatmap (lambda (i) (spawn-triplet i n))
             (enumerate-interval 1 n)))

  (define (distinct-sum-equals-s? triplet)
    (let ((i (car triplet))
          (j (cadr triplet))
          (k (caddr triplet)))
      (and
       ;; check if i, j, k are distinct.
       (cond ((= i j) #f)
             ((= i k) #f)
             ((= j k) #f)
             (else #t))
       ;; check if i + j + k = s
       (= (+ i j k) s))))

  (filter distinct-sum-equals-s? (spawn-triplets n)))

;;;
;;; Guile REPL
;;;
;;; scheme@(guile-user)> ,re (net ricketyspace sicp two fortyone)
;;; scheme@(guile-user)> (bastard-triplets 10 6)
;;; $8 = ((1 2 3) (1 3 2) (2 1 3) (2 3 1) (3 1 2) (3 2 1))
;;;