Lisp: список узлов на заданном уровне k

Question

Я хочу найти все атомы из дерева, расположенные на заданном уровне k. Я пытался что-то вроде:

(defun atoms (l) 
    ;returns the list of atoms of l at its superficial level 
    (cond 
    ((null l) l) 
    ((or (atom l) (numberp l)) l) 
    ((or (atom (car l)) (numberp (car l))) 
     (append (list(car l)) (atom (cdr l)))) 
    (T (atoms (Cdr l))))) 

(defun findat (l pos k) 
    (cond 
    ((null l) l) 
    ((= k pos) (atoms l)) 
    ((and (or (atom l) (numberp l)) (= k pos)) l) 
    (T (cons '() (mapcar #'(lambda (l) (findat l (+ pos 1) k)) l))))) 

так, для образца: л = (а (Ь (г)) (с (д (е)) (е))), поз = 0 и к = 2, я должен получить результат: (gdf), но вместо этого, я получаю некоторую ошибку, говоря, что «A не имеет тип LIST». Кто-нибудь знает, как исправить мой код? Заранее спасибо!

Answer 1

Вот несколько измененная версия вашего findat. Возможно, вам захочется подумать о более лучшем имени для функции (по крайней мере, выписать ее полностью (find-atoms) вместо того, чтобы аббревиатура излишне).

(defun findat (l k) 
    "Find atoms on level K in list L." 
    (cond 
    ((atom l) '()) 
    ((= k 0) (remove-if (complement #'atom) l)) 
    (t (mapcan #'(lambda (l) 
        (findat l (1- k))) 
       l)))) 

(findat '(a (b (g)) (c (d (e)) (f))) 2) 
; => (G D F) 

Изменения:

• Я удалил аргумент pos. Легче просто пересчитать k.
• Я изменил первый случай, чтобы проверить наличие атомов (включая nil). Он возвращает пустой список, чтобы append в последнем случае отменил его. Фактически возвращение желаемых атомов обрабатывается вторым случаем.
• Я использовал стандарт remove-if вместо вашего atoms. Вызов complement создает предикат, который соответствует любому, что не является атомом. Таким образом, remove-if содержит только атомы. Вы также можете использовать remove-if-not и оставить complement, но это устарело.
• В последнем случае я использовал (reduce #'append ...) для создания плоского списка атомов. Редактировать: Изменено для использования mapcan.

Обратите внимание также, как я положил строку в начале функции. Это докшрин. Вы должны сделать это вместо того, чтобы оставлять комментарий там, как в вашем atoms. Таким образом вы можете использовать (describe #'findat), чтобы увидеть документацию (или вы можете использовать Emacs/IDE для ее просмотра).