aboutsummaryrefslogtreecommitdiff
path: root/racket/leetcode/lc-999-available-captures.rkt
diff options
context:
space:
mode:
authorH.J <thechairman@thechairman.info>2024-10-09 11:36:55 -0400
committerH.J <thechairman@thechairman.info>2024-10-09 11:36:55 -0400
commit8777ff071f7bb37631baa7b6717ad29961e50911 (patch)
tree6d59c4ed58e454b960339c3d1151f0a879e8d7cb /racket/leetcode/lc-999-available-captures.rkt
parent6156a9ef7be4012063a042aafb4e9b0d7eadde8e (diff)
downloadgleam_aoc-8777ff071f7bb37631baa7b6717ad29961e50911.tar.gz
gleam_aoc-8777ff071f7bb37631baa7b6717ad29961e50911.zip
sorting by language
Diffstat (limited to 'racket/leetcode/lc-999-available-captures.rkt')
-rw-r--r--racket/leetcode/lc-999-available-captures.rkt28
1 files changed, 28 insertions, 0 deletions
diff --git a/racket/leetcode/lc-999-available-captures.rkt b/racket/leetcode/lc-999-available-captures.rkt
new file mode 100644
index 0000000..1b1a3a9
--- /dev/null
+++ b/racket/leetcode/lc-999-available-captures.rkt
@@ -0,0 +1,28 @@
+#lang racket
+(define/contract (num-rook-captures board)
+ (-> (listof (listof string?)) exact-integer?)
+
+ (define (get-rook-space [board-state board])
+ (for/or ([board-rank (in-list board-state)]
+ [rank (in-range 0 (length board-state))]
+ #:when (index-of board-rank "R"))
+ (list rank (index-of board-rank "R"))))
+
+ (define (check-for-capturable-pawns spaces)
+ (match spaces
+ [(list _ ... "p" "." ... "R" "." ... "p" _ ...) 2]
+ [(list _ ... "R" "." ... "p" _ ...) 1]
+ [(list _ ... "p" "." ... "R" _ ...) 1]
+ [_ 0]))
+
+ (define (check-rank rank [board-state board])
+ (let ([spaces (list-ref board-state rank)])
+ (check-for-capturable-pawns spaces)))
+
+ (define (check-file file [board-state board])
+ (let ([spaces (map (curryr list-ref file) board)])
+ (check-for-capturable-pawns spaces)))
+
+ (match (get-rook-space board)
+ [(list rank file) (+ (check-rank rank)
+ (check-file file))])) \ No newline at end of file