Сначала извлеките страну, связанную с входной строкой:
(define (get-country l s)
(cond [(empty? (rest l)) (second (second (first l)))]
[else (if (equal? s (first (first l)))
(second (second (first l)))
(get-country (rest l) s))]))
Затем извлеките все строки, с которыми связана эта страна:
(define (get-countries l s)
(cond [(empty? l) '()]
[else (if (equal? s (second (second (first l))))
(cons (first (first l)) (get-countries (rest l) s))
(get-countries (rest l) s))]))
Затем сложите их вместе:
(define (same-country l s)
(get-countries l (get-country l s)))
Когда мы оцениваем, мы получаем результат, отличный от (list "YUL" "YVR" "YWG" "YYZ")
:
> (same-country alist "YUL")
(list "YYZ" "YWG" "YUL" "YVR")
Поэтому мы проверяем, является ли результат перестановкой нужного списка. Сначала мы делаем is-permutation
:
(define (is-permutation l1 l2)
(and (not (and (cons? l1) (empty? l2)))
(not (and (empty? l1) (cons? l2)))
(or (and (empty? l1) (empty? l2))
(and (is-member (first l1) l2)
(is-permutation (rest l1)
(remove-one (first l1) l2))))))
(define (is-member e l)
(and (not (empty? l))
(or (equal? (first l) e)
(is-member e (rest l)))))
(define (remove-one e nel)
(cond [(empty? (rest nel)) '()]
[else (if (equal? (first nel) e)
(rest nel)
(cons (first nel) (remove-one e (rest nel))))]))
Затем мы можем проверить:
> (is-permutation (same-country alist "YUL")
(list "YUL" "YVR" "YWG" "YYZ"))
#true